artemis/src/cli/cache.toit at 3a495b82455eeb5f8d266bebf4dde819cc8959ea · toitware/artemis · GitHub

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
// Copyright (C) 2022 Toitware ApS. All rights reserved.

import crypto.sha1
import crypto.sha256
import encoding.base64
import uuid show Uuid
import .server-config

/**
Manages cache keys.
*/

cache-key-application-image id/Uuid --broker-config/ServerConfig -> string:
  return "$broker-config.cache-key/application/images/$(id).image"

cache-key-pod-parts -> string
    --broker-config/ServerConfig
    --organization-id/Uuid
    --part-id/string:
  return "$broker-config.cache-key/$organization-id/pod/parts/$part-id"

cache-key-pod-manifest -> string
    --broker-config/ServerConfig
    --organization-id/Uuid
    --pod-id/Uuid:
  return "$broker-config.cache-key/$organization-id/pod/manifest/$pod-id"

cache-key-patch -> string
    --broker-config/ServerConfig
    --organization-id/Uuid
    --patch-id/string:
  return "$broker-config.cache-key/$organization-id/patches/$patch-id"

CACHE-ARTIFACT-KIND-ENVELOPE ::= "envelope"
CACHE-ARTIFACT-KIND-PARTITION-TABLE ::= "partitions"

cache-key-url-artifact --url/string --kind/string -> string:
  HTTP-URL-PREFIX ::= "http://"
  HTTPS-URL-PREFIX ::= "https://"
  filename/string := ?
  if kind == CACHE-ARTIFACT-KIND-ENVELOPE:
    filename = "firmware.envelope"
  else if kind == CACHE-ARTIFACT-KIND-PARTITION-TABLE:
    filename = "partitions"  // We don't know whether it's a '.bin' or '.csv'.
  else:
    throw "Unknown artifact kind: $kind"
  url_without_prefix := (url.trim --left HTTP-URL-PREFIX).trim --left HTTPS-URL-PREFIX
  return "$(kind)s/$url_without_prefix/$filename"

cache-key-git-app --url/string -> string:
  // The URL might have weird characters or might be very long, so hash it.
  // Any hash function would do.
  hash64 := base64.encode --url-mode (sha1.sha1 url)
  url = url.trim --right ".git"
  // For the most common URLs, add the organization and name.
  // This isn't really necessary, but makes it nicer to understand what's in the cache.
  // For example: https://github.com/toitware/cellular.git
  //     becomes: hSGHt2yVMnTuWlP8z6GtRHc0_l8-toitware-cellular
  GIT-PROVIDERS ::= [
    "https://github.com/",
    "git@github.com:",
    "https://gitlab.com/",
    "git@gitlab.com:",
  ]
  GIT-PROVIDERS.do: | provider/string |
    if url.starts-with provider:
      url = url.trim --left provider
      human := url.replace --all "/" "-"
      return "git-app/$hash64-$human"
  return hash64

cache-key-sdk --version/string -> string:
  return "sdks/$version"