From 28d5f07ecbfb086f6cf55a68356ae0b5bce911b9 Mon Sep 17 00:00:00 2001
From: Christoph Ladurner <christoph.ladurner@tugraz.at>
Date: Wed, 17 Jul 2024 13:16:22 +0200
Subject: [PATCH] cli: add two commands

---
 repository_cli/records.py | 62 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 62 insertions(+)

diff --git a/repository_cli/records.py b/repository_cli/records.py
index 2d18f2c..2bee6a6 100644
--- a/repository_cli/records.py
+++ b/repository_cli/records.py
@@ -694,6 +694,68 @@ def publish(data_model: str, record_ids: list, record_id: str) -> None:
         secho(f"record ({record.id}) published", fg=Color.success)
 
 
+@group_records.command("single-ip")
+@option_data_model
+@option_input_file(
+    type_=JSON(),
+    name="record_ids",
+    help_="json array of ids",
+    required=False,
+)
+@option("--record-id", type=STRING)
+@with_appcontext
+def single_ip(record_ids: list, record_id: str, data_model: str) -> None:
+    """Set single ip to record."""
+    identity = get_identity("system_process", role_name="admin")
+    service = get_records_service(data_model=data_model)
+
+    if not record_ids and record_id:
+        record_ids = [record_id]
+
+    for rec_id in record_ids:
+        try:
+            data = get_data(service, rec_id, identity)
+        except RuntimeError as error:
+            secho(error.msg, fg=Color.error)
+        else:
+            data["custom_fields"]["single_ip"] = True
+
+            service.edit(id_=rec_id, identity=identity)
+            service.update_draft(id_=rec_id, identity=identity, data=data)
+            service.publish(id_=rec_id, identity=identity)
+
+
+@group_records.command("ip-network")
+@option_data_model
+@option_input_file(
+    type_=JSON(),
+    name="record_ids",
+    help_="json array of ids",
+    required=False,
+)
+@option("--record-id", type=STRING)
+@with_appcontext
+def ip_network(record_ids: list, record_id: str, data_model: str) -> None:
+    """Set single ip to record."""
+    identity = get_identity("system_process", role_name="admin")
+    service = get_records_service(data_model=data_model)
+
+    if not record_ids and record_id:
+        record_ids = [record_id]
+
+    for rec_id in record_ids:
+        try:
+            data = get_data(service, rec_id, identity)
+        except RuntimeError as error:
+            secho(error.msg, fg=Color.error)
+        else:
+            data["custom_fields"]["ip_network"] = True
+
+            service.edit(id_=rec_id, identity=identity)
+            service.update_draft(id_=rec_id, identity=identity, data=data)
+            service.publish(id_=rec_id, identity=identity)
+
+
 @group_pids.command("add")
 @option_pid
 @option_pid_identifier