mirror of
https://github.com/ansible-collections/community.general.git
synced 2026-04-24 20:59:16 +00:00
Make all doc fragments, module utils, and plugin utils private (#11896)
* Make all doc fragments private. * Make all plugin utils private. * Make all module utils private. * Reformat. * Changelog fragment. * Update configs and ignores. * Adjust unit test names.
This commit is contained in:
parent
9ef1dbb6d5
commit
4fa82b9617
807 changed files with 2041 additions and 1702 deletions
|
|
@ -1,142 +0,0 @@
|
|||
# Copyright (c) 2016, Hiroaki Nakamura <hnakamur@gmail.com>
|
||||
# Simplified BSD License (see LICENSES/BSD-2-Clause.txt or https://opensource.org/licenses/BSD-2-Clause)
|
||||
# SPDX-License-Identifier: BSD-2-Clause
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import http.client as http_client
|
||||
import json
|
||||
import os
|
||||
import socket
|
||||
import ssl
|
||||
import typing as t
|
||||
from urllib.parse import urlparse
|
||||
|
||||
from ansible.module_utils.urls import generic_urlparse
|
||||
|
||||
# httplib/http.client connection using unix domain socket
|
||||
HTTPConnection = http_client.HTTPConnection
|
||||
HTTPSConnection = http_client.HTTPSConnection
|
||||
|
||||
|
||||
class UnixHTTPConnection(HTTPConnection):
|
||||
def __init__(self, path: str) -> None:
|
||||
HTTPConnection.__init__(self, "localhost")
|
||||
self.path = path
|
||||
|
||||
def connect(self):
|
||||
sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
||||
sock.connect(self.path)
|
||||
self.sock = sock
|
||||
|
||||
|
||||
class LXDClientException(Exception):
|
||||
def __init__(self, msg: str, **kwargs) -> None:
|
||||
self.msg = msg
|
||||
self.kwargs = kwargs
|
||||
|
||||
|
||||
class LXDClient:
|
||||
def __init__(
|
||||
self,
|
||||
url: str,
|
||||
key_file: str | None = None,
|
||||
cert_file: str | None = None,
|
||||
debug: bool = False,
|
||||
server_cert_file: str | None = None,
|
||||
server_check_hostname: bool = True,
|
||||
) -> None:
|
||||
"""LXD Client.
|
||||
|
||||
:param url: The URL of the LXD server. (e.g. unix:/var/lib/lxd/unix.socket or https://127.0.0.1)
|
||||
:param key_file: The path of the client certificate key file.
|
||||
:param cert_file: The path of the client certificate file.
|
||||
:param debug: The debug flag. The request and response are stored in logs when debug is true.
|
||||
:param server_cert_file: The path of the server certificate file.
|
||||
:param server_check_hostname: Whether to check the server's hostname as part of TLS verification.
|
||||
"""
|
||||
self.url = url
|
||||
self.debug = debug
|
||||
self.logs: list[dict[str, t.Any]] = []
|
||||
self.connection: UnixHTTPConnection | HTTPSConnection
|
||||
if url.startswith("https:"):
|
||||
self.cert_file = cert_file
|
||||
self.key_file = key_file
|
||||
parts = generic_urlparse(urlparse(self.url))
|
||||
ctx = ssl.create_default_context(ssl.Purpose.SERVER_AUTH)
|
||||
if server_cert_file:
|
||||
# Check that the received cert is signed by the provided server_cert_file
|
||||
ctx.load_verify_locations(cafile=server_cert_file)
|
||||
ctx.check_hostname = server_check_hostname
|
||||
ctx.load_cert_chain(cert_file, keyfile=key_file) # type: ignore # TODO!
|
||||
self.connection = HTTPSConnection(parts.get("netloc"), context=ctx)
|
||||
elif url.startswith("unix:"):
|
||||
unix_socket_path = url[len("unix:") :]
|
||||
self.connection = UnixHTTPConnection(unix_socket_path)
|
||||
else:
|
||||
raise LXDClientException("URL scheme must be unix: or https:")
|
||||
|
||||
def do(self, method: str, url: str, body_json=None, ok_error_codes=None, timeout=None, wait_for_container=None):
|
||||
resp_json = self._send_request(method, url, body_json=body_json, ok_error_codes=ok_error_codes, timeout=timeout)
|
||||
if resp_json["type"] == "async":
|
||||
url = f"{resp_json['operation']}/wait"
|
||||
resp_json = self._send_request("GET", url)
|
||||
if wait_for_container:
|
||||
while resp_json["metadata"]["status"] == "Running":
|
||||
resp_json = self._send_request("GET", url)
|
||||
if resp_json["metadata"]["status"] != "Success":
|
||||
self._raise_err_from_json(resp_json)
|
||||
return resp_json
|
||||
|
||||
def authenticate(self, trust_password):
|
||||
body_json = {"type": "client", "password": trust_password}
|
||||
return self._send_request("POST", "/1.0/certificates", body_json=body_json)
|
||||
|
||||
def _send_request(self, method: str, url: str, body_json=None, ok_error_codes=None, timeout=None):
|
||||
try:
|
||||
body = json.dumps(body_json)
|
||||
self.connection.request(method, url, body=body)
|
||||
resp = self.connection.getresponse()
|
||||
resp_data = resp.read()
|
||||
resp_json = json.loads(resp_data)
|
||||
self.logs.append(
|
||||
{
|
||||
"type": "sent request",
|
||||
"request": {"method": method, "url": url, "json": body_json, "timeout": timeout},
|
||||
"response": {"json": resp_json},
|
||||
}
|
||||
)
|
||||
resp_type = resp_json.get("type", None)
|
||||
if resp_type == "error":
|
||||
if ok_error_codes is not None and resp_json["error_code"] in ok_error_codes:
|
||||
return resp_json
|
||||
if resp_json["error"] == "Certificate already in trust store":
|
||||
return resp_json
|
||||
self._raise_err_from_json(resp_json)
|
||||
return resp_json
|
||||
except OSError as e:
|
||||
raise LXDClientException("cannot connect to the LXD server", err=e) from e
|
||||
|
||||
def _raise_err_from_json(self, resp_json):
|
||||
err_params = {}
|
||||
if self.debug:
|
||||
err_params["logs"] = self.logs
|
||||
raise LXDClientException(self._get_err_from_resp_json(resp_json), **err_params)
|
||||
|
||||
@staticmethod
|
||||
def _get_err_from_resp_json(resp_json):
|
||||
err = None
|
||||
metadata = resp_json.get("metadata", None)
|
||||
if metadata is not None:
|
||||
err = metadata.get("err", None)
|
||||
if err is None:
|
||||
err = resp_json.get("error", None)
|
||||
return err
|
||||
|
||||
|
||||
def default_key_file() -> str:
|
||||
return os.path.expanduser("~/.config/lxc/client.key")
|
||||
|
||||
|
||||
def default_cert_file() -> str:
|
||||
return os.path.expanduser("~/.config/lxc/client.crt")
|
||||
Loading…
Add table
Add a link
Reference in a new issue