mycode/mycode.py at main · marsdevx/mycode · 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
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
#    File:         mycode.py                                                    ⣿⣿⣿⣿⣿⣿⣿⣿⣘⠿⠿⠿⢆⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⡸⣿⣿⣿⣿⣿⡇⣿⣿⣿⣿⣿⣿    #
#                                                                               ⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣶⣮⣭⣭⣴⣿⣿⣿⣿⣿⣿    #
#    Project:      mycode                                                       ⣿⣿⣿⣿⠿⠿⠿⠿⠿⠏⠻⡿⠿⠿⢿⣿⣿⣿⣿⣿⣿⣿⣿⣿⠿⠿⠿⠿⠟⠹⠿⠿⠿⠿⠿⣿⣿⣿⣿⣿    #
#    Github:       marsdevx                                                     ⣿⣿⣿⣇⠀⢀⣀⣀⢀⠀⠀⠀⠀⠀⠈⣿⣿⣿⣿⣿⣿⣿⣿⠅⢀⠀⠀⠀⠀⠀⠀⢀⣀⣀⠀⠈⣿⣿⣿⣿    #
#                                                                               ⣿⣿⣿⣿⣷⡘⣿⣿⡇⢀⣤⣤⣤⣀⢸⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⡇⠀⣀⣀⣐⡀⡀⣿⣿⣿⡏⣼⣿⣿⣿    #
#    Created:      15:51   21/05/2025                                           ⣿⣿⣿⣿⣿⣿⣾⣿⣻⣝⣛⣛⣛⣫⣼⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣛⠿⠿⠿⢟⣴⢿⣿⣴⣿⣿⣿⣿⣿    #
#    Updated:      19:07   21/05/2025                                           ⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⢻⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿    #
#                                                                               ⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿    #
#    Path:         ./mycode.py                                                  ⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿    #
#                                                                               ⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⡛⠿⠟⣋⣍⣛⣋⣽⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿    #

import os
import time
import json
import parse
import argparse
import subprocess
import argcomplete
import create_proj as cp
from argcomplete.completers import ChoicesCompleter

def check_json():
  if not os.path.exists(cp.github_path):
    data = {
      "github": {
        "username": "",
        "token": ""
      }
    }

    with open(cp.github_path, "w") as f:
      json.dump(data, f, indent=2)

def create(project_name, target_dir):
  target_dir = os.path.expanduser(target_dir)
  if not os.path.isdir(target_dir):
      print(f"Target directory does not exist: {target_dir}")
      return

  with open(cp.github_path) as f:
    config_file = json.load(f)
  username = config_file["github"].get("username", "").strip()
  token = config_file["github"].get("token", "").strip()

  cp.create_project(project_name, target_dir, username, token)

def open_project(project_name, not_close):
  all_projects = parse.collect_projects(parse.config["projects_dir"], parse.config["specific_projects"], parse.config["remove_rules"])
  if project_name in all_projects:
    project_path = all_projects[project_name]
    print("\033[0;96mEnjoy your work, sir.\033[0m")
    time.sleep(1.3)
    print(f"Opening project '{project_name}'...")
    time.sleep(0.3)
    subprocess.run(["code", project_path])
    if not not_close:
      print("Closing terminal...")
      time.sleep(0.1)
      script = '''
      tell application "iTerm2"
          tell the current window
              close
          end tell
      end tell
      '''
      subprocess.run(["osascript", "-e", script])
  else:
    print(f"Project '{project_name}' not found.")

desc = ("✨ Manage and organize your projects with mycode ✨\n\n"
"📂 Requires a configuration file located at: `~/.config/mycode/mycoderc`\n\n"
"💾 Configuration File Rules:\n"
"  - `--add <path>`: Adds all subfolders in the specified path to global projects.\n"
"  - `--addspecific <path>`: Adds only the specified folder to global projects.\n"
"  - `--remove <path> --from <path>`: Excludes a specific subfolder from a parent path.\n\n"
"🔗 Grouping Projects:\n"
"Groups can be defined using `[group:group_name]` syntax in the configuration file.\n"
"  - Group-specific rules:\n"
"      - `--add <path>`: Adds subfolders to the group only.\n"
"      - `--addspecific <path>`: Adds only the specified folder to the group.\n"
"      - `--remove <path> --from <path>`: Excludes a specific subfolder in the group.\n"
"  - Example:\n"
"      [group:example_group]\n"
"      {\n"
"          --add ~/example/path\n"
"          --addspecific ~/example/specific_project\n"
"          --remove ~/example/path/excluded_folder --from ~/example/path\n"
"      }\n\n"
"🛠  Commands:\n"
"  - `mycode -s all`: Show all global projects.\n"
"  - `mycode -s groups`: Show all groups and their projects.\n"
"  - `mycode -s <group_name>`: Show all projects in the specified group.\n"
"  - `mycode <project_name>`: Open the specified project in VS Code.\n"
"  - `mycode <project_name> -n`: Open the project without closing the terminal.\n"
"  - `mycode -c <project_name> <target_dir>`: Create a new project and repository.\n\n"
"🎯 Shell Autocompletion:\n"
"  - Autocompletion is supported for Zsh and Bash.\n"
"  - Make sure to add the following to your shell config (e.g. ~/.zshrc or ~/.bashrc):\n\n"
"      autoload -Uz compinit bashcompinit\n"
"      compinit\n"
"      bashcompinit\n\n"
"      eval \"$(register-python-argcomplete mycode)\"\n\n"
"      _mycode() {\n"
"        if (( CURRENT > 2 )) &&\n"
"           [[ ${words[CURRENT-2]} == --create || ${words[CURRENT-2]} == -c ]]; then\n"
"          _files\n"
"          return\n"
"        fi\n\n"
"        if (( CURRENT > 1 )) &&\n"
"           [[ ${words[CURRENT-1]} == --create || ${words[CURRENT-1]} == -c ]]; then\n"
"          return\n"
"        fi\n\n"
"        _python_argcomplete \"$@\"\n"
"      }\n\n"
"      compdef _mycode mycode\n\n"
"  - Then run:\n"
"      source ~/.zshrc\n"
"      or\n"
"      source ~/.bashrc\n")

def main():
  check_json()
  prog = argparse.ArgumentParser(prog="mycode", add_help=False)

  prog.add_argument("-s", "--show", nargs=1, help="Show the data list").completer = ChoicesCompleter(parse.get_group())
  prog.add_argument("-c", "--create", nargs=2, metavar=("project_name", "target_dir"), help="Create a new project and repository")
  prog.add_argument("project", nargs="?", help="Name of project to open").completer = ChoicesCompleter(parse.get_proj())
  prog.add_argument("-n", "--not-close", action="store_true", help="Don't close terminal after opening project")
  prog.add_argument("-h", "--help", action="store_true", help=argparse.SUPPRESS)

  argcomplete.autocomplete(prog, always_complete_options=False)
  args = prog.parse_args()

  if args.create:
    project_name, target_dir = args.create
    create(project_name, target_dir)
    all_projects = parse.collect_projects(parse.config["projects_dir"], parse.config["specific_projects"], parse.config["remove_rules"])
    if project_name in all_projects:
        open_project(project_name, False)
    if project_name not in all_projects:
      print("\033[0;96mEnjoy your work, sir.\033[0m")
  elif args.show:
    show_arg = args.show[0]
    if show_arg == "all":
      parse.print_proj()
      print("\033[0;96mHere are your projects, sir\033[0m")
    elif show_arg == "groups":
      parse.print_groups()
      print("\033[0;96mHere are your projects, sir\033[0m")
    elif show_arg in parse.config["groups"]:
      parse.print_group(show_arg)
      print("\033[0;96mHere are your projects, sir\033[0m")
    else:
      print(f"Group '{show_arg}' not found.")
  elif args.project:
    open_project(args.project, args.not_close)
  if args.help or (not args.create and not args.show and not args.project):
    print(desc)
    return

if __name__ == "__main__":
  main()