twtxt-registry-client/twtxt_registry_client/__main__.py

#!/usr/bin/env python3
from urllib.parse import urlsplit, urlunsplit
from objtools.collections import Namespace
from requests.exceptions import HTTPError
from twtxt.config import Config
from twtxt_registry_client import RegistryClient, output
import click


@click.group(name='twtxt-registry')
@click.argument('registry_url', required=True)
@click.version_option('-V', '--version')
@click.option(
    '-k', '--insecure',
    is_flag=True,
    help='Disable SSL certificate checks.',
)
@click.option(
    '-f', '--format',
    type=click.Choice(output.registry.keys()),
    default='pretty',
    help='Change the output format.',
)
@click.pass_context
def cli(ctx, registry_url, insecure, format):
    """
    Command-line client for the twtxt registry API.

    Takes a mandatory registry URL argument, as the base API URL
    (ex. https://registry.twtxt.org/api)
    """
    ctx.obj = Namespace()

    try:
        ctx.obj.conf = Config.discover()
    except ValueError:
        ctx.obj.conf = Namespace()

    scheme, netloc, path, query, fragment = urlsplit(registry_url)
    if not scheme:
        scheme = 'https'
    if not netloc and path:
        netloc, _, path = path.partition('/')
    registry_url = urlunsplit((scheme, netloc, path, query, fragment))
    ctx.obj.client = RegistryClient(registry_url, insecure=insecure)

    ctx.obj.formatter = output.registry[format]()


@cli.command()
@click.option(
    '-n', '--nickname',
    help='Nickname to register with. '
         'Defaults to the configured twtxt nickname, if available.',
    metavar='[NICK]',
)
@click.option(
    '-u', '--url',
    help='URL to the twtxt file to register with. '
         'Defaults to the configured twtxt URL, if available.',
    metavar='[URL]',
)
@click.pass_context
def register(ctx, nickname, url):
    """
    Register a user on a registry.
    """
    if not nickname or not url:
        if not ctx.obj.conf:
            raise click.UsageError(
                'Nickname or URL were omitted from the command-line, but they'
                'could not be deduced from the twtxt config.',
                ctx=ctx,
            )
        nickname = nickname or ctx.obj.conf.nick
        url = url or ctx.obj.conf.twturl

    click.echo(ctx.obj.formatter.format_response(
        ctx.obj.client.register(nickname, url, raise_exc=False)
    ))


@cli.command()
@click.option(
    '-q', '--query',
    help='An optional search query to filter users.',
)
@click.pass_context
def users(ctx, query):
    """
    List and search users on a registry.
    """
    try:
        click.echo(ctx.obj.formatter.format_users(
            ctx.obj.client.list_users(q=query)
        ))
    except HTTPError as e:
        click.echo(ctx.obj.formatter.format_response(e.response))


@cli.command()
@click.option(
    '-q', '--query',
    help='An optional search query to filter tweets.',
)
@click.pass_context
def tweets(ctx, query):
    """
    List and search tweets on a registry.
    """
    try:
        click.echo(ctx.obj.formatter.format_tweets(
            ctx.obj.client.list_tweets(q=query)
        ))
    except HTTPError as e:
        click.echo(ctx.obj.formatter.format_response(e.response))


@cli.command()
@click.argument('name_or_url', required=False)
@click.pass_context
def mentions(ctx, name_or_url):
    """
    List mentions to someone on a registry.

    Without arguments, will try to use twtxt's configured URL to list your own
    mentions.
    """
    if name_or_url:
        scheme = urlsplit(name_or_url).scheme
        if ctx.obj.conf and not scheme:  # it could be a nick
            source = ctx.obj.conf.get_source_by_nick(name_or_url)
            if source:
                url = source.url
        url = url or name_or_url  # Fallback
    elif not ctx.obj.conf:
        raise click.UsageError(
            'URL was omitted from the command-line, but it could not '
            'be deduced from the twtxt config.',
            ctx=ctx,
        )
    else:
        url = ctx.obj.conf.twturl

    try:
        click.echo(ctx.obj.formatter.format_tweets(
            ctx.obj.client.list_mentions(url)
        ))
    except HTTPError as e:
        click.echo(ctx.obj.formatter.format_response(e.response))


@cli.command()
@click.argument('name', required=True)
@click.pass_context
def tag(ctx, name):
    """
    Search for tweets containing a tag.

    Requires a tag name as a positional argument.
    """
    try:
        click.echo(ctx.obj.formatter.format_tweets(
            ctx.obj.client.list_tag_tweets(name)
        ))
    except HTTPError as e:
        click.echo(ctx.obj.formatter.format_response(e.response))


if __name__ == '__main__':
    cli()