push sheeet

pkgs/by-name/tr/treefmt/modules/default.nix

@@ -0,0 +1,9 @@
+{
+  _class = "treefmtConfig";
+
+  imports = [
+    ./options.nix
+    ./settings.nix
+    ./wrapper.nix
+  ];
+}

pkgs/by-name/tr/treefmt/modules/options.nix

@@ -0,0 +1,40 @@
+{ lib, config, ... }:
+{
+  options = {
+    name = lib.mkOption {
+      type = lib.types.str;
+      default = lib.getName config.package + "-with-config";
+      defaultText = lib.literalExpression "\"\${getName package}-with-config\"";
+      description = ''
+        Name to use for the wrapped treefmt package.
+      '';
+    };
+
+    runtimeInputs = lib.mkOption {
+      type = with lib.types; listOf package;
+      default = [ ];
+      description = ''
+        Packages to include on treefmt's PATH.
+      '';
+    };
+
+    configFile = lib.mkOption {
+      type = lib.types.path;
+      # Ensure file is copied to the store
+      apply = file: if lib.isDerivation file then file else "${file}";
+      defaultText = lib.literalMD "generated from [](#opt-treefmt-settings)";
+      description = ''
+        The treefmt config file.
+      '';
+    };
+
+    package = lib.mkOption {
+      type = lib.types.package;
+      defaultText = lib.literalExpression "pkgs.treefmt";
+      description = ''
+        The treefmt package to wrap.
+      '';
+      internal = true;
+    };
+  };
+}

pkgs/by-name/tr/treefmt/modules/settings.nix

@@ -0,0 +1,26 @@
+{
+  lib,
+  pkgs,
+  config,
+  modulesPath,
+  ...
+}:
+let
+  settingsFormat = pkgs.formats.toml { };
+in
+{
+  options.settings = lib.mkOption {
+    type = lib.types.submoduleWith {
+      specialArgs = { inherit modulesPath; };
+      modules = [
+        { freeformType = settingsFormat.type; }
+      ];
+    };
+    default = { };
+    description = ''
+      Settings used to build a treefmt config file.
+    '';
+  };
+
+  config.configFile = lib.mkOptionDefault (settingsFormat.generate "treefmt.toml" config.settings);
+}

pkgs/by-name/tr/treefmt/modules/wrapper.nix

@@ -0,0 +1,40 @@
+{
+  lib,
+  pkgs,
+  config,
+  options,
+  ...
+}:
+{
+  options.result = lib.mkOption {
+    type = lib.types.package;
+    description = ''
+      The wrapped treefmt package.
+    '';
+    readOnly = true;
+    internal = true;
+  };
+
+  config.result =
+    pkgs.runCommand config.name
+      {
+        nativeBuildInputs = [ pkgs.makeBinaryWrapper ];
+        env = {
+          inherit (config) configFile;
+          binPath = lib.makeBinPath config.runtimeInputs;
+        };
+        passthru = {
+          inherit (config) runtimeInputs;
+          inherit config options;
+        };
+        inherit (config.package) meta version;
+      }
+      ''
+        mkdir -p $out/bin
+        makeWrapper \
+          ${lib.getExe config.package} \
+          $out/bin/treefmt \
+          --prefix PATH : "$binPath" \
+          --add-flags "--config-file $configFile"
+      '';
+}