Optimized nginx health-check and some regex

This commit is contained in:
Kevin Veen-Birkenbach 2023-12-12 17:43:13 +01:00
parent 78ee502ba4
commit 04b3ce18e0
2 changed files with 28 additions and 27 deletions

View File

@ -4,52 +4,57 @@ import sys
import re import re
# file in which fqdn server configs are deposit # file in which fqdn server configs are deposit
config_path = {{nginx_servers_directory}} config_path = '{{nginx_servers_directory}}'
# Initialize the error counter # Initialize the error counter
error_counter = 0 error_counter = 0
# Regex pattern to match domain.tld or subdomain.domain.tld # Regex pattern to match domain.tld or *.domain.tld
pattern = re.compile(r"^(?:[\w-]+\.)?[\w-]+\.[\w-]+\.conf$") pattern = re.compile(r"^(?:[\w-]+\.)*[\w-]+\.[\w-]+\.conf$")
# Iterate over each file in the configuration directory # Iterate over each file in the configuration directory
for filename in os.listdir(config_path): for filename in os.listdir(config_path):
if filename.endswith('.conf') and pattern.match(filename): if filename.endswith('.conf') and pattern.match(filename):
# Extract the domain and subdomain from the filename # Extract the domain and subdomain from the filename
name = filename.replace('.conf', '') domain = filename.replace('.conf', '')
parts = name.split('.') parts = domain.split('.')
# Prepare the URL and expected status codes # Prepare the URL and expected status codes
url = f"https://{name}" url = f"https://{domain}"
# Default: Expect status code 200 for a domain # Default: Expect status code 200 or 302 for a domain
expected_statuses = [200] expected_statuses = [200,302]
# Determine expected status codes based on subdomain redirected_domains = [domain['source'] for domain in {{redirect_domain_mappings}}]
if len(parts) == 3: redirected_domains.append("{{domain_mailu}}")
if parts[0] == '{{domain_listmonk}}':
expected_statuses = [401] # Determine expected status codes based on the domain
{% if nginx_matomo_tracking | bool %} if domain == '{{domain_listmonk}}':
elif parts[0] == '{{nginx_www_redirect}}': expected_statuses = [401]
expected_statuses = [200,301] {% if nginx_matomo_tracking | bool %}
{% endif %} elif parts[0] == 'www' or domain in redirected_domains:
elif parts[0] == '{{domain_yourls}}': expected_statuses = [301]
expected_statuses = [403] {% endif %}
elif domain == '{{domain_yourls}}':
expected_statuses = [403]
try: try:
# Send a HEAD request to get only the response header # Send a HEAD request to get only the response header
response = requests.head(url, allow_redirects=True) response = requests.head(url)
# Check if the status code matches the expected statuses # Check if the status code matches the expected statuses
if response.status_code in expected_statuses: if response.status_code in expected_statuses:
print(f"{name}: ok") print(f"{domain}: OK")
else: else:
print(f"{name}: error") print(f"{domain}: ERROR: Expected {expected_statuses}. Got {response.status_code}.")
error_counter += 1 error_counter += 1
except requests.RequestException as e: except requests.RequestException as e:
# Handle exceptions for requests like connection errors # Handle exceptions for requests like connection errors
print(f"{name}: error due to {e}") print(f"{domain}: error due to {e}")
error_counter += 1 error_counter += 1
if error_counter > 0:
print(f"Warning: {error_counter} domains responded with an unexpected https status code.")
# Exit the script with the number of errors as the exit code # Exit the script with the number of errors as the exit code
sys.exit(error_counter) sys.exit(error_counter)

View File

@ -16,12 +16,8 @@
debug: debug:
var: filtered_domains var: filtered_domains
- name: Prepare redirect domain mappings
set_fact:
redirect_domain_mappings: "{{ filtered_domains | map('regex_replace', '^(.*)$', '{ source: \"www.\\1\", target: \"\\1\" }') | map('from_yaml') | list }}"
- name: Include nginx-domain-redirect role with dynamic domain mappings - name: Include nginx-domain-redirect role with dynamic domain mappings
include_role: include_role:
name: nginx-domain-redirect name: nginx-domain-redirect
vars: vars:
domain_mappings: "{{ redirect_domain_mappings }}" domain_mappings: "{{ filtered_domains | map('regex_replace', '^(.*)$', '{ source: \"www.\\1\", target: \"\\1\" }') | map('from_yaml') | list }}"