GitSetup/git_setup.py at main · AsadRahim/GitSetup · 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
80
81
82
83
84
85
86
87
88
89
import os

from Crypto.PublicKey import RSA

HOME_DIR = os.path.expanduser("~")
PREFIX = ''
REPO = {
    'github': 'github.com',
    'gitlab': 'gitlab.com',
    'bikbucket': 'bitbucket.org'
}
GLOBAL_CONFIG = """
[includeIf "gitdir:{projects_path}"]
    path = ~/.gitconfigs/.gitconfig-{client_name}
"""
PROJECT_CONFIG = """
[core]
    compression = 0
[http]
    postBuffer = 1048576000
[user]
    name = {username}
    email = {email}
"""
SSH_CONFIG = """
Host {client_name}.{repo_host}
    HostName {repo_host}
    User git
    IdentityFile ~/.ssh/id_rsa_{client_name}
"""


def main():
    repo = input("Enter repo (github/gitlab/bitbucket): ")
    client_name = input("Enter client's name eg client1: ")
    projects_path = (input(
        ("Enter client's projects root path without project name eg. ~/PycharmProjects/client1/ "
         "(check forward-slash in the end): ")
    ) or "/")
    username = input("Enter username: ")
    email = input("Enter email: ")

    repo_host = REPO[repo]
    # Global Config
    if os.path.exists(os.path.join(HOME_DIR, PREFIX, ".gitconfig")):
        with open(os.path.join(HOME_DIR, PREFIX, ".gitconfig"), 'a', encoding='utf-8') as file:
            file.write(GLOBAL_CONFIG.format(projects_path=projects_path, client_name=client_name))
    else:
        with open(os.path.join(HOME_DIR, PREFIX, ".gitconfig"), 'w', encoding='utf-8') as file:
            file.write(GLOBAL_CONFIG.format(projects_path=projects_path, client_name=client_name))
        os.chmod(os.path.join(HOME_DIR, PREFIX, ".gitconfig"), 0o644)
    # Project Config
    if os.path.exists(os.path.join(HOME_DIR, PREFIX, ".gitconfigs")):
        with open(os.path.join(HOME_DIR, PREFIX, f".gitconfigs/.gitconfig-{client_name}"), 'w',
                  encoding='utf-8') as file:
            file.write(PROJECT_CONFIG.format(username=username, email=email))
        os.chmod(os.path.join(
            HOME_DIR, PREFIX, f".gitconfigs/.gitconfig-{client_name}"), 0o644)
    else:
        os.mkdir(os.path.join(HOME_DIR, PREFIX, ".gitconfigs"))
        with open(os.path.join(HOME_DIR, PREFIX, f".gitconfigs/.gitconfig-{client_name}"), 'w',
                  encoding='utf-8') as file:
            file.write(PROJECT_CONFIG.format(username=username, email=email))
        os.chmod(os.path.join(
            HOME_DIR, PREFIX, f".gitconfigs/.gitconfig-{client_name}"), 0o644)
    # SSH Config
    if os.path.exists(os.path.join(HOME_DIR, PREFIX, ".ssh/config")):
        with open(os.path.join(HOME_DIR, PREFIX, ".ssh/config"), 'a', encoding='utf-8') as file:
            file.write(SSH_CONFIG.format(client_name=client_name, repo_host=repo_host))
    else:
        with open(os.path.join(HOME_DIR, PREFIX, ".ssh/config"), 'w', encoding='utf-8') as file:
            file.write(SSH_CONFIG.format(client_name=client_name, repo_host=repo_host))

    # Generate RSA Keys
    key = RSA.generate(2048)
    with open(os.path.join(HOME_DIR, PREFIX, f".ssh/id_rsa_{client_name}"), 'wb') as content_file:
        os.chmod(os.path.join(
            HOME_DIR, PREFIX, f".ssh/id_rsa_{client_name}"), 0o600)
        content_file.write(key.exportKey('PEM'))
    pubkey = key.publickey()
    with open(os.path.join(HOME_DIR, PREFIX, f".ssh/id_rsa_{client_name}.pub"), 'wb') as content_file:
        content_file.write(pubkey.exportKey('OpenSSH'))

    os.system(f'ssh-add ~/.ssh/id_rsa_{client_name}')
    print('Id RSA public Key\n', pubkey.exportKey('OpenSSH'))


if __name__ == "__main__":
    main()