Infra/lemur
5
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Refactor Acme plugin into AcmeChallenge objects, dns01

Browse Source
This commit is contained in:
Mathias Petermann
2020-10-21 17:01:12 +02:00
parent b91cebf245
commit 812e1dee92
4 changed files with 118 additions and 115 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
lemur/plugins/lemur_acme/acme_handlers.py
View File

@ -229,6 +229,12 @@ class AcmeDnsHandler(AcmeHandler):
current_app.logger.error(f"Unable to fetch DNS Providers: {e}")
self.all_dns_providers = []
def get_all_zones(self, dns_provider):
dns_provider_options = json.loads(dns_provider.credentials)
account_number = dns_provider_options.get("account_id")
dns_provider_plugin = self.get_dns_provider(dns_provider.provider_type)
return dns_provider_plugin.get_zones(account_number=account_number)
def get_dns_challenges(self, host, authorizations):
"""Get dns challenges for provided domain"""
@ -393,27 +399,8 @@ class AcmeDnsHandler(AcmeHandler):
def finalize_authorizations(self, acme_client, authorizations):
for authz_record in authorizations:
self.complete_dns_challenge(acme_client, authz_record)
for authz_record in authorizations:
dns_challenges = authz_record.dns_challenge
for dns_challenge in dns_challenges:
dns_providers = self.dns_providers_for_domain.get(authz_record.host)
for dns_provider in dns_providers:
# Grab account number (For Route53)
dns_provider_plugin = self.get_dns_provider(
dns_provider.provider_type
)
dns_provider_options = json.loads(dns_provider.credentials)
account_number = dns_provider_options.get("account_id")
host_to_validate, _ = self.strip_wildcard(authz_record.host)
host_to_validate = self.maybe_add_extension(
host_to_validate, dns_provider_options
)
dns_provider_plugin.delete_txt_record(
authz_record.change_id,
account_number,
dns_challenge.validation_domain_name(host_to_validate),
dns_challenge.validation(acme_client.client.net.key),
)
self.cleanup_dns_challenges(acme_client, authorizations)
return authorizations