#include <atomic>

#include "libmain/shared.hh"
#include "libstore/store-api.hh"
#include "libutil/sync.hh"
#include "libutil/thread-pool.hh"
#include "nix/command.hh"

namespace nix {
struct CmdCopy final : StorePathsCommand {
  std::string srcUri, dstUri;

  CheckSigsFlag checkSigs = CheckSigs;

  SubstituteFlag substitute = NoSubstitute;

  CmdCopy() : StorePathsCommand(true) {
    mkFlag()
        .longName("from")
        .labels({"store-uri"})
        .description("URI of the source Nix store")
        .dest(&srcUri);
    mkFlag()
        .longName("to")
        .labels({"store-uri"})
        .description("URI of the destination Nix store")
        .dest(&dstUri);

    mkFlag()
        .longName("no-check-sigs")
        .description("do not require that paths are signed by trusted keys")
        .set(&checkSigs, NoCheckSigs);

    mkFlag()
        .longName("substitute-on-destination")
        .shortName('s')
        .description(
            "whether to try substitutes on the destination store (only "
            "supported by SSH)")
        .set(&substitute, Substitute);
  }

  std::string name() override { return "copy"; }

  std::string description() override { return "copy paths between Nix stores"; }

  Examples examples() override {
    return {
        Example{"To copy Firefox from the local store to a binary cache in "
                "file:///tmp/cache:",
                "nix copy --to file:///tmp/cache $(type -p firefox)"},
        Example{"To copy the entire current NixOS system closure to another "
                "machine via SSH:",
                "nix copy --to ssh://server /run/current-system"},
        Example{"To copy a closure from another machine via SSH:",
                "nix copy --from ssh://server "
                "/nix/store/a6cnl93nk1wxnq84brbbwr6hxw9gp2w9-blender-2.79-rc2"},
#ifdef ENABLE_S3
        Example{"To copy Hello to an S3 binary cache:",
                "nix copy --to s3://my-bucket?region=eu-west-1 nixpkgs.hello"},
        Example{"To copy Hello to an S3-compatible binary cache:",
                "nix copy --to "
                "s3://my-bucket?region=eu-west-1&endpoint=example.com "
                "nixpkgs.hello"},
#endif
    };
  }

  ref<Store> createStore() override {
    return srcUri.empty() ? StoreCommand::createStore() : openStore(srcUri);
  }

  void run(ref<Store> srcStore, Paths storePaths) override {
    if (srcUri.empty() && dstUri.empty()) {
      throw UsageError("you must pass '--from' and/or '--to'");
    }

    ref<Store> dstStore = dstUri.empty() ? openStore() : openStore(dstUri);

    copyPaths(srcStore, dstStore, PathSet(storePaths.begin(), storePaths.end()),
              NoRepair, checkSigs, substitute);
  }
};
}  // namespace nix

static nix::RegisterCommand r1(nix::make_ref<nix::CmdCopy>());