From 25cb8835edb7ec2ac77942a102e9ee3184b28030 Mon Sep 17 00:00:00 2001
From: Lars Kiesow <lkiesow@uos.de>
Date: Sat, 22 Jul 2023 16:55:48 +0200
Subject: [PATCH] Auto-register new hosts

This patch adds a flag to auto-register newly added hosts with the
firewall. This can be enabled in general via the configuration and
disabled via a command line flag if necessary, or just enabled via
command line flag.
---
 README.md                | 4 ++++
 deterrerscli/__main__.py | 8 +++++++-
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index dbf7ed0..10d41f8 100644
--- a/README.md
+++ b/README.md
@@ -23,6 +23,10 @@ To configure the client, create a file `~/.deterrers.yml` with the following con
 ```yaml
 url: https://deterrers.example.com
 token: <api-token>
+
+# If you want to automatically register newly added hosts
+# Default: false
+auto-register: false
 ```
 
 ## Usane
diff --git a/deterrerscli/__main__.py b/deterrerscli/__main__.py
index 8ec9308..142229d 100644
--- a/deterrerscli/__main__.py
+++ b/deterrerscli/__main__.py
@@ -7,6 +7,7 @@
 from deterrerscli.types import IPV4_TYPE
 
 deterrers = None
+auto_register = False
 
 profiles = click.Choice(
         ('', 'HTTP', 'SSH', 'HTTP+SSH', 'Multipurpose'),
@@ -26,9 +27,11 @@ def print_format(data, format: str):
 @click.group()
 def cli():
     global deterrers
+    global auto_register
     with open(pathlib.Path().home() / '.deterrers.yml', 'r') as f:
         config = yaml.safe_load(f)
     deterrers = deterrersapi.Deterrers(config['url'], config['token'])
+    auto_register = config.get('auto-register', False)
 
 
 @cli.command()
@@ -62,11 +65,14 @@ def delete(ipv4):
 @click.option('--admin', '-a', multiple=True, required=True)
 @click.option('--profile', '-p', default='', type=profiles)
 @click.option('--firewall', '-f', default='', type=host_firewalls)
+@click.option('--register/--no-register', default=False)
 @click.argument('ipv4', type=IPV4_TYPE)
-def add(ipv4, admin, profile, firewall):
+def add(ipv4, admin, profile, firewall, register):
     '''Add IP address to DETERRERS.
     '''
     deterrers.add(ipv4, admin, profile, firewall)
+    if auto_register or register:
+        deterrers.action(ipv4, 'register')
 
 
 @cli.command()