ledoxide/flake.nix at main · zhufucdev/ledoxide · 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
81
82
83
84
85
86
87
{
  description = "A Nix-flake-based Rust development environment";

  # Nixpkgs / NixOS version to use.
  inputs = {
    nixpkgs.url = "nixpkgs/nixos-unstable";
    crane.url = "github:ipetkov/crane";
  };

  outputs =
    {
      self,
      nixpkgs,
      crane,
    }:
    let

      # to work with older version of flakes
      lastModifiedDate = self.lastModifiedDate or self.lastModified or "19700101";

      # Generate a user-friendly version number.
      version = builtins.substring 0 8 lastModifiedDate;

      # System types to support.
      supportedSystems = [
        "x86_64-linux"
        "x86_64-darwin"
        "aarch64-linux"
        "aarch64-darwin"
      ];

      # Helper function to generate an attrset '{ x86_64-linux = f "x86_64-linux"; ... }'.
      forAllSystems = nixpkgs.lib.genAttrs supportedSystems;

      # Nixpkgs instantiated for supported system types.
      nixpkgsFor = forAllSystems (
        system:
        import nixpkgs {
          inherit system;
          overlays = [ self.overlay ];
          config = {
            allowUnfree = true;
          };
        }
      );

    in

    {

      # A Nixpkgs overlay.
      overlay =
        final: prev:
        let
          craneLib = crane.mkLib final;
          gpuFeatures = if final.stdenv.hostPlatform.isDarwin then [ "metal" ] else [ "cuda" ];
        in
        {
          ledoxide = final.callPackage (import ./nix/package.nix) {
            inherit craneLib;
            inherit version;
            features = gpuFeatures;
          };
        };

      # Provide some binary packages for selected system types.
      packages = forAllSystems (system: {
        default = nixpkgsFor.${system}.ledoxide;
      });

      # A NixOS module, if applicable (e.g. if the package provides a system service).
      nixosModules = {
        ledoxide = import ./nix/module.nix;
        package =
          { pkgs, ... }:
          {
            nixpkgs.overlays = [ self.overlay ];
          };
      };

      # Tests run by 'nix flake check' and by Hydra.
      checks = forAllSystems (system: {
        inherit (self.packages.${system}) default;
      });

    };
}