ollama dev env

flake.nix

{
  description = "ollama dev";
  inputs.nixpkgs.url = github:NixOS/nixpkgs/nixpkgs-unstable;
  outputs = {self, nixpkgs}: 
    let
      forAllSystems = nixpkgs.lib.genAttrs nixpkgs.lib.systems.flakeExposed;
      nixpkgsFor = forAllSystems(system: import nixpkgs {
        inherit system;
      });
    in {
      packages = forAllSystems(system: 
        let
          pkgs = nixpkgsFor.${system};
        in {
          generate-api = pkgs.writeScriptBin "generate-api" ''
            GITROOT=$(git rev-parse --show-toplevel)

            rm -rf $GITROOT/ollama/generated || true

            mkdir $GITROOT/ollama/generated

            ${pkgs.oapi-codegen}/bin/oapi-codegen -package ollama_openapi -generate types,gin $GITROOT/ollama/docs/openapi.yaml > $GITROOT/ollama/generated/ollama_openapi.go
          '';
        });
      devShells = forAllSystems(system: 
        let
          pkgs = nixpkgsFor.${system};
        in {
          default = pkgs.mkShell {
            buildInputs = with pkgs; [
              self.packages.${system}.generate-api
              go
              gopls
              golangci-lint
              delve
            ];
            shellHook = ''
              export PS1='[$PWD]\n❄ '
            '';
          };
        });
    };
}