tuxbot-bot/tuxbot/setup.py

396 lines
11 KiB
Python

import argparse
import importlib
import logging
import os
import re
import sys
import json
from argparse import Namespace
from pathlib import Path
from typing import List
from urllib import request
from rich.prompt import Prompt, IntPrompt
from rich.console import Console
from rich.rule import Rule
from rich.style import Style
from rich.traceback import install
from tuxbot import version_info
from tuxbot.core.config import set_for
from tuxbot.logging import formatter
from tuxbot.core.utils.data_manager import (
config_path,
config_file,
cogs_data_path,
)
from tuxbot.core import config
console = Console()
install(console=console, show_locals=True)
try:
config_path.mkdir(parents=True, exist_ok=True)
except PermissionError:
console.print(
f"mkdir: cannot create directory '{config_path}': Permission denied"
)
sys.exit(1)
def get_name() -> str:
"""Get instance name via input.
Returns
-------
str
The instance name choose by user.
"""
name = ""
while not name:
name = Prompt.ask(
"What name do you want to give this instance?\n"
"[i](valid characters: A-Z, a-z, 0-9, _, -)[/i]\n",
default="prod",
console=console,
)
if re.fullmatch(r"[a-zA-Z0-9_\-]*", name) is None:
console.print()
console.print("[prompt.invalid]ERROR: Invalid characters provided")
name = ""
return name
def get_token() -> str:
"""Get token via input.
Returns
-------
str
The token choose by user.
"""
token = ""
while not token:
token = Prompt.ask(
"Please enter the bot token "
"(you can find it at https://discord.com/developers/applications)",
console=console,
)
if (
re.fullmatch(
r"([a-zA-Z0-9]{24}\.[a-zA-Z0-9_]{6}\.[a-zA-Z0-9_\-]{27}"
r"|mfa\.[a-zA-Z0-9_\-]{84})",
token,
)
is None
):
console.print("[prompt.invalid]ERROR: Invalid token provided")
token = ""
return token
def get_ip() -> str:
"""Get ip via input.
Returns
-------
str
The ip choose by user.
"""
ip = ""
# pylint: disable=line-too-long
ipv4_pattern = r"^(([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\.){3}([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])$"
ipv6_pattern = r"^(([0-9a-fA-F]{1,4}:){7,7}[0-9a-fA-F]{1,4}|([0-9a-fA-F]{1,4}:){1,7}:|([0-9a-fA-F]{1,4}:){1,6}:[0-9a-fA-F]{1,4}|([0-9a-fA-F]{1,4}:){1,5}(:[0-9a-fA-F]{1,4}){1,2}|([0-9a-fA-F]{1,4}:){1,4}(:[0-9a-fA-F]{1,4}){1,3}|([0-9a-fA-F]{1,4}:){1,3}(:[0-9a-fA-F]{1,4}){1,4}|([0-9a-fA-F]{1,4}:){1,2}(:[0-9a-fA-F]{1,4}){1,5}|[0-9a-fA-F]{1,4}:((:[0-9a-fA-F]{1,4}){1,6})|:((:[0-9a-fA-F]{1,4}){1,7}|:)|fe80:(:[0-9a-fA-F]{0,4}){0,4}%[0-9a-zA-Z]{1,}|::(ffff(:0{1,4}){0,1}:){0,1}((25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])\.){3,3}(25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])|([0-9a-fA-F]{1,4}:){1,4}:((25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])\.){3,3}(25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9]))"
while not ip:
ip = Prompt.ask(
"Please the ip of this machine "
"(you can find it with `curl ifconfig.me`)",
console=console,
)
ipv4 = re.match(ipv4_pattern, ip)
ipv6 = re.match(ipv6_pattern, ip)
if not ipv4 and not ipv6:
console.print("[prompt.invalid]ERROR: Invalid ip provided")
ip = ""
return ip
def get_multiple(
question: str, confirmation: str, value_type: type
) -> List[str | int]:
"""Give possibility to user to fill multiple value.
Parameters
----------
question:str
First question.
confirmation:str
Asking text if user want to add another.
value_type:type
The type of values inside the list.
Returns
-------
List[str | int]
List containing user filled values.
"""
prompt: IntPrompt | Prompt = IntPrompt() if value_type is int else Prompt()
user_input = prompt.ask(question, console=console)
if not user_input:
return []
values = [user_input]
while (
Prompt.ask(
confirmation, choices=["y", "n"], default="n", console=console
)
!= "n"
):
new = prompt.ask("Other")
if new not in values:
values.append(new)
else:
console.print(
f"[prompt.invalid]"
f"ERROR: `{new}` is already present, [i]ignored[/i]"
)
return values
def get_extra(question: str, value_type: type) -> str | int:
prompt: IntPrompt | Prompt = IntPrompt() if value_type is int else Prompt()
return prompt.ask(question, console=console)
def additional_config(cogs: str | list = "**"):
"""Asking for additional configs in cogs.
Returns
-------
dict:
Dict with cog name as key and configs as value.
"""
if cogs is None or "all" in sum(cogs, []):
cogs = []
else:
cogs = sum(cogs, [])
if len(cogs) == 0:
paths = list(Path("tuxbot/cogs").glob("**/config.py"))
else:
paths = [Path(f"tuxbot/cogs/{cog}/config.py") for cog in cogs]
for path in paths:
cog_name = str(path.parent).split("/")[-1]
if path.exists():
console.print(Rule(f"\nConfiguration for `{cog_name}` module"))
mod = importlib.import_module(str(path).replace("/", ".")[:-3])
mod_config_type = getattr(mod, cog_name.capitalize() + "Config")
mod_extra = mod.extra # type: ignore
mod_config = config.ConfigFile(
str(cogs_data_path(cog_name) / "config.yaml"),
mod_config_type,
).config
extras = {}
for key, value in mod_extra.items():
extras[key] = get_extra(value["description"], value["type"])
set_for(mod_config, **extras)
else:
console.print(
Rule(
f"\nFailed to fetch information for `{cog_name}` module",
style=Style(color="red"),
)
)
def finish_setup() -> None:
"""Configs who directly refer to the bot."""
name = get_name()
console.print(
Rule("Now, it's time to finish this setup by giving bot information")
)
console.print()
token = get_token()
ip = get_ip()
console.print()
prefixes = get_multiple(
"Choice a (or multiple) prefix for the bot",
"Add another prefix ?",
str,
)
console.print()
mentionable = (
Prompt.ask(
"Does the bot answer if it's mentioned?",
choices=["y", "n"],
default="y",
)
== "y"
)
console.print()
owners_id = get_multiple(
"Give the owner id of this bot", "Add another owner ?", int
)
console.print("\n" * 4)
console.print(Rule("\nAnd to finish, the configuration for PostgreSQL"))
console.print()
database = {
"username": Prompt.ask(
"Please enter the username for PostgreSQL",
console=console,
),
"password": Prompt.ask(
"Please enter the password for PostgreSQL",
console=console,
),
"domain": Prompt.ask(
"Please enter the domain for PostgreSQL",
console=console,
default="localhost",
),
"port": IntPrompt.ask(
"Please enter the port for PostgreSQL",
console=console,
default="5432",
),
"db_name": Prompt.ask(
"Please enter the database name for PostgreSQL", console=console
),
}
_config_file = config.ConfigFile(str(config_file), config.Config)
_config_file.config.Core.owners_id = owners_id
_config_file.config.Core.prefixes = prefixes
_config_file.config.Core.token = token
_config_file.config.Core.ip = ip
_config_file.config.Core.mentionable = mentionable
_config_file.config.Core.locale = "en-US"
_config_file.config.Core.instance_name = name
_config_file.config.Core.Database.username = database["username"]
_config_file.config.Core.Database.password = database["password"]
_config_file.config.Core.Database.domain = database["domain"]
_config_file.config.Core.Database.port = database["port"]
_config_file.config.Core.Database.db_name = database["db_name"]
def basic_setup() -> None:
"""Configs who refer to instances."""
console.print(
Rule(
"Hi ! it's time for you to give me information about you instance"
)
)
finish_setup()
console.print()
console.print(
"Instance successfully created! "
"You can now run `tuxbot` to launch it now or "
"setup the additional configs by running "
"`tuxbot-setup --additional-config=all`"
)
def update() -> None:
response = json.load(
request.urlopen(
"https://api.github.com/repos/Rom1-J/tuxbot-bot/commits/master"
) # skipcq: BAN-B310
)
if response.get("sha")[:6] == version_info.build:
print("Nothing to update, you can run `tuxbot` to start the bot")
else:
print(f"Updating to {response.get('sha')[:6]}...")
os.popen("/usr/bin/git pull")
print("Done!")
def parse_cli_flags(args: list) -> Namespace:
"""Parser for cli values.
Parameters
----------
args:list
Is a list of all passed values.
Returns
-------
Namespace
"""
parser = argparse.ArgumentParser(
description="Tuxbot Setup - OpenSource bot",
usage="tuxbot-setup [arguments]",
)
parser.add_argument(
"-a",
"--additional-config",
action="append",
nargs="+",
help="Execute setup to additional configs",
)
parser.add_argument(
"-U",
"--update",
action="store_true",
help="Check for update",
)
return parser.parse_args(args)
def setup() -> None:
cli_flags = parse_cli_flags(sys.argv[1:])
try:
if cli_flags.update:
update()
sys.exit()
# Create a new instance.
level = logging.DEBUG
base_logger = logging.getLogger("tuxbot")
base_logger.setLevel(level)
stdout_handler = logging.StreamHandler(sys.stdout)
stdout_handler.setFormatter(formatter)
base_logger.addHandler(stdout_handler)
if cli_flags.additional_config:
additional_config(cli_flags.additional_config)
else:
console.clear()
basic_setup()
except KeyboardInterrupt:
console.print("Exiting...")
except Exception:
console.print_exception()
if __name__ == "__main__":
setup()