kai/flake.nix at main · linuxmobile/kai · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
{
  description = "Bug Bounty Hunter Development Environment";

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

  outputs = inputs @ {
    self,
    nixpkgs,
    flake-parts,
    ...
  }:
    flake-parts.lib.mkFlake {inherit inputs;} {
      systems = ["x86_64-linux" "aarch64-linux" "x86_64-darwin" "aarch64-darwin"];

      perSystem = {
        config,
        self',
        inputs',
        pkgs,
        system,
        ...
      }: let
        pkgs = import nixpkgs {
          inherit system;
          config = {
            allowUnfree = true;
            segger-jlink.acceptLicense = true;
            permittedInsecurePackages = [
              "mbedtls-2.28.10"
            ];
          };
          overlays = [
            (final: prev: import ./pkgs {pkgs = prev;})
          ];
        };

        modules = import ./modules {
          inherit pkgs system;
        };
      in {
        packages = import ./pkgs {inherit pkgs;};

        devShells.default = pkgs.mkShell {
          buildInputs = with pkgs;
            [
              httpx
              jq
              ripgrep
              skim
              wget
              parallel
              moreutils
              inetutils
              go
              python3
            ]
            # All package modules automatically included
            ++ builtins.attrValues modules.packages
            ++ modules.scripts;

          shellHook = ''

            export NUCLEI_TEMPLATES="${pkgs.nuclei-templates}/share/nuclei-templates"
            export WORDLISTS="${pkgs.wordlists}/share/wordlists"
            export DIRBUSTER="${pkgs.dirbuster}/share/dirbuster"
            export SECLISTS="${pkgs.seclists}/share/wordlists/seclists"
            export MEDIUM="${pkgs.seclists}/share/wordlists/seclists/Discovery/Web-Content/raft-medium-directories.txt"
            export SMALL="${pkgs.seclists}/share/wordlists/seclists/Discovery/Web-Content/raft-small-directories.txt"
            export API="${pkgs.seclists}/share/wordlists/seclists/Discovery/Web-Content/api/api-endpoints-res.txt"
            export ROCKYOU="${pkgs.wordlists}/share/wordlists/rockyou.txt"
          '';
        };

        _module.args.modules = modules;
      };
    };
}