72 lines
2.6 KiB
Python
72 lines
2.6 KiB
Python
|
import tarfile
|
||
|
|
import tempfile
|
||
|
|
import os, sys
|
||
|
|
|
||
|
|
import requests
|
||
|
|
from requests.auth import HTTPBasicAuth
|
||
|
|
|
||
|
|
from rich.console import Console
|
||
|
|
from rich.prompt import Prompt
|
||
|
|
|
||
|
|
|
||
|
|
console = Console()
|
||
|
|
|
||
|
|
|
||
|
|
def publish(args):
|
||
|
|
if not "@" in args.name:
|
||
|
|
console.print(f"[b red]digpkg: failed to publish mineral: please include the version number in the package name. e.g: request@1.0.0")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
split_name = args.name.split("@")
|
||
|
|
mineral_name = split_name[0]
|
||
|
|
version = split_name[1]
|
||
|
|
|
||
|
|
# sanity checks
|
||
|
|
if not os.path.isdir(args.folder_path):
|
||
|
|
console.print(f"[b red]digpkg: failed to publish mineral: \"{args.folder_path}\" is not a directory")
|
||
|
|
sys.exit(1)
|
||
|
|
if not os.path.isfile(os.path.join(args.folder_path, "mineral.ini")):
|
||
|
|
console.print(f"[b red]digpkg: failed to publish mineral: mineral has no \"mineral.ini\" file")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
# ask for user and pass
|
||
|
|
console.print("[b]Please authenticate.\n[/]")
|
||
|
|
try:
|
||
|
|
username = Prompt.ask("Username", console=console)
|
||
|
|
password = Prompt.ask("Password (or PAT)", console=console, password=True)
|
||
|
|
except KeyboardInterrupt:
|
||
|
|
return
|
||
|
|
|
||
|
|
console.print()
|
||
|
|
|
||
|
|
with console.status("Compressing...", spinner="bouncingBall", spinner_style="blue") as status:
|
||
|
|
# compress to a tar file
|
||
|
|
with tempfile.TemporaryFile(mode="wb+") as f:
|
||
|
|
tar_file = tarfile.open(fileobj=f, mode="w:gz")
|
||
|
|
tar_file.add(args.folder_path, arcname=os.path.basename(args.folder_path))
|
||
|
|
|
||
|
|
console.print("[d][:white_check_mark:] Compressed![/]")
|
||
|
|
|
||
|
|
# send the request
|
||
|
|
status.update("Uploading...")
|
||
|
|
response = requests.put(
|
||
|
|
url=f"https://chookspace.com/api/packages/{username}/generic/{mineral_name}/{version}/mineral.tar",
|
||
|
|
data=f,
|
||
|
|
auth=HTTPBasicAuth(username, password)
|
||
|
|
)
|
||
|
|
|
||
|
|
tar_file.close()
|
||
|
|
|
||
|
|
if response.status_code == 401:
|
||
|
|
console.print("[b red]digpkg: failed to publish mineral: authentication failed[/]")
|
||
|
|
sys.exit(1)
|
||
|
|
elif response.status_code == 400:
|
||
|
|
console.print("[b red]digpkg: failed to publish mineral: the package name or version number are invalid[/]")
|
||
|
|
sys.exit(1)
|
||
|
|
elif response.status_code == 409:
|
||
|
|
console.print("[b red]digpkg: failed to publish mineral: that version number is already in use[/]")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
response.raise_for_status()
|
||
|
|
console.print("[d][:white_check_mark:] Uploaded![/]")
|
||
|
|
|