aboutsummaryrefslogtreecommitdiff
path: root/labbot/__main__.py
blob: 551f0f145b638c0198d37ce141b9500c659e28a9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
import click
import logging
from typing import List
from importlib import import_module
import json

import labbot.bot
import labbot.config
import labbot.logger

DEFAULT_ADDONS = [
    "merge-label",
    "approve-merge"
]

@click.group()
def main():
    pass

@main.command(help="Create a new instance")
@click.option("--name", prompt=True, help="Name the instance will be given")
@click.option("--access_token", prompt="Access Token for the Bot account", hide_input=True, help="Access Token to interact with the API")
@click.option("--secret", prompt="Webhook Secret (optional)", default="", help="Secret to receive webhook requests (optional)")
def setup(**data):
    instance_name = data.pop("name", "").replace(" ", "_").lower()
    data["addons"] = DEFAULT_ADDONS
    data["addon_path"] = ""

    if not labbot.config.read_instance_config(instance_name):
        labbot.config.write_instance_config(instance_name, data)
        click.echo(f"You can start your instance by running `lab-bot run {instance_name}`")
    else:
        click.echo(f"an instance with the name {instance_name} already exists")

@main.command(help="Configure an existing instance")
@click.argument('name')
@click.option("--access_token", required=False, help="Access Token to interact with the API")
@click.option("--secret", required=False, help="Secret to receive webhook requests (can be empty)")
@click.option("--addons", required=False, help="List of addons to load")
@click.option("--addon_path", required=False, help="Path to load custom addons from")
@click.option("--print", is_flag=True, required=False, help="Print the current config with redacted values")
def config(name, **data):

    data = {k:v for k,v in data.items() if v}
    print_config = data.pop("print", False)

    conf = labbot.config.read_instance_config(name)
    if conf:
        if data:
            conf.update(data)
            labbot.config.write_instance_config(name, conf)
            click.echo("configured")
        elif not print_config:
            click.echo("run with `--help` to show usage")

        if print_config:
            conf["access_token"] = "************"
            conf["secret"] = "******"
            click.echo(json.dumps(conf, indent=4))
    else:
        click.echo(f"{name} is not an instance")
    pass


@main.command(help="Run an instance")
@click.option("--port", default=8080, show_default=True, help="change the webhook port")
@click.option("--debug", is_flag=True, default=False, help="enable debug logging")
@click.argument('name')
def run(name, port: str, debug: bool):
    conf = labbot.config.read_instance_config(name)

    if not conf:
        click.echo(f"{name} is not an instance")
        return

    if debug:
        logger_level = logging.DEBUG
    else:
        logger_level = logging.INFO

    labbot.logger.init(logger_level)

    instance = labbot.bot.Bot(
            name=name,
            config=conf,
            secret=conf["secret"],
            access_token=conf["access_token"]
        )

    instance.run(
        port=port
    )

@main.command(name="list", help="List all available instances")
def list_instances():
    print("Available Instances:")
    for ins in labbot.config.list_instances():
        print(f"- {ins}")

if __name__ == "__main__":
    main()