87 lines
1.9 KiB
Python
87 lines
1.9 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import os
|
||
|
|
|
||
|
|
import click
|
||
|
|
import requests
|
||
|
|
|
||
|
|
|
||
|
|
def get_username():
|
||
|
|
response = requests.post(
|
||
|
|
"https://meta.sr.ht/query",
|
||
|
|
json={"query": "{ me { canonicalName } }"},
|
||
|
|
headers={"Authorization": f"Bearer {os.getenv('SRHT')}"},
|
||
|
|
).json()
|
||
|
|
|
||
|
|
return response["data"]["me"]["canonicalName"]
|
||
|
|
|
||
|
|
|
||
|
|
@click.group()
|
||
|
|
def cli():
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
@cli.command(help="Prints out canonical name of the authenticated user")
|
||
|
|
def whoami():
|
||
|
|
click.echo(get_username())
|
||
|
|
|
||
|
|
|
||
|
|
@cli.command(help="Provides paste to the paste.sr.ht")
|
||
|
|
@click.option(
|
||
|
|
"--public",
|
||
|
|
"visibility",
|
||
|
|
flag_value="public",
|
||
|
|
help="Public paste that is visible on the user's profile",
|
||
|
|
)
|
||
|
|
@click.option(
|
||
|
|
"--unlisted",
|
||
|
|
"visibility",
|
||
|
|
flag_value="unlisted",
|
||
|
|
default=True,
|
||
|
|
help="Unlisted paste that is accessible via a link",
|
||
|
|
)
|
||
|
|
@click.option(
|
||
|
|
"--private",
|
||
|
|
"visibility",
|
||
|
|
flag_value="private",
|
||
|
|
help="Private paste that is accessible only to the authenticated user",
|
||
|
|
)
|
||
|
|
@click.argument(
|
||
|
|
"src",
|
||
|
|
type=click.File("r"),
|
||
|
|
nargs=-1,
|
||
|
|
)
|
||
|
|
def paste(visibility, src):
|
||
|
|
request = {
|
||
|
|
"visibility": visibility,
|
||
|
|
"files": [
|
||
|
|
{"filename": s.name.split("/")[-1], "contents": s.read()}
|
||
|
|
for s in src
|
||
|
|
],
|
||
|
|
}
|
||
|
|
|
||
|
|
response = requests.post(
|
||
|
|
"https://paste.sr.ht/api/pastes",
|
||
|
|
json=request,
|
||
|
|
headers={"Authorization": f"token {os.getenv('SRHT_LEGACY')}"},
|
||
|
|
).json()
|
||
|
|
|
||
|
|
if "sha" in response:
|
||
|
|
click.secho("*** Paste link ***", fg="green")
|
||
|
|
click.secho(
|
||
|
|
f"https://paste.sr.ht/{response['user']['canonical_name']}/{response['sha']}"
|
||
|
|
)
|
||
|
|
|
||
|
|
click.secho("\n*** Files ***", fg="green")
|
||
|
|
for file in response["files"]:
|
||
|
|
filename = file["filename"]
|
||
|
|
blob_id = file["blob_id"]
|
||
|
|
|
||
|
|
click.secho(f"https://paste.sr.ht/blob/{blob_id}\t{filename}")
|
||
|
|
else:
|
||
|
|
click.secho(response, fg="red")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
cli()
|