#pragma once

#include "rust-ffi.hh"

namespace nix {

/* See path.rs. */
struct StorePath;

class Store;

extern "C" {
    void ffi_StorePath_drop(void *);
    bool ffi_StorePath_less_than(const StorePath & a, const StorePath & b);
    bool ffi_StorePath_eq(const StorePath & a, const StorePath & b);
    void ffi_StorePath_clone_to(const StorePath & _other, StorePath & _this);
    unsigned char * ffi_StorePath_hash_data(const StorePath & p);
}

struct StorePath : rust::Value<3 * sizeof(void *) + 24, ffi_StorePath_drop>
{
    StorePath() = delete;

    static StorePath make(std::string_view path, std::string_view storeDir);

    static StorePath make(unsigned char hash[20], std::string_view name);

    static StorePath fromBaseName(std::string_view baseName);

    rust::String to_string() const;

    bool operator < (const StorePath & other) const
    {
        return ffi_StorePath_less_than(*this, other);
    }

    bool operator == (const StorePath & other) const
    {
        return ffi_StorePath_eq(*this, other);
    }

    bool operator != (const StorePath & other) const
    {
        return !(*this == other);
    }

    StorePath(StorePath && that) = default;

    StorePath(const StorePath & that)
    {
        ffi_StorePath_clone_to(that, *this);
    }

    void operator = (const StorePath & that)
    {
        (rust::Value<3 * sizeof(void *) + 24, ffi_StorePath_drop>::operator = (that));
        ffi_StorePath_clone_to(that, *this);
    }

    StorePath clone() const;

    /* Check whether a file name ends with the extension for
       derivations. */
    bool isDerivation() const;

    std::string_view name() const;

    unsigned char * hashData() const
    {
        return ffi_StorePath_hash_data(*this);
    }

    static StorePath dummy;
};

typedef std::set<StorePath> StorePathSet;
typedef std::vector<StorePath> StorePaths;

StorePathSet cloneStorePathSet(const StorePathSet & paths);
StorePathSet storePathsToSet(const StorePaths & paths);

StorePathSet singleton(const StorePath & path);

/* Size of the hash part of store paths, in base-32 characters. */
const size_t storePathHashLen = 32; // i.e. 160 bits

/* Extension of derivations in the Nix store. */
const std::string drvExtension = ".drv";

struct StorePathWithOutputs
{
    StorePath path;
    std::set<std::string> outputs;

    StorePathWithOutputs(const StorePath & path, const std::set<std::string> & outputs = {})
        : path(path.clone()), outputs(outputs)
    { }

    StorePathWithOutputs(StorePath && path, std::set<std::string> && outputs)
        : path(std::move(path)), outputs(std::move(outputs))
    { }

    StorePathWithOutputs(const StorePathWithOutputs & other)
        : path(other.path.clone()), outputs(other.outputs)
    { }

    std::string to_string(const Store & store) const;
};

std::pair<std::string_view, StringSet> parsePathWithOutputs(std::string_view s);

}

namespace std {

template<> struct hash<nix::StorePath> {
    std::size_t operator()(const nix::StorePath & path) const noexcept
    {
        return * (std::size_t *) path.hashData();
    }
};

}