about summary refs log tree commit diff
path: root/humblebundle/fetch-humble-bundle/default.nix
blob: 3ef2cbe6993f9a1c658b6c71e43f6f9b56d98e89 (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
{ stdenv, curl, cacert, email, password, writeText, fetchFromGitHub
, python, buildPythonPackage, pythonPackages, fetchpatch
}:

{ machineName, downloadName ? "Download", suffix ? "humblebundle", md5 }: let
  cafile = "${cacert}/etc/ca-bundle.crt";

  humbleAPI = buildPythonPackage rec {
    name = "humblebundle-${version}";
    version = "0.1.0";

    src = fetchFromGitHub {
      owner = "saik0";
      repo = "humblebundle-python";
      rev = version;
      sha256 = "04kwghlwgdrhhr8dzc7avr5banpcwcaflh1v81r35jm0yf8inzsa";
    };

    patches = stdenv.lib.singleton (fetchpatch {
      url = "https://github.com/saik0/humblebundle-python/pull/7.patch";
      sha256 = "106306029drs6pi8jp520bm3903w06j7sh1hq57ncbd85h15fzzx";
    });

    propagatedBuildInputs = with pythonPackages; [ requests2 ];
  };

  getDownloadURL = writeText "gethburl.py" ''
    import sys, humblebundle

    def get_products(client):
      gamekeys = client.get_gamekeys()
      for gamekey in gamekeys:
        order = hb.get_order(gamekey)
        if order.subproducts is None:
          continue
        for subproduct in order.subproducts:
          prodname = subproduct.human_name.encode('ascii', 'replace')
          downloads = [(download.machine_name, download.download_struct)
                       for download in subproduct.downloads]
          yield (prodname, downloads)

    def find_download(downloads):
      for machine_name, dstruct in sum(downloads.values(), []):
        if machine_name == '${machineName}':
          for ds in dstruct:
            if ds.name == '${downloadName}':
              return ds
          print >>sys.stderr, \
            'Unable to find ${downloadName} for ${machineName}!'
          print >>sys.stderr, 'Available download types:'
          for ds in dstruct:
            print >>sys.stderr, "  " + ds.name
          raise SystemExit(1)

    hb = humblebundle.HumbleApi()
    hb.login('${email}', '${password}')
    products = dict(get_products(hb))
    dstruct = find_download(products)

    if dstruct is None:
      print >>sys.stderr, 'Cannot find download for ${machineName}!'
      print >>sys.stderr, 'Available machine names:'
      for name, dstructs in sorted(products.items(), key=lambda x: x[0]):
        print >>sys.stderr, "  * " + name
        print >>sys.stderr, "    " + ', '.join(map(lambda x: x[0], dstructs))
      raise SystemExit(1)
    elif dstruct.md5 != '${md5}':
      print >>sys.stderr, \
        'MD5 for ${machineName} is not ${md5} but ' + dstruct.md5 + '.'
      raise SystemExit(1)
    else:
      print dstruct.url.web
  '';
in stdenv.mkDerivation {
  name = "${machineName}.${suffix}";
  outputHashAlgo = "md5";
  outputHash = md5;

  buildInputs = [ python humbleAPI ];

  buildCommand = ''
    url="$(python "${getDownloadURL}")"
    header "downloading $name from $url"
    ${curl}/bin/curl --cacert "${cafile}" --fail --output "$out" "$url"
    stopNest
  '';
}