about summary refs log tree commit diff
path: root/src/libstore/http-binary-cache-store.cc
blob: 9614d0b4cf35d9d92517b3a7fa11e90a1764e860 (plain) (blame)
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
#include "binary-cache-store.hh"
#include "download.hh"
#include "globals.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 & _cacheUri)
        : BinaryCacheStore(localStore, secretKeyFile)
        , 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),
        settings.get("binary-cache-secret-key-file", string("")),
        uri);
    store->init();
    return store;
});

}