Add bearer validation

This commit is contained in:
Matteo Bertucci
2021-01-14 16:08:24 +01:00
parent 8f8503d4f7
commit b8c4a114f6
4 changed files with 53 additions and 19 deletions

View File

@ -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")