recticode 0.1.0__tar.gz

recticode-0.1.0/PKG-INFO

@@ -0,0 +1,15 @@
+Metadata-Version: 2.3
+Name: recticode
+Version: 0.1.0
+Summary: Add your description here
+Author: VulcanWM
+Author-email: VulcanWM <vulcanwmemail@gmail.com>
+Requires-Dist: pytest>=9.0.2
+Requires-Dist: requests>=2.32.5
+Requires-Dist: typer>=0.24.1
+Requires-Python: >=3.13
+Description-Content-Type: text/markdown
+
+# Recticode
+
+It's cool

recticode-0.1.0/README.md

@@ -0,0 +1,3 @@
+# Recticode
+
+It's cool

recticode-0.1.0/pyproject.toml

@@ -0,0 +1,21 @@
+[project]
+name = "recticode"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+authors = [
+    { name = "VulcanWM", email = "vulcanwmemail@gmail.com" }
+]
+requires-python = ">=3.13"
+dependencies = [
+    "pytest>=9.0.2",
+    "requests>=2.32.5",
+    "typer>=0.24.1",
+]
+
+[project.scripts]
+recticode = "recticode.main:app"
+
+[build-system]
+requires = ["uv_build>=0.10.11,<0.11.0"]
+build-backend = "uv_build"

recticode-0.1.0/src/recticode/__init__.py

@@ -0,0 +1,2 @@
+def main() -> None:
+    print("Hello from recticode!")

recticode-0.1.0/src/recticode/git_code.py

@@ -0,0 +1,91 @@
+import subprocess
+import os
+import json
+import requests
+
+def get_user_code():
+    url = "https://github.com/login/device/code"
+
+    values = {
+        "client_id": "Ov23lizX5wFSpnR89gKJ",
+        "scope": "read:user repo"
+    }
+
+    headers = {
+        "Accept": "application/json"
+    }
+
+    r = requests.post(url, data=values, headers=headers)
+
+    r_json = r.json()
+
+    device_code = r_json["device_code"]
+    verification_uri = r_json['verification_uri']
+    expires_in = r_json['expires_in']
+    user_code = r_json['user_code']
+    interval = r_json['interval']
+
+    return device_code, verification_uri, expires_in, user_code, interval
+
+def get_access_token():
+    try:
+        config_dir = os.path.expanduser("~/.config/recticode")
+        os.makedirs(config_dir, exist_ok=True)
+
+        token_path = os.path.join(config_dir, "token.json")
+        with open(token_path) as f:
+            token = json.load(f)["access_token"]
+        return token
+    except:
+        return False
+
+
+def save_access_token(access_token):
+    try:
+        config_dir = os.path.expanduser("~/.config/recticode")
+        os.makedirs(config_dir, exist_ok=True)
+
+        token_path = os.path.join(config_dir, "token.json")
+
+        data = {"access_token": access_token}
+
+        with open(token_path, "w") as f:
+            json.dump(data, f)
+
+        return True
+    except:
+        return False
+
+
+def clone_repo(url):
+    cmd = ["git", "clone", url]
+    subprocess.run(cmd, check=True)
+    return True
+
+def get_user_data(access_token):
+    check_headers = {
+        "Authorization": f"Bearer {access_token}",
+        "Accept": "application/vnd.github+json"
+    }
+
+    check_post = requests.get("https://api.github.com/user", headers=check_headers)
+
+    user_json = check_post.json()
+    username = user_json['login']
+    email = user_json['email']
+    name = user_json['name']
+
+    return username, email, name
+
+def remove_access_token():
+    try:
+        config_dir = os.path.expanduser("~/.config/recticode")
+        os.makedirs(config_dir, exist_ok=True)
+
+        token_path = os.path.join(config_dir, "token.json")
+
+        os.remove(token_path)
+
+        return True
+    except:
+        return False

recticode-0.1.0/src/recticode/main.py

@@ -0,0 +1,136 @@
+import typer
+from rich import print
+from functools import wraps
+from time import sleep
+from recticode.git_code import clone_repo, save_access_token, get_access_token, get_user_data, remove_access_token, get_user_code
+import requests
+import os.path
+import subprocess
+import sys
+
+app = typer.Typer()
+
+def require_login(func):
+    @wraps(func)
+    def wrapper(*args, **kwargs):
+        access_token = get_access_token()
+        if not access_token:
+            print("[red]You must login first[/red]")
+            raise typer.Exit()
+        return func(*args, **kwargs)
+    return wrapper
+
+@app.command()
+def login():
+    access_token = get_access_token()
+    if not access_token:
+        headers = {
+            "Accept": "application/json"
+        }
+
+        device_code, verification_uri, expires_in, user_code, interval = get_user_code()
+
+        print(f"Go to {verification_uri} and enter code [bold]{user_code}[/bold]")
+
+        for i in range(expires_in // interval):
+            poll_values = {
+                "client_id": "Ov23lizX5wFSpnR89gKJ",
+                "device_code": device_code,
+                "grant_type": "urn:ietf:params:oauth:grant-type:device_code"
+            }
+            poll_post = requests.post("https://github.com/login/oauth/access_token", data=poll_values, headers=headers)
+
+            poll_json = poll_post.json()
+
+            if "access_token" in poll_json:
+                access_token = poll_json['access_token']
+
+                username, email, name = get_user_data(access_token=access_token)
+
+                save_access_token(access_token=access_token)
+                print(f"[green]✓ Logged in as {name}[/green]")
+                break
+
+            if "error" in poll_json:
+                if poll_json["error"] == "authorization_pending":
+                    sleep(interval)
+                    continue
+                elif poll_json["error"] == "slow_down":
+                    interval += 5
+                    sleep(interval)
+                elif poll_json["error"] == "expired_token":
+                    print("[red]Device code expired, try login again[/red]")
+                    break
+    else:
+        username, email, name = get_user_data(access_token=access_token)
+
+        print("You are already logged in")
+
+@app.command()
+@require_login
+def whoami():
+    access_token = get_access_token()
+    username, email, name = get_user_data(access_token=access_token)
+    print("Hi", name)
+
+@app.command()
+def logout():
+    access_token = get_access_token()
+    if not access_token:
+        print("You are not logged in")
+    else:
+        remove_access_token()
+
+        print("[yellow]✓ Logged out[/yellow]")
+
+
+@app.command()
+@require_login
+def start(challenge_name):
+    access_token = get_access_token()
+
+    request_url = "https://api.recticode.com/challenge_repo/"
+
+    response = requests.get(request_url + challenge_name + "?token=" + access_token)
+    if response.status_code == 200:
+        if "error" in response.json():
+            print(response.json()['error'])
+        else:
+            clone_repo(response.json()['repo_name'])
+            print(f"[green]Challenge cloned![/green] cd {challenge_name} to start")
+    else:
+        print("Error occurred")
+
+@app.command()
+@require_login
+def list_challenges():
+    access_token = get_access_token()
+
+    request_url = "https://api.recticode.com/list_challenges?token=" + access_token
+
+    response = requests.get(request_url)
+    if response.status_code == 200:
+        challenges = response.json()['challenges']
+
+        print("[bold][yellow]All Challenges[/yellow][/bold]")
+        for challenge in challenges:
+            print(f"[bold]{challenge['name']}[/bold]: {challenge['description']} ({challenge['language']})")
+    else:
+        print("Error occurred")
+
+@app.command()
+@require_login
+def check():
+    if os.path.exists("challenge.json"):
+        env = os.environ.copy()
+        env["PYTHONPATH"] = "."
+        subprocess.run(
+            [sys.executable, "-m", "pytest", "tests/"],
+            env=env
+        )
+    else:
+        print("This is not a valid challenge")
+
+
+if __name__ == "__main__":
+    app()