jj/flake.nix

{
  description = "jujutsu";

  inputs.nixpkgs-mozilla.url = "github:mozilla/nixpkgs-mozilla";

  outputs = { self, nixpkgs, nixpkgs-mozilla, ... }:
    let
      lib = nixpkgs.lib;
      systems = [
        "aarch64-linux"
        "aarch64-darwin"
        "i686-linux"
        "x86_64-darwin"
        "x86_64-linux"
      ];
      foreachSystem = f: lib.foldl' (attrs: system: lib.recursiveUpdate attrs (f system)) { } systems;
      version = (builtins.fromTOML (builtins.readFile ./Cargo.toml)).package.version;
    in
    {
      overlay = (final: prev: {
        jujutsu = final.callPackage
          (
            { stdenv
            , lib
            , fetchFromGitHub
            , rustPlatform
            , pkgconfig
            , openssl
            , dbus
            , sqlite
            , file
            , gzip
            , makeWrapper
            , Security
            , SystemConfiguration
            , libiconv
            , rust
            }:

            rustPlatform.buildRustPackage rec {
              pname = "jujutsu";
              inherit version;

              src = self;

              cargoLock = {
                lockFile = "${self}/Cargo.lock";
              };
              nativeBuildInputs = [ rust pkgconfig gzip makeWrapper ];
              buildInputs = [ openssl dbus sqlite ]
              ++ lib.optionals stdenv.isDarwin [
                Security
                SystemConfiguration
                libiconv
              ];
            }

          )
          {
            rust = (final.lib.rustLib.fromManifestFile ./toolchain-manifest.toml { inherit (final) stdenv lib fetchurl patchelf; }).rust;
            inherit (final.darwin.apple_sdk.frameworks) Security SystemConfiguration;
          };
      });
    } //
    (foreachSystem (system:
      let
        pkgs = import nixpkgs {
          inherit system;
          overlays = [ nixpkgs-mozilla.overlays.rust self.overlay ];
        };

        # Script to update the toolchain-manifest.toml file which is used in the nix build.
        #
        # nixpkgs doesn't package rust nightly, so the Mozilla nixpkgs Rust overlay is used,
        # which can download the rust toolchain if given a manifest file (a sha256 hash can also be used
        # in which case the overlay downloads the manifest file with fetchurl).
        # The nice thing about this manifest file is that the nix builds are fully hermetic.
        # The downside is that now both nixpkgs needs to be updated (with nix flake update)
        # and the toolchain manifest.
        #
        # To update the toolchain to the latest nightly, run 
        # $ nix develop
        # $ updateToolchainManifest
        # The script also accepts arguments for a different rust channel (first argument)
        # or a specific nightly date (e.g. updateToolchainManifest nightly 2022-02-19)
        updateToolchainManifest = pkgs.writeScriptBin "updateToolchainManifest" ''
          #! /usr/bin/env bash

          set -ex
          
          if [[ -z $1 ]]; then
            channel='"nightly"'
          else
            channel="\"$1\""
          fi

          if [[ -z $2 ]]; then
            date='null'
          else
            date="\"$2\""
          fi

          url=$(nix eval --raw --impure --expr "let flake = (builtins.getFlake (builtins.toString ./.)); in (import flake.inputs.nixpkgs { overlays = [ flake.inputs.nixpkgs-mozilla.overlays.rust ]; }).lib.rustLib.manifest_v2_url { channel = $channel; date = $date; }")
          curl $url > ./toolchain-manifest.toml
        '';
      in
      {
        devShell.${system} = pkgs.mkShell {
          inputsFrom = [ pkgs.jujutsu ];
          packages = [ updateToolchainManifest ];
        };
        packages.${system}.jujutsu = pkgs.jujutsu;
        defaultPackage.${system} = self.packages.${system}.jujutsu;
        defaultApp.${system} = {
          type = "app";
          program = "${pkgs.jujutsu}/bin/jj";
        };
        checks.${system}.jujutsu = pkgs.jujutsu.overrideAttrs ({ ... }: {
          cargoBuildType = "debug";
          cargoCheckType = "debug";
          preCheck = ''
            export RUST_BACKTRACE=1
          '';
        });
      }));
}