init manage.hs script

colors and helpful output

remove Justfile

build manage as a package

add manage as a precompiled tool to installer

nix/packages/by-name/manage/package.nix

@@ -0,0 +1,10 @@
+{
+  haskellPackages,
+  haskell,
+}:
+let
+  inherit (haskell.lib.compose) justStaticExecutables;
+
+  drv = haskellPackages.callCabal2nix "manage" ./src { };
+in
+justStaticExecutables drv

nix/packages/by-name/manage/src/.gitignore

@@ -0,0 +1 @@
+dist-newstyle/

nix/packages/by-name/manage/src/Main.hs

@@ -0,0 +1,101 @@
+#!/usr/bin/env runhaskell
+{-# LANGUAGE LambdaCase #-}
+{-# LANGUAGE ViewPatterns #-}
+
+import System.Environment
+import System.Exit
+import System.IO
+import System.Process
+
+-- Note: to type check this code
+-- `ghc manage.hs -Werror=all -fno-code'
+
+main :: IO ()
+main = do
+  let showHelpOr args elseDo =
+        if "--help" `elem` args then putStr help
+        else elseDo
+
+  getArgs >>= \case
+    ( "os" : host : action
+      : (partitionArgs -> (manageArgs, extraArgs))
+      ) ->
+        manageArgs `showHelpOr` do
+          nixpkgs <- readNixpkgsPath
+          -- NOTE: It seems like nom is able to do some pty magic and can print out to terminal in real time.
+          -- I thought this would be blocking?
+          readProcessFriendly "nixos-rebuild" (action : nixosRebuildArgs nixpkgs host <> extraArgs) ""
+            >>= putStr
+
+    ( "install" : host
+      : (partitionArgs -> (manageArgs, extraArgs))
+      ) ->
+        manageArgs `showHelpOr` do
+          nixpkgs <- readNixpkgsPath
+          readProcessFriendly "nixos-install" (nixosInstallArgs nixpkgs host <> extraArgs) ""
+            >>= putStr
+
+    _ -> putStr help >> exitFailure
+
+newtype StorePath = StorePath { unStorePath :: String }
+
+defaultNixosCmdArgs :: StorePath -> String -> [String]
+defaultNixosCmdArgs (unStorePath -> nixpkgsPath) hostname =
+  [ "-I", "nixpkgs=" <> nixpkgsPath
+  , "-I", "nixos-config=./nix/configurations/" <> hostname <> ".nix"
+  , "--file", "./default.nix"
+  , "--attr", "nixosConfigurations." <> hostname
+  ]
+
+nixosInstallArgs :: StorePath -> String -> [String]
+nixosInstallArgs = defaultNixosCmdArgs
+
+nixosRebuildArgs :: StorePath -> String -> [String]
+nixosRebuildArgs =
+  defaultNixosCmdArgs
+  -- This is not a nixos-install flag
+  <> (\_ _ -> ["--no-reexec"])
+
+help :: String
+help =
+  unlines
+    [ "Manage.hs"
+    , ""
+    , "Manage.hs is a thin wrapper to make nixos-{install,rebuild} easier to use."
+    , "Black lives matter. Trans rights are human rights. No nazi bullsh*t."
+    , ""
+    , "Available commands:"
+    , indent "os <hostname> <action>:" <> indent "run perhost action with nixos-rebuild"
+    , indent "install <hostname>:" <> indent "run perhost action with nixos-install"
+    , indent "--help:" <> indent "show this help menu"
+    ]
+  where indent = ("\t" <>)
+
+partitionArgs :: [String] -> ([String], [String])
+partitionArgs = finalize . span (/= "--")
+  where finalize (xs, ys) = (xs, drop 1 ys)
+
+quote :: String -> String
+quote x = "\"" <> x <> "\""
+
+-- https://stackoverflow.com/a/70162369
+blueForeground :: String -> String
+blueForeground x = "\ESC[34m" <> x <> "\ESC[0m"
+
+readProcessFriendly :: String -> [String] -> String -> IO String
+readProcessFriendly x args stdin_ = do
+  hPutStrLn stderr
+    $ "Executing: " <> (quote . blueForeground) (showCommandForUser x args)
+  readProcess x args stdin_
+
+readNixpkgsPath :: IO StorePath
+readNixpkgsPath =
+  StorePath . clean <$> readProcessFriendly
+    "nix-instantiate"
+    [ "--eval"
+    , "-E"
+    , "let sources = import ./npins; in sources.nixpkgs.outPath"
+    ]
+    ""
+  where
+    clean = filter (\c -> c /= '\n' && c /= '"')

nix/packages/by-name/manage/src/manage.cabal

@@ -0,0 +1,21 @@
+cabal-version:      3.0
+name:               manage
+version:            0.1.0.0
+description:        NixOS Helper
+author:             Léana 江
+maintainer:         leana.jiang+git@icloud.com
+build-type:         Simple
+
+common common
+  ghc-options:
+    -Wall -Wcompat -Widentities -Wincomplete-record-updates
+    -Wincomplete-patterns -Wincomplete-uni-patterns
+    -Wredundant-constraints -Werror=missing-fields
+  build-depends: base
+  default-language: Haskell2010
+
+executable manage
+  import: common
+  main-is: ./Main.hs
+  build-depends:
+    process