162 Zeilen
4.2 KiB
Python
162 Zeilen
4.2 KiB
Python
#!/usr/bin/python3
|
|
import pathlib
|
|
from typing import List, Union
|
|
|
|
try:
|
|
from ansible.module_utils.generic import SYSTEMD_NETWORK_CONFIG as SYSTEMD_PATH
|
|
from ansible.module_utils.generic import Types
|
|
from ansible.module_utils.module import SystemdUnitModule
|
|
except ImportError:
|
|
from module_utils.generic import SYSTEMD_NETWORK_CONFIG as SYSTEMD_PATH
|
|
from module_utils.generic import Types
|
|
from module_utils.module import SystemdUnitModule
|
|
|
|
def boolconvert(b: Union[bool, str]) -> str:
|
|
if b is True:
|
|
return "yes"
|
|
elif b is False:
|
|
return "no"
|
|
return b
|
|
|
|
|
|
class Module(SystemdUnitModule):
|
|
"""Sets up the systemd network unit"""
|
|
|
|
name = "systemd_network"
|
|
module_spec = dict(
|
|
argument_spec=dict(
|
|
mac=Types.str(),
|
|
device=Types.str(),
|
|
name=Types.str(required=True),
|
|
description=Types.str(),
|
|
dot=Types.bool(),
|
|
dnssec=Types.bool(),
|
|
dns=Types.list(elements=str),
|
|
domain=Types.list(elements=str),
|
|
defaultdns=Types.bool(),
|
|
address=Types.list(elements=str, required=True),
|
|
route=Types.list(elements=str),
|
|
),
|
|
required_if=(("defaultdns", True, ("dns",), False),),
|
|
required_one_of=(("mac", "device"),),
|
|
)
|
|
|
|
def prepare(self):
|
|
self.unitfile = SYSTEMD_PATH.joinpath(self.get("name")).with_suffix(".network")
|
|
self.__unit = None
|
|
|
|
def unit(self) -> str:
|
|
if self.__unit is None:
|
|
self.__unit = "\n".join(
|
|
(
|
|
self.match(),
|
|
self.network(),
|
|
self.addresses(),
|
|
self.routes(),
|
|
)
|
|
)
|
|
return self.__unit
|
|
|
|
def match(self) -> str:
|
|
matches = []
|
|
if self.get("mac", False):
|
|
matches.append("MACAddress={}\n".format(self.get("mac")))
|
|
if self.get("device", False):
|
|
matches.append("Name={}\n".format(self.get("device")))
|
|
return "[Match]\n" + "".join(matches)
|
|
|
|
def network(self) -> str:
|
|
output = "[Network]\n"
|
|
options = []
|
|
try:
|
|
options.append("Description={}".format(self.get("description")))
|
|
except KeyError:
|
|
pass
|
|
try:
|
|
for server in self.get("dns", []):
|
|
options.append(f"DNS={server}")
|
|
options.append("DNSDefaultRoute={}".format(self.get("defaultdns", False)))
|
|
except KeyError:
|
|
pass
|
|
try:
|
|
domain = self.get("domain")
|
|
self.set("domainlog", str(domain))
|
|
options.append("Domains={}".format(" ".join(domain)))
|
|
options.append(
|
|
"DNSOverTLS={}".format(boolconvert(self.get("dot", "opportunistic")))
|
|
)
|
|
options.append(
|
|
"DNSSEC={}".format(boolconvert(self.get("dnssec", "allow-downgrade")))
|
|
)
|
|
except KeyError:
|
|
pass
|
|
output += "\n".join(options)
|
|
return output
|
|
|
|
def addresses(self) -> str:
|
|
output = []
|
|
for address in self.get("address"):
|
|
output.append(f"[Address]\nAddress={address}\n")
|
|
return "\n".join(output)
|
|
|
|
def routes(self) -> str:
|
|
output = []
|
|
routes = self.get("route", [])
|
|
self.set("routes", routes)
|
|
for gw in routes:
|
|
output.append(f"[Route]\nGateway={gw}\nGatewayOnLink=yes\nQuickAck=yes\n")
|
|
self.set("routes", output)
|
|
return "\n".join(output)
|
|
|
|
|
|
DOCUMENTATION = """---
|
|
description:
|
|
- Sets up the systemd network unit
|
|
module: systemd_network
|
|
options:
|
|
address:
|
|
elements: str
|
|
required: true
|
|
type: list
|
|
defaultdns:
|
|
required: false
|
|
type: bool
|
|
description:
|
|
required: false
|
|
type: str
|
|
dns:
|
|
default: []
|
|
elements: str
|
|
required: false
|
|
type: list
|
|
dnssec:
|
|
required: false
|
|
type: bool
|
|
domain:
|
|
default: []
|
|
elements: str
|
|
required: false
|
|
type: list
|
|
dot:
|
|
required: false
|
|
type: bool
|
|
name:
|
|
required: true
|
|
type: str
|
|
mac:
|
|
elements: str
|
|
required: true
|
|
type: list
|
|
route:
|
|
default: []
|
|
elements: str
|
|
required: false
|
|
type: list
|
|
short_description: Sets up the systemd network unit
|
|
|
|
"""
|
|
|
|
|
|
if __name__ == "__main__":
|
|
Module()()
|