mirror of
https://github.com/Akarys42/cloudflare-ddns-docker.git
synced 2025-05-10 16:55:17 -06:00
Add bearer validation
This commit is contained in:
@ -1,39 +1,43 @@
|
||||
import logging
|
||||
|
||||
import click
|
||||
from email_validator import EmailNotValidError, validate_email
|
||||
|
||||
from .constants import DEFAULT_DELAY
|
||||
from .utils import parse_duration, validate_bearer
|
||||
|
||||
DEFAULT_DELAY = "5 minutes"
|
||||
log = logging.getLogger("ddns")
|
||||
|
||||
|
||||
@click.command()
|
||||
@click.option('--delay', '-d', default=DEFAULT_DELAY, show_default=True)
|
||||
@click.option('--email', '-u', prompt="Enter your Cloudflare Email address")
|
||||
@click.option('--key', '-k', prompt="Enter your Cloudflare Auth key", hide_input=True)
|
||||
def start(delay: str, email: str, key: str) -> None:
|
||||
@click.option('--token', '-k', prompt="Enter your Cloudflare Token", hide_input=True, show_envvar=True)
|
||||
@click.option('-v', '--verbose', is_flag=True, default=False)
|
||||
def start(delay: str, token: str, verbose: int) -> None:
|
||||
"""Main application entrypoint."""
|
||||
logging.basicConfig(
|
||||
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
|
||||
level=logging.DEBUG if verbose else logging.INFO
|
||||
)
|
||||
|
||||
try:
|
||||
duration = parse_duration(delay)
|
||||
except ValueError as e:
|
||||
logging.error(f"Failed to parse delay: {e}")
|
||||
logging.error("Exiting with code 64.")
|
||||
log.error(f"Failed to parse delay: {e}")
|
||||
log.error("Exiting with code 64.")
|
||||
exit(64)
|
||||
|
||||
try:
|
||||
validate_email(email)
|
||||
except EmailNotValidError:
|
||||
logging.warning(f"The email address {email} don't seem valid. Do you have a typo?")
|
||||
log.debug("Validating bearer token.")
|
||||
|
||||
try:
|
||||
validate_bearer(key)
|
||||
except ...:
|
||||
...
|
||||
validate_bearer(token)
|
||||
except ValueError as e:
|
||||
log.error(f"Failed to valid bearer token: {e}")
|
||||
log.error("Exiting with code 64.")
|
||||
exit(64)
|
||||
else:
|
||||
log.info("Successfully validated the bearer token.")
|
||||
|
||||
|
||||
# Main entrypoint
|
||||
if __name__ == "__main__":
|
||||
logging.basicConfig(format="%(asctime)s - %(levelname)s - %(message)s")
|
||||
|
||||
start(auto_envvar_prefix="CF_DDNS")
|
||||
|
Reference in New Issue
Block a user