about summary refs log tree commit diff
path: root/src/libstore/download.hh
blob: 752f918575f85e0780ab892c0398798631ed918f (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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
#pragma once

#include "types.hh"
#include "hash.hh"
#include "globals.hh"

#include <string>
#include <future>

namespace nix {

struct DownloadRequest
{
    std::string uri;
    std::string expectedETag;
    bool verifyTLS = true;
    bool head = false;
    size_t tries = 5;
    unsigned int baseRetryTimeMs = 250;
    ActivityId parentAct;
    bool decompress = true;
    std::shared_ptr<std::string> data;
    std::string mimeType;
    std::function<void(char *, size_t)> dataCallback;

    DownloadRequest(const std::string & uri)
        : uri(uri), parentAct(getCurActivity()) { }

    std::string verb()
    {
        return data ? "upload" : "download";
    }
};

struct DownloadResult
{
    bool cached = false;
    std::string etag;
    std::string effectiveUrl;
    std::shared_ptr<std::string> data;
    uint64_t bodySize = 0;
};

struct CachedDownloadResult
{
    // Note: 'storePath' may be different from 'path' when using a
    // chroot store.
    Path storePath;
    Path path;
    std::optional<std::string> etag;
};

class Store;

struct Downloader
{
    /* Enqueue a download request, returning a future to the result of
       the download. The future may throw a DownloadError
       exception. */
    virtual void enqueueDownload(const DownloadRequest & request,
        Callback<DownloadResult> callback) = 0;

    std::future<DownloadResult> enqueueDownload(const DownloadRequest & request);

    /* Synchronously download a file. The request will be retried in
       case of transient failures. */
    DownloadResult download(const DownloadRequest & request);

    /* Download a file, writing its data to a sink. The sink will be
       invoked on the thread of the caller. The request will not be
       retried in case of transient failures. */
    void download(DownloadRequest && request, Sink & sink);

    /* Check if the specified file is already in ~/.cache/nix/tarballs
       and is more recent than ‘tarball-ttl’ seconds. Otherwise,
       use the recorded ETag to verify if the server has a more
       recent version, and if so, download it to the Nix store. */
    CachedDownloadResult downloadCached(
        ref<Store> store, const string & uri, bool unpack, string name = "",
        const Hash & expectedHash = Hash(), string * effectiveUri = nullptr,
        int ttl = settings.tarballTtl);

    enum Error { NotFound, Forbidden, Misc, Transient, Interrupted };
};

/* Return a shared Downloader object. Using this object is preferred
   because it enables connection reuse and HTTP/2 multiplexing. */
ref<Downloader> getDownloader();

/* Return a new Downloader object. */
ref<Downloader> makeDownloader();

class DownloadError : public Error
{
public:
    Downloader::Error error;
    DownloadError(Downloader::Error error, const FormatOrString & fs)
        : Error(fs), error(error)
    { }

    bool isTransient() override
    {
        return error == Downloader::Error::Transient;
    }
};

bool isUri(const string & s);

}