aboutsummaryrefslogtreecommitdiff
path: root/setup.py
blob: 6f8ef5140a82c616b8f6873266403e12b1dfb94c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
#!/bin/env python3

"""Interactively install fischerlings dotfiles"""

import argparse
import os
import sys
import subprocess

home_dir = os.getenv("HOME")
cwd = os.getcwd()

def fish_config(quiet):
    """install and run fisherman"""
    # download fisherman
    import urllib.request

    if not quiet:
        print("Installing fisherman, see: https://github.com/fisherman/fisherman")
    with urllib.request.urlopen("https://git.io/fisherman") as f:
        fisher = f.read().decode("utf-8")

    dir_path = home_dir+"/.config/fish/functions/"
    # install fisherman at $fish_config/functions/
    if not os.path.exists(dir_path):
        os.mkdir(dir_path)
    else:
        if not os.path.exists(dir_path+"fisher.fish"):
            file = open(dir_path+"fisher.fish", 'w')
            file.write(fisher)
        else:
            if not quiet:
                print("fisherman already installed")
                return

    if not quiet:
        print("Installed fisherman -> execute fisher to pull plugins")

    subprocess.run(["fish", dir_path+"fisher.fish"])

# targets are a list of tuples<file name, link destination> and/or functions

targets = {
        "vimrc":
            [("vimrc", home_dir + "/.vimrc")],
        "vim.d":
            [("vim", home_dir + "/.vim")],
        "ssh-config":
            [("ssh/config", home_dir + "/.ssh/config")],
        "xinitrc":
            [("xinitrc", home_dir + "/.xinitrc")],
        "Xresources":
            [("Xresources", home_dir + "/.Xresources")],
        "mutt":
            [("muttrc", home_dir + "/.muttrc"),
            ("mutt", home_dir + "/.mutt"),
            ("scripts/mutt.sh", home_dir+"/.local/bin/mutt.sh")],
        "zshrc":
            [("zshrc", home_dir + "/zshrc")],
        "i3-config":
            [("i3", home_dir + "/.i3")],
        "dir_colors":
            [("dir_colors", home_dir + "/.dir_colors")],
        "fish-config":
            [("config/fish/config.fish", home_dir + "/.config/fish/config.fish"),
            ("config/fish/fishfile", home_dir + "/.config/fish/fishfile"),
            fish_config,
            ("fau_stud.conf.gpg", "/etc/wpa_supplicant/fau_stud.conf")],
        "terminator-config":
            [("config/terminator/config", home_dir + "/.config/terminator/config")]
        }

git_submodules_for = ["vim.d"]

def install_target(target, quiet):
    """Install a target"""

    if not quiet:
        print("Installing", target, "...")

    for instruction in targets[target]:
        if isinstance(instruction, tuple):
            if not quiet:
                print("Installing", instruction[0], "to", instruction[1])
            if os.path.exists(instruction[0]):
                # check if file is encrypted
                if instruction[0][-4:] == ".gpg":
                    if subprocess.run(["gpg", "--output", "decrypted/" +
                                instruction[0][:-4], "-d", instruction[0]]): 
                        instruction = ("decrypted/" + instruction[0][:-4],
                                        instruction[1])

                        # set privacy friendly file permissions 
                        if not subprocess.run(["chmod", "660", instruction[0]]):
                            print("Skipping because changing permissions of",
                                    instruction[1], "failed. This could be a privacy risk!")
                            continue
                    else:
                        print("Decrypting failed", file=sys.stderr)
                        continue

                try:
                    os.symlink(cwd + "/" + instruction[0], instruction[1])
                except Exception as e:
                    if isinstance(e, PermissionError):
                        subprocess.run(["sudo", "ln", "-s",
                                cwd + '/' + instruction[0], instruction[1]])
                    elif isinstance(e, FileExistsError):
                        print("File already exists")
                    else:
                        print(e, file=sys.stderr)
            else:
                print("Can't find", instruction[0] +
                        ". Please make sure you are in the right directory.", file=sys.stderr)
        elif callable(instruction):
            if not quiet:
                print("Running function", instruction)
            res = instruction(quiet)
            if res:
                if not quiet:
                    print("The instruction", instruction, "of", target, "failed with", str(res) + ".")
                    print("This can leaf your installation in a broken state")
                    print("ABORTING to prevent failures.")
                exit(res)
        else:
            print("I don't understand", instruction,
                    "\nThe instruction musst be a tuple or a function.\nSKIPPING IT")

    if target in git_submodules_for:
        choice = input(target + " uses git submodules want to pull them ? (Y/n) ")
        if choice in ["Y", "y", ""]:
            subprocess.run(["git", "submodule", "update", "--init"])


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("-q", "--quiet", action="store_true")
    parser.add_argument("targets", type=str, nargs='*')

    args = parser.parse_args()

    if args.targets:
        for t in args.targets:
            install_target(t, args.quiet)
    else:
        if not args.quiet:
            print("Welcome to the installation script of fischerlings dotfiles.")
            print("Located at: https://github.com/fischerling/dotfiles")
            print()

        targets_ord = [x for x in targets]
        for i, target in zip(range(len(targets_ord)), targets_ord):
            print(str(i)+":", target)
        print("Enter n° of configurations to be installed or all")
        choice = input("==> ")
        
        if choice == "":
            pass
        elif "all" in choice:
            for target in targets:
                install_target(target, args.quiet)
        else:
            for n in choice.strip().split(" "):
                install_target(targets_ord[int(n)], args.quiet)

        if not args.quiet:
            print("Good Bye. Have fun with your new configuration")


if __name__ == "__main__":
    main()