Merge branch 'new-cli'

This commit is contained in:
Eelco Dolstra 2016-03-04 15:21:23 +01:00
commit 7c9d7a253c
26 changed files with 1287 additions and 163 deletions

View file

@ -0,0 +1,76 @@
#include "binary-cache-store.hh"
#include "download.hh"
namespace nix {
class HttpBinaryCacheStore : public BinaryCacheStore
{
private:
Path cacheUri;
ref<Downloader> downloader;
public:
HttpBinaryCacheStore(std::shared_ptr<Store> localStore,
const Path & secretKeyFile, const Path & publicKeyFile,
const Path & _cacheUri)
: BinaryCacheStore(localStore, secretKeyFile, publicKeyFile)
, cacheUri(_cacheUri)
, downloader(makeDownloader())
{
if (cacheUri.back() == '/')
cacheUri.pop_back();
}
void init() override
{
// FIXME: do this lazily?
if (!fileExists("nix-cache-info"))
throw Error(format("%s does not appear to be a binary cache") % cacheUri);
}
protected:
bool fileExists(const std::string & path) override
{
try {
DownloadOptions options;
options.showProgress = DownloadOptions::no;
options.head = true;
downloader->download(cacheUri + "/" + path, options);
return true;
} catch (DownloadError & e) {
if (e.error == Downloader::NotFound)
return false;
throw;
}
}
void upsertFile(const std::string & path, const std::string & data)
{
throw Error("uploading to an HTTP binary cache is not supported");
}
std::string getFile(const std::string & path) override
{
DownloadOptions options;
options.showProgress = DownloadOptions::no;
return downloader->download(cacheUri + "/" + path, options).data;
}
};
static RegisterStoreImplementation regStore([](const std::string & uri) -> std::shared_ptr<Store> {
if (std::string(uri, 0, 7) != "http://" &&
std::string(uri, 0, 8) != "https://") return 0;
auto store = std::make_shared<HttpBinaryCacheStore>(std::shared_ptr<Store>(0),
"", "", // FIXME: allow the signing key to be set
uri);
store->init();
return store;
});
}

View file

@ -65,13 +65,21 @@ std::string LocalBinaryCacheStore::getFile(const std::string & path)
return readFile(binaryCacheDir + "/" + path);
}
ref<Store> openLocalBinaryCacheStore(std::shared_ptr<Store> localStore,
const Path & secretKeyFile, const Path & publicKeyFile,
const Path & binaryCacheDir)
{
auto store = std::make_shared<LocalBinaryCacheStore>(
localStore, secretKeyFile, publicKeyFile, binaryCacheDir);
store->init();
return ref<Store>(std::shared_ptr<Store>(store));
}
static RegisterStoreImplementation regStore([](const std::string & uri) -> std::shared_ptr<Store> {
if (std::string(uri, 0, 7) != "file://") return 0;
auto store = std::make_shared<LocalBinaryCacheStore>(std::shared_ptr<Store>(0),
return openLocalBinaryCacheStore(std::shared_ptr<Store>(0),
"", "", // FIXME: allow the signing key to be set
std::string(uri, 7));
store->init();
return store;
});
}

View file

@ -453,6 +453,11 @@ ref<Store> openStoreAt(const std::string & uri);
ref<Store> openStore();
ref<Store> openLocalBinaryCacheStore(std::shared_ptr<Store> localStore,
const Path & secretKeyFile, const Path & publicKeyFile,
const Path & binaryCacheDir);
/* Store implementation registration. */
typedef std::function<std::shared_ptr<Store>(const std::string & uri)> OpenStore;