dotfiles/flake.nix

{
  description = "Rasyid's nix-darwin dotfiles";

  inputs = {
    nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable";
    darwin.url = "github:LnL7/nix-darwin";
    darwin.inputs.nixpkgs.follows = "nixpkgs";

    home-manager.url = "github:nix-community/home-manager";
    home-manager.inputs.nixpkgs.follows = "nixpkgs";

    ghostty.url = "github:ghostty-org/ghostty";
    ghostty.inputs.nixpkgs.follows = "nixpkgs";

    nix-ai-tools.url = "github:numtide/nix-ai-tools";
    nix-ai-tools.inputs.nixpkgs.follows = "nixpkgs";

    sops-nix.url = "github:Mic92/sops-nix";
    sops-nix.inputs.nixpkgs.follows = "nixpkgs";

    chaotic.url = "github:chaotic-cx/nyx/nyxpkgs-unstable";

    try.url = "github:tobi/try";
    try.inputs.nixpkgs.follows = "nixpkgs";

    beads.url = "github:steveyegge/beads";
    beads.inputs.nixpkgs.follows = "nixpkgs";
  };

  outputs = inputs@{ self, nixpkgs, darwin, home-manager, ghostty, nix-ai-tools
    , sops-nix, chaotic, try, beads, ... }:
    let
      inherit (nixpkgs.lib) genAttrs;
      lib = nixpkgs.lib;

      darwinSystems = [ "aarch64-darwin" "x86_64-darwin" ];
      linuxSystems = [ "x86_64-linux" "aarch64-linux" ];
      systems = darwinSystems ++ linuxSystems;

      forEachSystem = f: genAttrs systems (system: f system);

      overlaysList = [ ghostty.overlays.default chaotic.overlays.default ];

      mkPkgs = system:
        import nixpkgs {
          inherit system;
          overlays = overlaysList;
          config.allowUnfree = true;
        };

      sharedDarwinModules = [ ./modules/darwin/system.nix ];
      sharedNixosModules =
        [ ./modules/nixos/system.nix chaotic.nixosModules.default ];

      user = "rasyidanakbar";

      mkDarwin = { system ? "aarch64-darwin", extraModules ? [ ]
        , homeFile ? ./modules/darwin/home/default.nix, }:
        let
          pkgs = mkPkgs system;
          customPkgs = import ./packages { inherit pkgs lib beads; };
        in darwin.lib.darwinSystem {
          inherit pkgs;
          system = pkgs.stdenv.hostPlatform.system;
          specialArgs = { inherit inputs overlaysList user; };
          modules = sharedDarwinModules ++ extraModules ++ [
            home-manager.darwinModules.home-manager
            {
              nix.registry.self.flake = self;
              nixpkgs.overlays = overlaysList;

              home-manager.useGlobalPkgs = true;
              home-manager.useUserPackages = true;
              home-manager.backupFileExtension = "backup";
              home-manager.extraSpecialArgs = {
                inherit inputs user try customPkgs;
              };
              home-manager.users.${user} = import homeFile;
            }
          ];
        };
      mkNixos = { system ? "x86_64-linux", extraModules ? [ ]
        , homeFile ? ./modules/nixos/home/default.nix, }:
        let
          pkgs = mkPkgs system;
          customPkgs = import ./packages { inherit pkgs lib beads; };
        in lib.nixosSystem {
          inherit pkgs;
          system = pkgs.stdenv.hostPlatform.system;
          specialArgs = { inherit inputs overlaysList user; };
          modules = sharedNixosModules ++ extraModules ++ [
            home-manager.nixosModules.home-manager
            {
              nix.registry.self.flake = self;
              nixpkgs.overlays = overlaysList;

              home-manager.useGlobalPkgs = true;
              home-manager.useUserPackages = true;
              home-manager.extraSpecialArgs = {
                inherit inputs user try customPkgs;
              };
              home-manager.users.${user} = import homeFile;
            }
          ];
        };
    in {
      darwinConfigurations = { macbook-pro = mkDarwin { }; };
      nixosConfigurations = {
        dev-vm = mkNixos {
          system = "x86_64-linux";
          extraModules = [ ./hosts/dev-vm.nix ];
          # Uses default homeFile: ./modules/nixos/home/default.nix
        };
        desktop = mkNixos {
          system = "x86_64-linux";
          extraModules = [ ./hosts/desktop.nix ];
          # Home Manager configs are imported directly in desktop modules
          # No homeFile needed - using inline home-manager.users.${user}
          homeFile = ./modules/nixos/home/default.nix;
        };
      };

      devShells = forEachSystem (system:
        let
          pkgs = mkPkgs system;
          python = if builtins.hasAttr "python312" pkgs then
            pkgs.python312
          else
            pkgs.python3;

          # Custom packages (osgrep, beads, etc.)
          customPkgs = import ./packages { inherit pkgs lib beads; };

          # Common arguments passed to all devshell imports
          shellArgs = { inherit pkgs lib python customPkgs; };

          # Import all shells from shells/ directory
          importShell = name: import (./shells + "/${name}.nix") shellArgs;

        in {
          default = importShell "default";
          python-uv = importShell "python-uv";
          ai-notebook = importShell "ai-notebook";
          jupyter-notebook = importShell "jupyter-notebook";
          go = importShell "go";
          web-bun = importShell "web-bun";
          rust = importShell "rust";
          ai-agent = importShell "ai-agent";
        });

      formatter = forEachSystem (system: (mkPkgs system).nixfmt-classic);

      packages = forEachSystem (system:
        let
          pkgs = mkPkgs system;
          customPkgs = import ./packages { inherit pkgs lib beads; };
        in customPkgs // { inherit (pkgs) git; });

      checks = forEachSystem (system:
        let
          pkgs = mkPkgs system;
          customPkgs = import ./packages { inherit pkgs lib beads; };
        in { inherit (customPkgs) osgrep opencode beads; });
    };
}