From 2d6414ea41811adead271b60483e89f338261362 Mon Sep 17 00:00:00 2001
From: Raito Bezarius <masterancpp@gmail.com>
Date: Mon, 26 Aug 2024 17:58:40 +0200
Subject: [PATCH] tftp: introduce the FIT enclosing boot.scr

This simplify TFTP.

Signed-off-by: Raito Bezarius <masterancpp@gmail.com>
---
 modules/outputs/tftpboot-fit.its | 14 +++++++++++++
 modules/outputs/tftpboot.nix     | 36 ++++++++++++++++++++++++++++++++
 2 files changed, 50 insertions(+)
 create mode 100644 modules/outputs/tftpboot-fit.its

diff --git a/modules/outputs/tftpboot-fit.its b/modules/outputs/tftpboot-fit.its
new file mode 100644
index 0000000..fcb58ab
--- /dev/null
+++ b/modules/outputs/tftpboot-fit.its
@@ -0,0 +1,14 @@
+/dts-v1/;
+
+/ {
+  description = "Liminix TFTP bootscript";
+  #address-cells = <1>;
+
+  images {
+      bootscript {
+        description = "Bootscript";
+        data = /incbin/("boot.scr");
+        type = "script";
+        compression = "none";
+      };           
+};
diff --git a/modules/outputs/tftpboot.nix b/modules/outputs/tftpboot.nix
index f8f0572..94d5ef4 100644
--- a/modules/outputs/tftpboot.nix
+++ b/modules/outputs/tftpboot.nix
@@ -51,11 +51,47 @@ in {
         It uses the Linux `phram <https://github.com/torvalds/linux/blob/master/drivers/mtd/devices/phram.c>`_ driver to emulate a flash device using a segment of physical RAM.
       '';
     };
+
+    tftpboot-fit = mkOption {
+      type = types.package;
+      description = ''
+        tftpboot-fit
+        ************
+
+        This output is a variant that encloses the `boot.scr` in a FIT
+        if that's simpler to transfer for you.
+      '';
+    };
   };
   config = {
     boot.ramdisk.enable = true;
 
     system.outputs = rec {
+      tftpboot-fit = 
+      let
+        tftpboot-fit = pkgs.writeText "tftpboot.its" ''
+        /dts-v1/;
+
+        / {
+          description = "Liminix TFTP bootscript";
+          #address-cells = <1>;
+
+          images {
+              bootscript {
+                description = "Bootscript";
+                data = /incbin/("${tftpboot}/boot.scr");
+                type = "script";
+                compression = "none";
+              };
+          };
+        };
+        '';
+      in
+        pkgs.runCommand "tftpboot-fit" { nativeBuildInputs = with pkgs.pkgsBuildBuild; [ ubootTools ]; } ''
+        mkdir -p $out/
+        cp -rf ${tftpboot}/* $out/
+        mkimage -f ${tftpboot-fit} $out/script.ub
+      '';
       tftpboot =
         let
           inherit (pkgs.lib.trivial) toHexString;