From ab84bd93a3988fcbb2380df6563e575634f4c06f Mon Sep 17 00:00:00 2001
From: ZHANG Yuntian <yt@radxa.com>
Date: Thu, 6 Feb 2025 18:53:08 +0800
Subject: [PATCH] radxa/rock-5b: init

---
 README.md                 |  1 +
 flake.nix                 |  1 +
 radxa/rock-5b/default.nix | 31 +++++++++++++++++++++++++++++++
 3 files changed, 33 insertions(+)
 create mode 100644 radxa/rock-5b/default.nix

diff --git a/README.md b/README.md
index 134c3d3f..4cad0390 100644
--- a/README.md
+++ b/README.md
@@ -348,6 +348,7 @@ See code for all available configurations.
 | [Purism Librem 15v3](purism/librem/13v3)                                          | `<nixos-hardware/purism/librem/15v3>`                   |
 | [Purism Librem 5r4](purism/librem/5r4)                                            | `<nixos-hardware/purism/librem/5r4>`                    |
 | [Radxa ROCK 4C+](radxa/rock-4c-plus)                                              | `<nixos-hardware/radxa/rock-4c-plus>`                   |
+| [Radxa ROCK 5 Model B](radxa/rock-5b)                                             | `<nixos-hardware/radxa/rock-5b>`                        |
 | [Radxa ROCK Pi 4](radxa/rock-pi-4)                                                | `<nixos-hardware/radxa/rock-pi-4>`                      |
 | [Raspberry Pi 2](raspberry-pi/2)                                                  | `<nixos-hardware/raspberry-pi/2>`                       |
 | [Raspberry Pi 3](raspberry-pi/3)                                                  | `<nixos-hardware/raspberry-pi/3>`                       |
diff --git a/flake.nix b/flake.nix
index 8a1dc042..0051c92b 100644
--- a/flake.nix
+++ b/flake.nix
@@ -300,6 +300,7 @@
         raspberry-pi-4 = import ./raspberry-pi/4;
         raspberry-pi-5 = import ./raspberry-pi/5;
         rock-4c-plus = import ./radxa/rock-4c-plus;
+        rock-5b = import ./radxa/rock-5b;
         rock-pi-4 = import ./radxa/rock-pi-4;
         kobol-helios4 = import ./kobol/helios4;
         samsung-np900x3c = import ./samsung/np900x3c;
diff --git a/radxa/rock-5b/default.nix b/radxa/rock-5b/default.nix
new file mode 100644
index 00000000..992b5ca2
--- /dev/null
+++ b/radxa/rock-5b/default.nix
@@ -0,0 +1,31 @@
+{ lib
+, pkgs
+, config
+, ...
+}:
+let
+  cfg = config.hardware.radxa.rock-5b;
+  rkCfg = config.hardware.rockchip;
+in {
+  imports = [
+    ../.
+    ../../rockchip
+  ];
+
+  options.hardware.radxa.rock-5b = {
+    platformFirmware = lib.mkPackageOption pkgs "ubootRock5ModelB" { };
+  };
+
+  config = {
+    hardware = {
+      radxa.enable = true;
+      rockchip = {
+        rk3588.enable = true;
+        diskoExtraPostVM = ''
+          dd conv=notrunc,fsync if=${cfg.platformFirmware}/idbloader.img of=$out/${rkCfg.diskoImageName} bs=512 seek=64
+          dd conv=notrunc,fsync if=${cfg.platformFirmware}/u-boot.itb of=$out/${rkCfg.diskoImageName} bs=512 seek=16384
+        '';
+      };
+    };
+  };
+}