From 9d0d54c2e403941b77832fcc1505d0450397da2c Mon Sep 17 00:00:00 2001
From: tristan <tristan@tristans.cloud>
Date: Sun, 31 Dec 2023 03:41:46 +0000
Subject: [PATCH] styled components custom behavior

---
 nixite/html.nix        |  4 +++-
 nixite/style.nix       |  8 +++++---
 testing/style.test.nix | 15 ++++++++++++++-
 3 files changed, 22 insertions(+), 5 deletions(-)

diff --git a/nixite/html.nix b/nixite/html.nix
index 4c780b4..aca58eb 100644
--- a/nixite/html.nix
+++ b/nixite/html.nix
@@ -8,7 +8,9 @@ rec {
 
   writeAttrs = attrs:
     toString (builtins.attrValues (
-      builtins.mapAttrs (key: value: ''${key}="${toString value}"'') attrs
+      builtins.mapAttrs (key: value:
+        if (builtins.substring 0 2 key) == "__" then ""
+          else ''${key}="${toString value}"'') attrs
     ));
 
   tag = tag: attrs: child: {
diff --git a/nixite/style.nix b/nixite/style.nix
index 948561b..7f941ca 100644
--- a/nixite/style.nix
+++ b/nixite/style.nix
@@ -8,7 +8,7 @@ let
       // {style = self.style + new.style;};
   };
 
-  mkStyle = identifier: styles: ''
+  mkStyle = identifier: styles: if styles == {} then "" else ''
     ${identifier} {
       ${toString (builtins.attrValues (
       builtins.mapAttrs (
@@ -34,8 +34,10 @@ let
     + id;
 in {
   styled = name: tag: cprops: styles:
-    {
-      ${name} = props: child: (html.tag tag (props // cprops) child);
+    let
+      custom = cprops.__child or (child: child);
+    in{
+      ${name} = props: child: (html.tag tag (props // cprops) (custom child));
 
       style = mkStyle (mkIdentifier tag cprops) styles;
     }
diff --git a/testing/style.test.nix b/testing/style.test.nix
index c57e0bf..fc7c8f7 100644
--- a/testing/style.test.nix
+++ b/testing/style.test.nix
@@ -21,7 +21,13 @@ let
     })
     (style.style "body" {
       foo = "bar";
-    });
+    })
+    (style.styled "list" "ul" {
+      __child = child:
+        assert builtins.isList child;
+        (map (html.tag "li" {}) child);
+    } {})
+  ;
 in [
   (it "makes a p component" {
     expected = html.tag "p" {} "yes";
@@ -42,6 +48,13 @@ in [
     expected = html.tag "div" {class = ["foo" "bar"];} "foobar";
     actual = my.foobar {} "foobar";
   })
+  (it "does custom behavour" {
+    expected = toString (html.tag "ul" {} [
+      (html.tag "li" {} "1")
+      (html.tag "li" {} "2")
+    ]);
+    actual = toString ( my.list {} ["1" "2"] );
+  })
   (it "combines attrs" {
     expected = html.tag "div" {
       id = "foo";