mirror of
https://git.pleroma.social/pleroma/relay.git
synced 2025-04-19 17:16:42 +00:00
e85215d986
* add `typing-extensions` dev dependency * make sure `Self` import falls back to `typing-extensions` * let setuptools find sub-packages * pass global `Application` instance to cli commands * move `RELAY_SOFTWARE` to relay/misc.py * allow `str` for `follow` in `Message.new_unfollow`
73 lines
1.9 KiB
Python
73 lines
1.9 KiB
Python
import click
|
|
|
|
from . import cli, pass_app
|
|
|
|
from ..application import Application
|
|
from ..database.schema import Whitelist
|
|
|
|
|
|
@cli.group("whitelist")
|
|
def cli_whitelist() -> None:
|
|
"Manage the instance whitelist"
|
|
|
|
|
|
@cli_whitelist.command("list")
|
|
@click.pass_context
|
|
@pass_app
|
|
def cli_whitelist_list(app: Application, ctx: click.Context) -> None:
|
|
"List all the instances in the whitelist"
|
|
|
|
click.echo("Current whitelisted domains:")
|
|
|
|
with app.database.session() as conn:
|
|
for row in conn.execute("SELECT * FROM whitelist").all(Whitelist):
|
|
click.echo(f"- {row.domain}")
|
|
|
|
|
|
@cli_whitelist.command("add")
|
|
@click.argument("domain")
|
|
@pass_app
|
|
def cli_whitelist_add(app: Application, domain: str) -> None:
|
|
"Add a domain to the whitelist"
|
|
|
|
with app.database.session() as conn:
|
|
if conn.get_domain_whitelist(domain):
|
|
click.echo(f"Instance already in the whitelist: {domain}")
|
|
return
|
|
|
|
conn.put_domain_whitelist(domain)
|
|
click.echo(f"Instance added to the whitelist: {domain}")
|
|
|
|
|
|
@cli_whitelist.command("remove")
|
|
@click.argument("domain")
|
|
@pass_app
|
|
def cli_whitelist_remove(app: Application, domain: str) -> None:
|
|
"Remove an instance from the whitelist"
|
|
|
|
with app.database.session() as conn:
|
|
if not conn.del_domain_whitelist(domain):
|
|
click.echo(f"Domain not in the whitelist: {domain}")
|
|
return
|
|
|
|
if conn.get_config("whitelist-enabled"):
|
|
if conn.del_inbox(domain):
|
|
click.echo(f"Removed inbox for domain: {domain}")
|
|
|
|
click.echo(f"Removed domain from the whitelist: {domain}")
|
|
|
|
|
|
@cli_whitelist.command("import")
|
|
@pass_app
|
|
def cli_whitelist_import(app: Application) -> None:
|
|
"Add all current instances to the whitelist"
|
|
|
|
with app.database.session() as conn:
|
|
for row in conn.get_inboxes():
|
|
if conn.get_domain_whitelist(row.domain) is not None:
|
|
click.echo(f"Domain already in whitelist: {row.domain}")
|
|
continue
|
|
|
|
conn.put_domain_whitelist(row.domain)
|
|
|
|
click.echo("Imported whitelist from inboxes")
|