import json
import logging
import subprocess

import click
from rich.logging import RichHandler

from restic.console import console, logger
from restic.loki import send_to_loki


def snapshot(loki_url: str = None):
    cmd = ['restic', 'snapshots', '--json']

    with console.status('Getting snapshots...'):
        logger.debug(f'Running cmd [bright_black]{" ".join(cmd)}[/]')
        result = subprocess.run(cmd, capture_output=True, text=True)

    line = result.stdout
    data = json.loads(line)
    logger.info(f'Got {len(data)} snapshots')

    if loki_url is not None:
        send_to_loki(loki_url=loki_url, line=line, backup='snapshots')


@click.command()
@click.option('--loki-url', type=str, help='Loki URL for logging', envvar='LOKI_URL')
def main(loki_url: str = None):
    logging.basicConfig(
        level='DEBUG', format='%(message)s', handlers=[RichHandler(markup=True, console=console)]
    )

    snapshot(loki_url)


if __name__ == '__main__':
    main()