link.py

#!/usr/bin/env python3

import os
import ctypes
import platform
import subprocess
import shutil
Linux = 'Linux'
Windows = 'Windows'

def link(src,dest,folder):
    if platform.system() == Windows:
        if folder == True:
            subprocess.run(['cmd','/c','mklink','/J',dest,src],shell=True)
        else:
            os.link(src,dest)
    elif platform.system() == Linux:
        if folder == True:
            os.symlink(src,dest)
        else:
            os.symlink(src,dest)
    

class LinkFile:
    # 'type' can be: 'msg', 'file', 'backupFile', 'dir'
    def __init__(self, origin_parts, destination_parts, type="file",message = None):
        self.type = type
        if not type == 'msg':
            self.origin = os.path.join(*origin_parts)
            self.destination = os.path.join(*destination_parts)
            self.target_is_directory = True if type == "dir" else False
            if type == 'backupFile':
                self.backup = os.path.join(*message)
            self.message = None
        else:
            self.origin = None
            self.destination = None
            self.target_is_directory = None
            self.message = message
    def exist_dest(self):
        exists = (os.path.islink(self.destination) or os.path.exists(self.destination)) 
        return exists

    def rm(self):
        if self.type== 'msg':
            return
        # TODO: (os.path.islink()) This does not work on windows with folder. 
        # But deleting the folder(as it currently does) fails, and the code 
        # moves on.
        if os.path.islink(self.destination):
            os.unlink(self.destination)
        elif os.path.exists(self.destination):
            if self.target_is_directory:
                shutil.rmtree(self.destination)
            else:
                os.remove(self.destination)

    # Works on linux
    def bkup(self):
        if self.type == 'msg':
            return
        shutil.move(self.destination,self.backup)

    def lnk(self):
        if self.type == 'msg':
            print('MESSAGE:    '+self.message)
            return

        if not os.path.exists(os.path.dirname(self.destination)):
            os.makedirs(os.path.dirname(self.destination))
        link(self.origin,self.destination,self.target_is_directory)

    def is_dest_link_origin(self):
        installed = False
        if os.path.islink(self.destination) and self.exist_dest():
            installed = os.readlink(self.destination) == self.origin
        return installed


class Program:
    def __init__(self, name, files):
        self.name = name
        self.files = files

    def install(self):
        for file in self.files:
            if file.type == 'msg':
                print('MESSAGE:    ' + file.message)
                return

            try:
                if file.is_dest_link_origin():
                    print(f'INFO   :    Already installed: {file.destination}' )
                    continue
                if file.exist_dest() and file.type != 'backupFile':
                    print('WARNING:    A different file already exits ... removing: ' + file.destination)
                    file.rm()
                if file.exist_dest() and file.type == 'backupFile':
                    print(f'INFO   :    Moving old file at: {file.destination}')
                    print(f'            to {file.backup}')
                    file.bkup()
                file.lnk()
                print('INFO   :    Linked to this file: ' + file.destination)
            except FileExistsError:
                print('ERROR  :    This shouldn\'t happen, file exists error: ' + file.destination)
                pass


def main():
    curr_dir = os.getcwd()

    if platform.system() == Windows:
        osConfDir = os.getenv('APPDATA')
        print("INFO   :    Platform is Windows, config directory is set to %APPDATA% -> ")
        print("       " + osConfDir)
    elif platform.system() == Linux:
        print("INFO   :    Platform is Linux")
        osConfDir = os.getenv('HOME') + '/.config'
        if "root" in osConfDir:
            print()
            print("User detected as root, do you want to choose a user in /home? If not the")
            print("script will instead install to the root user.")
            answer = input("Do you want to choose in /home? (y/n)")
            print()
            if answer.lower() == "y":
                allUsers = next(os.walk(r'/home'))[1]

                print("0 - Choose custom directory")
                i = 1
                for user in allUsers:
                    print(f"{i} - {user} (/home/{user})")
                    i = i + 1
                print()
                print("Chose a number from the list abbove to install to that directory. The mainly")
                print("used directories are '~/.config/' and some files such as '~/.bashrc'.")
                answer = input("Number: ")
                if int(answer) > 0 and int(answer) <= len(allUsers) :
                    osConfDir = f"/home/{allUsers[int(answer) - 1]}/.config"
                elif int(answer) == 0:
                    answer = input("The user dir is:")
                    osConfDir = answer
                else:
                    print("bad nubmer, exiting")
                    return
            print()
        print("INFO   :    Config directory is set to -> "+osConfDir )
    else:
        print('ERROR  :    This is not a windows or linux machine, update the script to work with this as well.')
        return 0
    print()

    programs = [
        # Program('Fastfetch', [
        #     LinkFile([curr_dir, 'fastfetch_presets', 'tami'], ['/usr/share/fastfetch/presets', 'tami'])
        # ]),
        Program('Fastfetch', [
            LinkFile([curr_dir, 'fastfetch', 'config.jsonc'], [osConfDir, 'fastfetch', 'config.jsonc'])
        ]),
        Program('Bash', [
            LinkFile([curr_dir,'bash','.bashrc'],[osConfDir,'..' , '.bashrc'], 'backupFile', [osConfDir, 'old.bashrc']),
            LinkFile([curr_dir,'bash','.profile'],[osConfDir,'..' , '.profile']),
            LinkFile([curr_dir,'bash','.inputrc'],[osConfDir,'..' , '.inputrc'])
        ]),
        Program('Kitty', [
            LinkFile([curr_dir, 'kitty'], [osConfDir, 'kitty'], 'dir')
        ]),
        Program('Ricemood', [
            LinkFile([curr_dir, 'ricemood'], [osConfDir, 'ricemood'], 'dir')
        ]),
        Program('Fish', [
            LinkFile([curr_dir, 'fish'], [osConfDir, 'fish'], 'dir')
        ]),
        Program('Tmux', [
            LinkFile([curr_dir, 'tmux.conf'], [osConfDir, 'tmux', 'tmux.conf'])
        ]),
        Program('Helix', [
            LinkFile([curr_dir, 'helix', 'config.toml'], [osConfDir, 'helix', 'config.toml']),
            LinkFile([curr_dir, 'helix', 'languages.toml'], [osConfDir, 'helix', 'languages.toml']),
            LinkFile([curr_dir, 'helix', 'themes'], [osConfDir, 'helix', 'themes'], 'dir'),
            LinkFile([curr_dir, "helix", "runtime", "queries" ], [osConfDir, "helix", "runtime", "queries" ], 'dir')
        ]),
        Program('Cmus', [
            LinkFile([curr_dir, 'cmus', 'autosave'], [osConfDir, 'cmus', 'autosave'])
        ]),
        Program('i3', [
            LinkFile([curr_dir, 'i3', 'config'], [osConfDir, 'i3', 'config']),
            LinkFile([curr_dir, 'i3', 'laptop'], [osConfDir, 'i3', 'laptop']),
            LinkFile([curr_dir, 'i3', 'default'], [osConfDir, 'i3', 'default'])
        ]),
        Program('picom', [
            LinkFile([curr_dir, 'picom', 'picom.conf'], [osConfDir, 'picom', 'picom.conf'])
        ]),
        Program('polybar', [
            LinkFile([curr_dir, 'polybar', 'launch.sh'], [osConfDir, 'polybar', 'launch.sh']),
            LinkFile([curr_dir, 'polybar', 'config.ini'], [osConfDir, 'polybar', 'config.ini'])
        ]),
        Program('rofi', [
            LinkFile([curr_dir, 'rofi' ], [osConfDir, 'rofi' ], 'dir'),
        ]),
        Program('starship', [
            LinkFile([curr_dir, 'starship.toml'], [osConfDir, 'starship.toml'])
        ]),
        Program('eww', [
            LinkFile([curr_dir, 'eww', 'eww.scss'], [osConfDir, 'eww', 'eww.scss']),
            LinkFile([curr_dir, 'eww', 'eww.yuck'], [osConfDir, 'eww', 'eww.yuck'])
        ]),
        Program('ghci', [
            LinkFile([curr_dir, '.ghci'], [osConfDir, '..' ,'.ghci'])
        ])
    ]

    print("This script will overwrite the folders as well. Don't run it if you aren't sure what you're doing.")
    answer = input("Do you want to proceed? (y/n): ")
    if answer.lower() != "y":
        print('Script execution aborted. Good choice! >.<')
        return

    if platform.system() == Windows:
        for program in [x for x in programs if x.name == "Helix"]:
            print(f'--- Installing: {program.name} ---')
            program.install()
    else:
        for program in programs:
            print(f'--- Installing: {program.name} ---')
            program.install()

    print("")
    print("Script execution completed. All required links created.")


if __name__ == "__main__":
    main()