From d38864740bb11fba16072b525981943b34f8dbee Mon Sep 17 00:00:00 2001
From: atse <atse@m-labs.hk>
Date: Fri, 23 Aug 2024 11:30:16 +0800
Subject: [PATCH] flake: Add pytec runnables to devShell

Makes it possible to directly run `plot` and `autotune` in the devShell.
---
 flake.nix         | 14 ++++++++++++++
 pytec/autotune.py |  0
 pytec/plot.py     |  0
 3 files changed, 14 insertions(+)
 mode change 100644 => 100755 pytec/autotune.py
 mode change 100644 => 100755 pytec/plot.py

diff --git a/flake.nix b/flake.nix
index af4492f..dad082e 100644
--- a/flake.nix
+++ b/flake.nix
@@ -68,6 +68,19 @@
           matplotlib
         ];
       };
+
+      pytec-dev-wrappers = pkgs.runCommandNoCC "pytec-dev-wrappers" { } ''
+        mkdir -p $out/bin
+        for program in ${self}/pytec/*.py; do
+          if [ -x $program ]; then
+            progname=`basename -s .py $program`
+            outname=$out/bin/$progname
+            echo "#!${pkgs.bash}/bin/bash" >> $outname
+            echo "exec python3 -m $progname \"\$@\"" >> $outname
+            chmod 755 $outname
+          fi
+        done
+      '';
     in
     {
       packages.x86_64-linux = {
@@ -89,6 +102,7 @@
             openocd
             dfu-util
             rlwrap
+            pytec-dev-wrappers
           ]
           ++ (with python3Packages; [
             numpy
diff --git a/pytec/autotune.py b/pytec/autotune.py
old mode 100644
new mode 100755
diff --git a/pytec/plot.py b/pytec/plot.py
old mode 100644
new mode 100755