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", multiple=True, 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()
|
