flake.nix

{
  description = "Dank Material Shell";

  inputs = {
    nixpkgs.url = "github:nixos/nixpkgs/nixos-unstable";
    quickshell = {
      url = "git+https://git.outfoxxed.me/quickshell/quickshell?rev=41828c4180fb921df7992a5405f5ff05d2ac2fff";
      inputs.nixpkgs.follows = "nixpkgs";
    };
    flake-compat = {
      url = "github:NixOS/flake-compat";
      flake = false;
    };
  };

  outputs =
    {
      self,
      nixpkgs,
      quickshell,
      ...
    }:
    let
      goModVersion =
        let
          content = builtins.readFile ./core/go.mod;
          lines = builtins.filter builtins.isString (builtins.split "\n" content);
          goLines = builtins.filter (l: builtins.match "go [0-9]+\\..*" l != null) lines;
          matched =
            if goLines != [ ] then builtins.match "go ([0-9]+)\\.([0-9]+).*" (builtins.head goLines) else null;
        in
        if matched != null then
          {
            major = builtins.elemAt matched 0;
            minor = builtins.elemAt matched 1;
          }
        else
          {
            major = "1";
            minor = "25";
          };
      goForPkgs = pkgs: pkgs.${"go_${goModVersion.major}_${goModVersion.minor}"};
      forEachSystem =
        fn:
        nixpkgs.lib.genAttrs [ "aarch64-darwin" "aarch64-linux" "x86_64-darwin" "x86_64-linux" ] (
          system: fn system nixpkgs.legacyPackages.${system}
        );
      buildDmsPkgs = pkgs: {
        dms-shell = self.packages.${pkgs.stdenv.hostPlatform.system}.default;
        quickshell = quickshell.packages.${pkgs.stdenv.hostPlatform.system}.default;
      };
      mkModuleWithDmsPkgs =
        modulePath:
        args@{ pkgs, ... }:
        {
          imports = [
            (import modulePath (args // { dmsPkgs = buildDmsPkgs pkgs; }))
          ];
        };
      mkQmlImportPath =
        pkgs: qmlPkgs:
        pkgs.lib.concatStringsSep ":" (map (o: "${o}/${pkgs.qt6.qtbase.qtQmlPrefix}") qmlPkgs);

      mkQtPluginPath =
        pkgs: qtPkgs:
        pkgs.lib.concatStringsSep ":" (map (o: "${o}/${pkgs.qt6.qtbase.qtPluginPrefix}") qtPkgs);

      qmlPkgs =
        pkgs: with pkgs.kdePackages; [
          kirigami.unwrapped
          sonnet
          qtmultimedia
          qtimageformats
          kimageformats
        ];
    in
    {
      packages = forEachSystem (
        system: pkgs:
        let
          mkDate =
            longDate:
            pkgs.lib.concatStringsSep "-" [
              (builtins.substring 0 4 longDate)
              (builtins.substring 4 2 longDate)
              (builtins.substring 6 2 longDate)
            ];
          version =
            let
              rawVersion = pkgs.lib.removePrefix "v" (pkgs.lib.trim (builtins.readFile ./quickshell/VERSION));
              cleanVersion = builtins.replaceStrings [ " " ] [ "" ] rawVersion;
              dateSuffix = "+date=" + mkDate (self.lastModifiedDate or "19700101");
              revSuffix = "_" + (self.shortRev or "dirty");
            in
            "${cleanVersion}${dateSuffix}${revSuffix}";
        in
        {
          dms-shell = pkgs.lib.makeOverridable (
            {
              extraQtPackages ? [ ],
            }:
            (pkgs.buildGoModule.override { go = goForPkgs pkgs; }) (
              let
                rootSrc = ./.;
                qtPackages = (qmlPkgs pkgs) ++ extraQtPackages;
              in
              {
                inherit version;
                pname = "dms-shell";
                src = ./core;
                vendorHash = "sha256-dEk7IOd6aQwaxZruxQclN7TGMyb8EJOl6NBWRsoZ9HQ=";

                subPackages = [ "cmd/dms" ];

                ldflags = [
                  "-s"
                  "-w"
                  "-X 'main.Version=${version}'"
                ];

                nativeBuildInputs = with pkgs; [
                  installShellFiles
                  makeWrapper
                ];

                postInstall = ''
                  mkdir -p $out/share/quickshell/dms
                  cp -r ${rootSrc}/quickshell/. $out/share/quickshell/dms/

                  chmod u+w $out/share/quickshell/dms/VERSION
                  echo "${version}" > $out/share/quickshell/dms/VERSION

                  # Install desktop file and icon
                  install -D ${rootSrc}/assets/dms-open.desktop \
                    $out/share/applications/dms-open.desktop
                  install -D ${rootSrc}/core/assets/danklogo.svg \
                    $out/share/hicolor/scalable/apps/danklogo.svg

                  wrapProgram $out/bin/dms \
                    --add-flags "-c $out/share/quickshell/dms" \
                    --prefix "NIXPKGS_QT6_QML_IMPORT_PATH" ":" "${mkQmlImportPath pkgs qtPackages}" \
                    --prefix "QT_PLUGIN_PATH" ":" "${mkQtPluginPath pkgs qtPackages}"

                  install -Dm644 ${rootSrc}/assets/systemd/dms.service \
                    $out/lib/systemd/user/dms.service

                  substituteInPlace $out/lib/systemd/user/dms.service \
                    --replace-fail /usr/bin/dms $out/bin/dms \
                    --replace-fail /usr/bin/pkill ${pkgs.procps}/bin/pkill

                  substituteInPlace $out/share/quickshell/dms/Modules/Greetd/assets/dms-greeter \
                    --replace-fail /bin/bash ${pkgs.bashInteractive}/bin/bash

                  substituteInPlace $out/share/quickshell/dms/assets/pam/fprint \
                    --replace-fail pam_fprintd.so ${pkgs.fprintd}/lib/security/pam_fprintd.so

                  substituteInPlace $out/share/quickshell/dms/assets/pam/u2f \
                    --replace-fail pam_u2f.so ${pkgs.pam_u2f}/lib/security/pam_u2f.so

                  installShellCompletion --cmd dms \
                    --bash <($out/bin/dms completion bash) \
                    --fish <($out/bin/dms completion fish) \
                    --zsh <($out/bin/dms completion zsh)
                '';

                meta = {
                  description = "Desktop shell for wayland compositors built with Quickshell & GO";
                  homepage = "https://danklinux.com";
                  changelog = "https://github.com/AvengeMedia/DankMaterialShell/releases/tag/v${version}";
                  license = pkgs.lib.licenses.mit;
                  mainProgram = "dms";
                  platforms = pkgs.lib.platforms.linux;
                };
              }
            )
          ) { };

          quickshell = quickshell.packages.${system}.default;

          default = self.packages.${system}.dms-shell;
        }
      );

      homeModules.dank-material-shell = mkModuleWithDmsPkgs ./distro/nix/home.nix;

      homeModules.default = self.homeModules.dank-material-shell;

      homeModules.niri = import ./distro/nix/niri.nix;

      homeModules.dankMaterialShell.default = builtins.warn "dank-material-shell: flake output `homeModules.dankMaterialShell.default` has been renamed to `homeModules.dank-material-shell`" self.homeModules.dank-material-shell;

      homeModules.dankMaterialShell.niri = builtins.warn "dank-material-shell: flake output `homeModules.dankMaterialShell.niri` has been renamed to `homeModules.niri`" self.homeModules.niri;

      nixosModules.dank-material-shell = mkModuleWithDmsPkgs ./distro/nix/nixos.nix;

      nixosModules.default = self.nixosModules.dank-material-shell;

      nixosModules.greeter = mkModuleWithDmsPkgs ./distro/nix/greeter.nix;

      nixosModules.dankMaterialShell = builtins.warn "dank-material-shell: flake output `nixosModules.dankMaterialShell` has been renamed to `nixosModules.dank-material-shell`" self.nixosModules.dank-material-shell;

      devShells = forEachSystem (
        system: pkgs:
        let
          devQmlPkgs = [
            quickshell.packages.${system}.default
            pkgs.kdePackages.qtdeclarative
          ]
          ++ (qmlPkgs pkgs);
        in
        {
          default = pkgs.mkShell {
            buildInputs =
              with pkgs;
              [
                (goForPkgs pkgs)
                go-mockery_2
                gopls
                delve
                go-tools
                gnumake

                prek
                uv # for prek
                shellcheck

                # Nix development tools
                nixd
                nil
              ]
              ++ devQmlPkgs;

            shellHook = ''
              touch quickshell/.qmlls.ini 2>/dev/null
              if [ ! -f .git/hooks/pre-commit ]; then prek install; fi
            '';

            QML2_IMPORT_PATH = mkQmlImportPath pkgs devQmlPkgs;
            QT_PLUGIN_PATH = mkQtPluginPath pkgs devQmlPkgs;
          };
        }
      );
    };
}