118 lines
3.5 KiB
Python
118 lines
3.5 KiB
Python
import click
|
|
import json as pyjson
|
|
|
|
from itertools import chain
|
|
from typing import Optional
|
|
|
|
from toot import api
|
|
from toot.cli.validators import validate_instance
|
|
from toot.entities import Instance, Status, from_dict, Account
|
|
from toot.exceptions import ApiError, ConsoleError
|
|
from toot.output import print_account, print_instance, print_search_results, print_status, print_timeline
|
|
from toot.cli import cli, get_context, json_option, pass_context, Context
|
|
|
|
|
|
@cli.command()
|
|
@json_option
|
|
@pass_context
|
|
def whoami(ctx: Context, json: bool):
|
|
"""Display logged in user details"""
|
|
response = api.verify_credentials(ctx.app, ctx.user)
|
|
|
|
if json:
|
|
click.echo(response.text)
|
|
else:
|
|
account = from_dict(Account, response.json())
|
|
print_account(account)
|
|
|
|
|
|
@cli.command()
|
|
@click.argument("account")
|
|
@json_option
|
|
@pass_context
|
|
def whois(ctx: Context, account: str, json: bool):
|
|
"""Display account details"""
|
|
account_dict = api.find_account(ctx.app, ctx.user, account)
|
|
|
|
# Here it's not possible to avoid parsing json since it's needed to find the account.
|
|
if json:
|
|
click.echo(pyjson.dumps(account_dict))
|
|
else:
|
|
account_obj = from_dict(Account, account_dict)
|
|
print_account(account_obj)
|
|
|
|
|
|
@cli.command()
|
|
@click.argument("instance", callback=validate_instance, required=False)
|
|
@json_option
|
|
def instance(instance: Optional[str], json: bool):
|
|
"""Display instance details
|
|
|
|
INSTANCE can be a domain or base URL of the instance to display.
|
|
e.g. 'mastodon.social' or 'https://mastodon.social'. If not
|
|
given will display details for the currently logged in instance.
|
|
"""
|
|
if not instance:
|
|
context = get_context()
|
|
if not context.app:
|
|
raise click.ClickException("INSTANCE argument not given and not logged in")
|
|
instance = context.app.base_url
|
|
|
|
try:
|
|
response = api.get_instance(instance)
|
|
except ApiError:
|
|
raise ConsoleError(
|
|
f"Instance not found at {instance}.\n" +
|
|
"The given domain probably does not host a Mastodon instance."
|
|
)
|
|
|
|
if json:
|
|
click.echo(response.text)
|
|
else:
|
|
print_instance(from_dict(Instance, response.json()))
|
|
|
|
|
|
@cli.command()
|
|
@click.argument("query")
|
|
@click.option("-r", "--resolve", is_flag=True, help="Resolve non-local accounts")
|
|
@json_option
|
|
@pass_context
|
|
def search(ctx: Context, query: str, resolve: bool, json: bool):
|
|
"""Search for users or hashtags"""
|
|
response = api.search(ctx.app, ctx.user, query, resolve)
|
|
if json:
|
|
click.echo(response.text)
|
|
else:
|
|
print_search_results(response.json())
|
|
|
|
|
|
@cli.command()
|
|
@click.argument("status_id")
|
|
@json_option
|
|
@pass_context
|
|
def status(ctx: Context, status_id: str, json: bool):
|
|
"""Show a single status"""
|
|
response = api.fetch_status(ctx.app, ctx.user, status_id)
|
|
if json:
|
|
click.echo(response.text)
|
|
else:
|
|
status = from_dict(Status, response.json())
|
|
print_status(status)
|
|
|
|
|
|
@cli.command()
|
|
@click.argument("status_id")
|
|
@json_option
|
|
@pass_context
|
|
def thread(ctx: Context, status_id: str, json: bool):
|
|
"""Show thread for a toot."""
|
|
context_response = api.context(ctx.app, ctx.user, status_id)
|
|
if json:
|
|
click.echo(context_response.text)
|
|
else:
|
|
toot = api.fetch_status(ctx.app, ctx.user, status_id).json()
|
|
context = context_response.json()
|
|
|
|
statuses = chain(context["ancestors"], [toot], context["descendants"])
|
|
print_timeline(from_dict(Status, s) for s in statuses)
|