blob: ad29d21078f4352beb4bbe2e83cb87206bf4409a (
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
|
from argparse import Namespace
import re
from ._base import AppCommandFeatureProvider
from ._config import ConfigManager
from ._template import TemplateManager
class NginxManager(AppCommandFeatureProvider):
def __init__(self) -> None:
super().__init__("nginx-manager")
self._domains_cache: list[str] | None = None
self._domain_config_value_cache: str | None = None
def setup(self) -> None:
pass
@property
def domains(self) -> list[str]:
if self._domains_cache is None:
self._domains_cache = self._get_domains()
return self._domains_cache
@property
def _domain_config_name(self) -> str:
return self.app.get_feature(ConfigManager).get_domain_item_name()
def _get_domain_config_value(self) -> str:
if self._domain_config_value_cache is None:
self._domain_config_value_cache = self.app.get_feature(
ConfigManager
).get_item_value_str(self._domain_config_name)
return self._domain_config_value_cache
def _get_domains_from_text(self, text: str) -> set[str]:
domains: set[str] = set()
regex = re.compile(r"server_name\s+(\S+)\s*;")
domain_variable_str = f"${self._domain_config_name}"
brace_domain_variable_regex = re.compile(
r"\$\{\s*" + self._domain_config_name + r"\s*\}"
)
for match in regex.finditer(text):
domain_part = match.group(1)
if domain_variable_str in domain_part:
domains.add(
domain_part.replace(
domain_variable_str, self._get_domain_config_value()
)
)
continue
m = brace_domain_variable_regex.search(domain_part)
if m:
domains.add(
domain_part.replace(m.group(0), self._get_domain_config_value())
)
continue
domains.add(domain_part)
return domains
def _get_nginx_conf_template_text(self) -> str:
template_manager = self.app.get_feature(TemplateManager)
text = ""
for path, template in template_manager.template_tree.templates:
if path.as_posix().startswith("nginx/"):
text += template.raw_text
return text
def _get_domains(self) -> list[str]:
text = self._get_nginx_conf_template_text()
domains = list(self._get_domains_from_text(text))
domains.remove(self._get_domain_config_value())
return [self._get_domain_config_value(), *domains]
def _print_domains(self) -> None:
for domain in self.domains:
print(domain)
def get_command_info(self):
return "nginx", "Manage nginx related things."
def setup_arg_parser(self, arg_parser):
subparsers = arg_parser.add_subparsers(
dest="nginx_command", required=True, metavar="NGINX_COMMAND"
)
_list_parser = subparsers.add_parser("list", help="list domains")
def run_command(self, args: Namespace) -> None:
if args.nginx_command == "list":
self._print_domains()
|