firstboot/init.py at master · RaytingSec/firstboot · GitHub

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
#!/usr/bin/python3

import subprocess
import argparse
import time


Environment = 'gnome'  # 'cinnamon'
Machine = 'laptop'  # 'desktop'
Hypervisor = 'vmware'


parser = argparse.ArgumentParser()

parser.add_argument('--vm', action='store_true', default=False)
parser.add_argument('--gui', action='store_true', default=False)
parser.add_argument('--host', action='store_true', default=False)
parser.add_argument('--server', action='store_true', default=False)

parser.add_argument('--files', action='store_true', default=False)
parser.add_argument('--cat', action='store_true', default=False)

args = parser.parse_args()

sh = []

sh.append('core.sh')

# data=~/Data
# if [[ -d $data ]]; then
#     echo "Using ~/Data"
# elif [[ -d "/data/" ]]; then
#     echo "Using /data/"
#     ln -s /data/ ~/Data
# else
#     echo "Warning, no data directory found"
#     exit 1
# fi

if args.vm:
    sh.append('vm.sh')
if args.gui:
    sh.append('core-gui.sh')
    sh.append('packages-gui.sh')
    sh.append('sublime.sh')
    sh.append('arc-theme.sh')
    if Environment is 'gnome':
        sh.append('gnome.sh')
        sh.append('extensions.sh')
if args.vm and args.gui:
    if Environment is 'gnome':
        sh.append('gnome-vm.sh')
if args.host:
    if Machine is 'laptop':
        sh.append('laptop.sh')
    sh.append('directories.sh')
    sh.append('packages-host.sh')
    if Environment is 'gnome':
        sh.append('gnome-host.sh')
        sh.append('extensions-host.sh')
    # Applications
    sh.append('flashplayerdebugger.sh')
    sh.append('vmware.sh')
    sh.append('thinkorswim.sh')
if args.server:
    sh.append('server.sh')

for script in sh:
    if args.files:
        print(script)
    elif args.cat:
        with open(script) as file:
            contents = file.read()
        print('# {}'.format(script))
        print(contents)
    else:
        print('\nRunning {}\n'.format(script) + '=' * 30)
        time.sleep(3)
        subprocess.call(['bash', script])