From 353a199ab28fd052015e61bbec01166949ed7afb Mon Sep 17 00:00:00 2001
From: Daniel Barlow <dan@telent.net>
Date: Tue, 5 Sep 2023 22:42:25 +0100
Subject: [PATCH] odhcpc script: parse ADDRESSES and PREFIXES

so downstream doesn't have to
---
 pkgs/anoia/init.fnl                 |  5 ++-
 pkgs/odhcp-script/odhcp6-script.fnl | 52 +++++++++++++++++++++++++++--
 2 files changed, 54 insertions(+), 3 deletions(-)

diff --git a/pkgs/anoia/init.fnl b/pkgs/anoia/init.fnl
index 06695ba7..795a8eba 100644
--- a/pkgs/anoia/init.fnl
+++ b/pkgs/anoia/init.fnl
@@ -13,4 +13,7 @@
 
 (fn system [s] (assert (os.execute s)))
 
-{ : merge : split : file-exists? : system }
+(fn mkdir [directory]
+  (os.execute (.. "mkdir -p " directory)))
+
+{ : merge : split : file-exists? : system : mkdir }
diff --git a/pkgs/odhcp-script/odhcp6-script.fnl b/pkgs/odhcp-script/odhcp6-script.fnl
index 36544dbf..ccd52797 100644
--- a/pkgs/odhcp-script/odhcp6-script.fnl
+++ b/pkgs/odhcp-script/odhcp6-script.fnl
@@ -1,6 +1,9 @@
+(local { : split : merge : mkdir } (require :anoia))
+(local { : view } (require :fennel))
 
 (local state-directory (assert (os.getenv "SERVICE_STATE")))
-(os.execute (.. "mkdir -p " state-directory))
+
+(mkdir state-directory)
 
 (fn write-value [name value]
   (let [path (.. state-directory "/" name)]
@@ -10,6 +13,49 @@
 (fn write-value-from-env [name]
   (write-value name (os.getenv (string.upper name))))
 
+;; Format: <prefix>/<length>,preferred,valid[,excluded=<excluded-prefix>/<length>][,class=<prefix class #>]
+
+;;(parse-prefix "2001:8b0:de3a:40dc::/64,7198,7198")
+;;(parse-prefix "2001:8b0:de3a:1001::/64,7198,7188,excluded=1/2,thi=10")
+
+(fn parse-prefix [str]
+  (fn parse-extra [s]
+    (let [out {}]
+      (each [name val (string.gmatch s ",(.-)=([^,]+)")]
+        (tset out name val))
+      out))
+  (let [(prefix len preferred valid extra)
+        (string.match str "(.-)::/(%d+),(%d+),(%d+)(.*)$")]
+    (merge {: prefix : len : preferred : valid} (parse-extra extra))))
+
+
+(fn parse-address [str]
+  (fn parse-extra [s]
+    (let [out {}]
+      (each [name val (string.gmatch s ",(.-)=([^,]+)")]
+        (tset out name val))
+      out))
+  (let [(address len preferred valid extra)
+        (string.match str "(.-)/(%d+),(%d+),(%d+)(.*)$")]
+    (merge {: address : len : preferred : valid} (parse-extra extra))))
+
+
+(fn write-addresses [addresses]
+  (each [_ a (ipairs (split " " addresses))]
+    (let [address (parse-address a)
+          keydir (.. "address/" (address.address:gsub ":" "-"))]
+      (mkdir (.. state-directory "/" keydir))
+      (each [k v (pairs address)]
+        (write-value (.. keydir "/" k) v)))))
+
+(fn write-prefixes [prefixes]
+  (each [_ a (ipairs (split " " prefixes))]
+    (let [prefix (parse-prefix a)
+          keydir (.. "prefix/" (prefix.prefix:gsub ":" "-"))]
+      (mkdir (.. state-directory "/" keydir))
+      (each [k v (pairs prefix)]
+        (write-value (.. keydir "/" k) v)))))
+
 ;; we remove state before updating to ensure that consumers don't get
 ;; a half-updated snapshot
 (os.remove (.. state-directory "/state"))
@@ -48,7 +94,9 @@
        :sntp_fqdn
        ]]
   (each [_ n (ipairs wanted)]
-    (write-value-from-env n)))
+    (write-value-from-env n))
+  (write-addresses (os.getenv :ADDRESSES))
+  (write-prefixes (os.getenv :PREFIXES)))
 
 (let [[ifname state] arg
       ready (match state