mirror of
https://github.com/nova-r/fediplug.git
synced 2025-03-25 22:54:59 +01:00
e14b1af877
* Add style indicatorsr * add type hinting, format with black Co-authored-by: Mæve Rey <42996147+m-rey@users.noreply.github.com>
105 lines
3 KiB
Python
105 lines
3 KiB
Python
"""Entry point for command-line interface."""
|
|
|
|
from typing import Dict, Tuple
|
|
|
|
options: Dict[str, bool] = {
|
|
"debug": False
|
|
} # needs to be initialized before imports, to avoid circular import error (yes, i know...)
|
|
|
|
import os
|
|
|
|
path = os.path
|
|
import sys
|
|
|
|
import click as click
|
|
import asyncio
|
|
|
|
from fediplug.dirs import DIRS
|
|
import fediplug.mastodon as mastodon
|
|
import fediplug.keyring as keyring
|
|
import fediplug.buttplugio as buttplugio
|
|
|
|
|
|
def get_access_token(instance: str) -> str:
|
|
"""Ensure the user credential exists."""
|
|
|
|
keyring.migrate_access_token(instance)
|
|
|
|
if not keyring.has_credential(instance, keyring.CREDENTIAL_ACCESS_TOKEN):
|
|
click.echo(
|
|
f"user credential for {instance} does not exist; try `fediplug login`"
|
|
)
|
|
sys.exit(1)
|
|
|
|
return keyring.get_credential(instance, keyring.CREDENTIAL_ACCESS_TOKEN)
|
|
|
|
|
|
def get_client_credentials(instance: str) -> Tuple[str, str]:
|
|
"""Ensure the client credentials exist."""
|
|
|
|
keyring.migrate_client_credentials(instance)
|
|
|
|
if not (
|
|
keyring.has_credential(instance, keyring.CREDENTIAL_CLIENT_ID)
|
|
and keyring.has_credential(instance, keyring.CREDENTIAL_CLIENT_SECRET)
|
|
):
|
|
click.echo(
|
|
f"client credentials for {instance} do not exist; try `fediplug register`"
|
|
)
|
|
sys.exit(1)
|
|
|
|
return (
|
|
keyring.get_credential(instance, keyring.CREDENTIAL_CLIENT_ID),
|
|
keyring.get_credential(instance, keyring.CREDENTIAL_CLIENT_SECRET),
|
|
)
|
|
|
|
|
|
@click.group()
|
|
@click.option("-d", "--debug", is_flag=True, help="Print debug messages.")
|
|
def cli(debug: bool) -> None:
|
|
"""A program to play music your friends post on Mastodon."""
|
|
|
|
options["debug"] = debug
|
|
|
|
|
|
@cli.command()
|
|
@click.argument("instance")
|
|
def register(instance: str) -> None:
|
|
"""Register fediplug on your Mastodon instance."""
|
|
|
|
mastodon.register(instance)
|
|
|
|
|
|
@cli.command()
|
|
@click.argument("instance")
|
|
def login(instance: str) -> None:
|
|
"""Log in to your Mastodon instance."""
|
|
|
|
client_id, client_secret = get_client_credentials(instance)
|
|
|
|
click.echo("Open this page in your browser and follow the instructions.")
|
|
click.echo("Paste the code here.")
|
|
click.echo("")
|
|
click.echo(mastodon.get_auth_request_url(instance, client_id, client_secret))
|
|
click.echo("")
|
|
|
|
grant_code = input("Code: ")
|
|
mastodon.login(instance, client_id, client_secret, grant_code)
|
|
|
|
|
|
@cli.command()
|
|
@click.argument("instance")
|
|
@click.argument("users", nargs=-1)
|
|
def stream(instance: str, users: Tuple[str]):
|
|
"""Control buttplug.io device from your timeline."""
|
|
|
|
event_loop = asyncio.get_event_loop()
|
|
plug_client = event_loop.run_until_complete(buttplugio.connect_plug_client())
|
|
plug_client = event_loop.run_until_complete(buttplugio.scan_devices(plug_client))
|
|
|
|
client_id, client_secret = get_client_credentials(instance)
|
|
access_token = get_access_token(instance)
|
|
|
|
mastodon.stream(
|
|
instance, users, client_id, client_secret, access_token, plug_client, event_loop
|
|
)
|