OdysseyDecomp/flake.nix

{
  inputs = {
    nixpkgs.url = "github:NixOS/nixpkgs/nixpkgs-unstable";
    flake-parts.url = "github:hercules-ci/flake-parts";
  };

  outputs =
    inputs:
    inputs.flake-parts.lib.mkFlake { inherit inputs; } {
      systems = [
        "aarch64-darwin"
        "aarch64-linux"
        #"x86_64-darwin"  # missing tools artifact for this system
        "x86_64-linux"
      ];
      perSystem =
        {
          system,
          ...
        }:
        let
          pkgs = import inputs.nixpkgs {
            inherit system;
          };

          tools = import ./tools/nix pkgs;
          cacheUrl = tools.cacheUrl;
          tools-artifact = tools.artifacts.${system};

          buildInputs = with pkgs; [
            cmake
            ninja
            llvmPackages_18.clang-unwrapped
            llvmPackages_18.bintools
            ccache
            pkg-config

            (python312.withPackages (python-pkgs: [
              python-pkgs.capstone
              python-pkgs.colorama
              python-pkgs.cxxfilt
              python-pkgs.pyelftools
              python-pkgs.watchdog
              python-pkgs.levenshtein
              python-pkgs.toml
            ]))
            openssl
            ncurses5
          ];
          env = {
            LD_LIBRARY_PATH = "${pkgs.lib.makeLibraryPath buildInputs}";
            LIBCLANG_PATH = "${pkgs.libclang.lib}/lib";
            NX_DECOMP_TOOLS_NX2ELF = "${tools-artifact}/bin/nx2elf";
          };
        in
        {
          _module.args.pkgs = import inputs.nixpkgs {
            config.allowUnfree = true;
          };
          formatter = pkgs.nixfmt-tree;
          devShells.default = pkgs.mkShell (
            {
              inherit buildInputs;
            }
            // env
          );
          packages = {
            setup = pkgs.writeShellApplication {
              name = "setup";
              text = ''
                link() {
                  rm -f "$2"
                  ln -s "$1" "$2"
                }

                echo -n "${cacheUrl}" > toolchain/cache-version-url.txt
                mkdir -p toolchain/bin
                link ${tools-artifact}/bin/clang toolchain/bin/clang
                link ${tools-artifact}/bin/ld.lld toolchain/bin/ld.lld
                link ${tools-artifact}/bin/decompme tools/decompme
                link ${tools-artifact}/bin/check tools/check
                link ${tools-artifact}/bin/listsym tools/listsym

                SMO_NIX_SETUP=1 python3 tools/setup.py "$@"
              '';
              runtimeInputs = buildInputs;
              runtimeEnv = env;
            };
          };
        };
    };
}