cratery 1.11.1

Cratery -- a private cargo registry
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

#!/usr/bin/env python3

import base64
import json
import sys
from urllib.request import urlopen, Request

def build_token(token_id, token_secret):
    token = f'{token_id}:{token_secret}'
    token_bytes = token.encode('ascii')
    base64_bytes = base64.b64encode(token_bytes)
    base64_str = base64_bytes.decode('ascii')
    return base64_str

def make_request(method, registry_uri, resource, token):
    url = f'{registry_uri}{resource}'
    with urlopen(Request(url, None, {"Authorization": f'Basic {token}', "Accept": "application/json"}, None, False, method)) as response:
        data = json.load(response)
        return data

def main():
    # read args
    registry_uri = sys.argv[1]
    token_id = sys.argv[2]
    token_secret = sys.argv[3]
    # compute final token
    token = build_token(token_id, token_secret)
    crates = make_request("GET", registry_uri, "/api/v1/crates/undocumented", token)
    for crate in crates:
        crate_name = crate["name"]
        crate_version = crate["version"]
        print(f"missing documentation for: {crate_name} {crate_version}")
        make_request("POST", registry_uri, f"/api/v1/crates/{crate_name}/{crate_version}/docsregen", token)

if __name__ == "__main__":
    main()