Sign in

flagon-io/g1t

public

g1t/services/packages/src/lib.rs

730 lines31,445 bytesCodeBlame

Pick any line to see why it is the way it is: the commit, the pull request and issue it came from, and what the agent was thinking.

Packages, with a container registry on g1t.sh; workspaces deleted whole and kept 30 days; Members for every member1//! The packages service: the registries a workspace publishes to and
2//! installs from, beside its code (docs/PACKAGES.md). Container images
3//! first, over OCI Distribution 1.1 on `g1t.sh/v2/` (oci.rs).
4//!
5//! Other services reach it over `POST /rpc/<method>`; see
6//! `g1t_contracts::packages` for the methods and their arguments. Any
7//! other request is a registry's own protocol. Files are kept through the
8//! `BlobStore` port (store/), metadata in D1 (db.rs).
9
10mod access;
11mod db;
12mod digest;
13mod limits;
14mod manifest;
15mod names;
npm on g1t.sh: publish and install @<workspace>/<name> with the npm CLI and a g1t token16mod npm;
17mod npm_http;
Packages, with a container registry on g1t.sh; workspaces deleted whole and kept 30 days; Members for every member18mod oci;
19mod quota;
20mod range;
21mod sigv4;
22mod store;
23mod token;
24mod upload;
25
26use std::cell::RefCell;
27use std::collections::HashMap;
28
29use g1t_contracts::audit::{AuditActor, AuditTarget, NewAuditEntry, RecordAuditArgs, Surface};
30use g1t_contracts::credentials::Decision;
31use g1t_contracts::events::{Event, NewEvent, PackageEvent, Publish};
32use g1t_contracts::identity::GitCredentialsArgs;
33use g1t_contracts::packages::*;
34use g1t_contracts::repos::{GetArgs, Repo, RepoPath};
35use g1t_contracts::{FailureCode, Outcome, User, Viewer, new_id};
36use g1t_kit::{args, now_ms, reply, rpc_method};
37use worker::{Context, Env, Fetcher, MessageBatch, Request, Response, Result, ScheduleContext, ScheduledEvent, event};
38
39use access::{Action, LinkedTo, Target};
40use db::{Db, PackageRow};
41use store::{BlobStore, Store};
42
43pub(crate) const SOURCE: &str = "packages";
44/// How large a request body may be: Cloudflare's limit on the zone's plan.
45const DEFAULT_MAX_REQUEST_BYTES: u64 = 100_000_000;
46/// How many packages a listing shows.
47const LIST_LIMIT: u32 = 200;
48const VERSIONS_SHOWN: u32 = 200;
49/// How much one sweep lets go of.
50const SWEEP_BATCH: u32 = 200;
51
52thread_local! {
53 /// Pulls counted since the last write, by package: written at most
54 /// every few seconds, so a busy image costs one write, not one a pull.
55 /// What an isolate holds when it goes away is lost: the count is
56 /// approximate.
57 static DOWNLOADS: RefCell<(HashMap<String, u64>, u64)> = RefCell::new((HashMap::new(), 0));
58}
59const DOWNLOADS_FLUSH_MS: u64 = 10_000;
60
61thread_local! {
62 /// What billing allows each workspace, as asked last, and when.
63 static ALLOWANCES: RefCell<HashMap<String, (quota::Allowance, u64)>> = RefCell::new(HashMap::new());
64}
65/// How long billing's answer is kept.
66const ALLOWANCE_TTL_MS: u64 = 5 * 60 * 1000;
67
68/// Who made a registry request, as its audit entries and versions name them.
69pub struct Caller {
70 pub actor: Option<AuditActor>,
71}
72
73/// What access decisions need about a package, owned.
74pub struct TargetOf {
75 workspace: String,
76 repo: Option<(String, String, bool)>,
77 public: bool,
78}
79
80impl TargetOf {
81 pub fn package(row: &PackageRow) -> TargetOf {
82 TargetOf {
83 workspace: row.workspace.clone(),
84 repo: row
85 .repo_id
86 .as_ref()
87 .map(|id| (id.clone(), row.repo_name.clone().unwrap_or_default(), !row.public())),
88 public: row.public(),
89 }
90 }
91
92 pub fn view(&self) -> Target<'_> {
93 Target {
94 workspace: &self.workspace,
95 repo: self.repo.as_ref().map(|(id, name, private)| LinkedTo { id, name, private: *private }),
96 public: self.public,
97 }
98 }
99}
100
101pub struct Packages {
102 pub db: Db,
103 pub store: Store,
104 identity: Fetcher,
105 repos: Fetcher,
106 events: Fetcher,
107 /// Signs registry tokens (PACKAGES_TOKEN_SECRET).
108 secret: Vec<u8>,
109 max_request: u64,
110 /// The host package addresses start with: `g1t.sh`.
111 host: String,
112 /// Whether free workspaces' package storage is limited (STORAGE_LIMITS
113 /// `on`); off when self-hosted.
114 storage_limits: bool,
115 env: Env,
116}
117
118/// What a workspace's deletion or restoring does to its packages.
119#[derive(Debug, PartialEq, Eq)]
120pub(crate) enum WorkspaceMark {
121 /// Hide them from `at`, while the workspace may still be restored.
122 Hide { workspace: String, at: String },
123 /// Show them again.
124 Show { workspace: String },
125}
126
127/// The mark `event` asks for, if any. A protected workspace (`protected`,
128/// slugs or ids, as identity's PROTECTED_WORKSPACES) is never hidden: it
129/// cannot be deleted, so an event saying so is a mistake.
130pub(crate) fn workspace_mark(event: &Event, protected: &[String], now: &str) -> Option<WorkspaceMark> {
131 let text = |key: &str| event.data[key].as_str().map(|v| v.trim().to_lowercase()).filter(|v| !v.is_empty());
132 let workspace = text("slug")?;
133 match event.kind.as_str() {
134 "workspace.deleting" => {
135 let id = text("workspaceId").unwrap_or_default();
136 if protected.iter().any(|name| *name == workspace || (!id.is_empty() && *name == id)) {
137 return None;
138 }
139 Some(WorkspaceMark::Hide { workspace, at: now.to_owned() })
140 }
141 "workspace.restored" => Some(WorkspaceMark::Show { workspace }),
142 _ => None,
143 }
144}
145
146fn not_found<T>() -> Outcome<T> {
147 Outcome::fail(FailureCode::NotFound, "Package not found.")
148}
149
150impl Packages {
151 pub fn from_env(env: &Env) -> Result<Packages> {
152 let secret = store::var(env, "PACKAGES_TOKEN_SECRET");
153 if secret.len() < 32 {
154 return Err(worker::Error::RustError("PACKAGES_TOKEN_SECRET is missing or shorter than 32 characters".into()));
155 }
156 // 0 is no limit, as self-hosted (nothing in front cuts bodies short).
157 let max_request = match store::var(env, "MAX_REQUEST_BYTES").parse::<u64>() {
158 Ok(0) => u64::MAX,
159 Ok(limit) => limit,
160 Err(_) => DEFAULT_MAX_REQUEST_BYTES,
161 };
162 let host = store::var(env, "REGISTRY_HOST");
163 Ok(Packages {
164 db: Db { db: env.d1("DB")? },
165 store: Store::from_env(env)?,
166 identity: env.service("IDENTITY")?,
167 repos: env.service("REPOS")?,
168 events: env.service("EVENTS")?,
169 secret: secret.into_bytes(),
170 max_request,
171 host: if host.is_empty() { "g1t.sh".to_owned() } else { host },
172 storage_limits: store::var(env, "STORAGE_LIMITS") == "on",
173 env: env.clone(),
174 })
175 }
176
177 async fn viewer_for(&self, username: &str, secret: &str) -> Result<Viewer> {
178 g1t_kit::call(
179 &self.identity,
180 "user_for_git_credentials",
181 &GitCredentialsArgs { username: username.to_owned(), secret: secret.to_owned() },
182 )
183 .await
184 }
185
186 /// A repository of the workspace by name, whoever may see it.
187 async fn repo_by_name(&self, workspace: &str, name: &str) -> Result<Option<Repo>> {
188 if !g1t_contracts::is_valid_repo_name(name) {
189 return Ok(None);
190 }
191 let found: Outcome<Repo> = g1t_kit::call(
192 &self.repos,
193 "get",
194 &GetArgs {
195 path: RepoPath { namespace: workspace.to_owned(), name: name.to_owned() },
196 viewer: Some(User::system(workspace)),
197 },
198 )
199 .await?;
200 Ok(found.into_result().ok().filter(|repo| repo.fork_of.is_none()))
201 }
202
203 /// Who may do what to an image: the package's own settings, or, before
204 /// its first push, the repository it will be linked to.
205 async fn target(&self, name: &names::ImageName, found: Option<&PackageRow>) -> Result<TargetOf> {
206 if let Some(found) = found {
207 return Ok(TargetOf::package(found));
208 }
209 let repo = self.repo_by_name(&name.workspace, name.repo_name()).await?;
210 Ok(TargetOf {
211 workspace: name.workspace.clone(),
212 repo: repo.map(|repo| (repo.id, repo.name, repo.is_private)),
213 public: false,
214 })
215 }
216
217 /// What billing allows the workspace, kept for five minutes. `None`
218 /// when billing cannot be asked: the push is then let through.
219 async fn allowance(&self, workspace: &str) -> Option<quota::Allowance> {
220 let now = now_ms();
221 let kept = ALLOWANCES.with(|kept| kept.borrow().get(workspace).copied());
222 if let Some((allowance, at)) = kept
223 && now.saturating_sub(at) < ALLOWANCE_TTL_MS
224 {
225 return Some(allowance);
226 }
227 let billing = self.env.service("BILLING").ok()?;
228 let asked: Result<quota::Allowance> = g1t_kit::call(
229 &billing,
230 "entitlements",
231 &g1t_contracts::billing::EntitlementsArgs { workspace: workspace.to_owned() },
232 )
233 .await;
234 match asked {
235 Ok(allowance) => {
236 ALLOWANCES.with(|kept| kept.borrow_mut().insert(workspace.to_owned(), (allowance, now)));
237 Some(allowance)
238 }
239 Err(error) => {
240 worker::console_error!("packages: billing could not be asked about {workspace}, letting the push through: {error}");
241 None
242 }
243 }
244 }
245
246 /// Why a push of these files (digest and size) into the package may
247 /// not be kept, if it may not: a free workspace past its free storage.
248 /// Files the workspace already holds add nothing.
249 pub(crate) async fn storage_refusal(&self, package: &PackageRow, files: &[(String, u64)]) -> Result<Option<String>> {
250 if !self.storage_limits || files.is_empty() {
251 return Ok(None);
252 }
253 let digests: Vec<String> = files.iter().map(|(digest, _)| digest.clone()).collect();
254 let held = self.db.held(&package.workspace, &digests).await?;
255 let mut seen = std::collections::HashSet::new();
256 let adding: u64 = files
257 .iter()
258 .filter(|(digest, _)| !held.contains(digest) && seen.insert(digest.clone()))
259 .map(|(_, size)| size)
260 .sum();
261 if adding == 0 {
262 return Ok(None);
263 }
264 let Some(allowance) = self.allowance(&package.workspace).await else {
265 return Ok(None);
266 };
267 let (public_bytes, private_bytes) = self.db.storage(&package.workspace).await?;
268 let public = package.public();
269 let used = if public { public_bytes } else { private_bytes };
270 Ok(quota::decide(&allowance, public, used, adding)
271 .err()
272 .map(|refusal| quota::message(&package.workspace, &refusal)))
273 }
274
275 fn count_download(&self, package_id: &str, ctx: &Context) {
276 let due = DOWNLOADS.with(|counts| {
277 let mut counts = counts.borrow_mut();
278 *counts.0.entry(package_id.to_owned()).or_default() += 1;
279 let now = now_ms();
280 if now.saturating_sub(counts.1) < DOWNLOADS_FLUSH_MS {
281 return None;
282 }
283 counts.1 = now;
284 Some(counts.0.drain().collect::<Vec<_>>())
285 });
286 if let Some(due) = due {
287 let env = self.env.clone();
288 ctx.wait_until(async move {
289 let written = match env.d1("DB") {
290 Ok(db) => Db { db }.add_downloads(&due).await,
291 Err(error) => Err(error),
292 };
293 if let Err(error) = written {
294 worker::console_error!("packages: downloads not counted: {error}");
295 }
296 });
297 }
298 }
299
300 async fn announce(&self, kind: &'static str, package: &PackageRow, data: PackageEvent, caller: &Caller) {
301 let event = NewEvent {
302 kind,
303 source: SOURCE,
304 repo_id: package.repo_id.clone(),
305 actor: caller.actor.as_ref().map(|actor| actor.actor_id.clone()),
306 data,
307 };
308 let published: Result<serde_json::Value> = g1t_kit::call(&self.events, "publish", &Publish { events: vec![event] }).await;
309 if let Err(error) = published {
310 worker::console_error!("packages: {kind} not published: {error}");
311 }
312 }
313
314 async fn audit(&self, caller: &Caller, action: &str, package: &PackageRow, path: Option<&str>, surface: Option<Surface>) {
315 let Some(actor) = caller.actor.clone() else {
316 return;
317 };
318 let target = AuditTarget {
319 workspace: package.workspace.clone(),
320 repo: package.repo_name.as_ref().map(|name| format!("{}/{name}", package.workspace)),
321 path: Some(path.map_or_else(|| format!("{}:{}/{}", package.ecosystem, package.workspace, package.name), str::to_owned)),
322 ..AuditTarget::default()
323 };
324 let mut entry = NewAuditEntry::new(
325 actor,
326 action,
327 surface.unwrap_or(Surface::Registry),
328 target,
329 &Decision::allow("packages"),
330 new_id("req", now_ms()),
331 );
332 entry.result = Some("ok".to_owned());
333 let recorded: Result<u32> = g1t_kit::call(&self.events, "audit_record", &RecordAuditArgs { entries: vec![entry] }).await;
334 if let Err(error) = recorded {
335 worker::console_error!("packages: audit entry not recorded: {error}");
336 }
337 }
338
339 fn summary(&self, row: &db::ListedRow) -> PackageSummary {
340 let p = &row.package;
341 PackageSummary {
342 id: p.id.clone(),
343 workspace: p.workspace.clone(),
344 ecosystem: Ecosystem::parse(&p.ecosystem).unwrap_or(Ecosystem::Container),
345 name: p.name.clone(),
npm on g1t.sh: publish and install @<workspace>/<name> with the npm CLI and a g1t token346 address: match p.ecosystem.as_str() {
347 "npm" => format!("{}/-/npm/@{}/{}", self.host, p.workspace, p.name),
348 _ => format!("{}/{}/{}", self.host, p.workspace, p.name),
349 },
Packages, with a container registry on g1t.sh; workspaces deleted whole and kept 30 days; Members for every member350 visibility: Visibility::parse(&p.visibility),
351 repo: p.repo_id.as_ref().map(|id| LinkedRepo {
352 id: id.clone(),
353 namespace: p.workspace.clone(),
354 name: p.repo_name.clone().unwrap_or_default(),
355 }),
356 description: p.description.clone(),
357 versions: row.version_count,
358 latest: row.latest_tag.clone().or_else(|| row.latest_version.clone()),
359 size: row.bytes,
360 downloads: p.downloads,
361 created_at: p.created_at.clone(),
362 updated_at: p.updated_at.clone(),
363 }
364 }
365
366 async fn listed(&self, workspace: &str, ecosystem: Ecosystem, name: &str) -> Result<Option<db::ListedRow>> {
367 let rows = self.db.list(workspace, Some(ecosystem.as_str()), None, None, LIST_LIMIT).await?;
368 if let Some(row) = rows.into_iter().find(|row| row.package.name == name) {
369 return Ok(Some(row));
370 }
371 // Past the first page: read it alone.
372 Ok(self.db.package(workspace, ecosystem.as_str(), name).await?.filter(|package| !package.hidden()).map(|package| db::ListedRow {
373 package,
374 version_count: 0,
375 bytes: 0,
376 latest_tag: None,
377 latest_version: None,
378 }))
379 }
380
381 async fn list_packages(&self, a: ListPackagesArgs) -> Result<Outcome<Vec<PackageSummary>>> {
382 let workspace = a.workspace.to_lowercase();
383 let rows = self
384 .db
385 .list(&workspace, a.ecosystem.map(Ecosystem::as_str), a.repo_id.as_deref(), a.query.as_deref(), LIST_LIMIT)
386 .await?;
387 let visible = rows
388 .iter()
389 .filter(|row| access::decide(a.viewer.as_ref(), &TargetOf::package(&row.package).view(), Action::Pull).allowed)
390 .map(|row| self.summary(row))
391 .collect();
392 Ok(Outcome::Ok(visible))
393 }
394
395 async fn get_package(&self, a: GetPackageArgs) -> Result<Outcome<PackageDetail>> {
396 let workspace = a.workspace.to_lowercase();
397 let Some(row) = self.listed(&workspace, a.ecosystem, &a.name).await? else {
398 return Ok(not_found());
399 };
400 let target = TargetOf::package(&row.package);
401 let permissions = access::permissions(a.viewer.as_ref(), &target.view());
402 if !permissions.pull {
403 return Ok(not_found());
404 }
405 let tags = self.db.tags(&row.package.id).await?;
406 let versions = self
407 .db
408 .versions(&row.package.id, VERSIONS_SHOWN)
409 .await?
410 .into_iter()
411 .map(|version| {
412 let meta = version.meta();
413 let text = |key: &str| meta[key].as_str().map(str::to_owned);
414 PackageVersion {
415 tags: tags.iter().filter(|tag| tag.version_id == version.id).map(|tag| tag.tag.clone()).collect(),
416 media_type: text("media_type"),
417 artifact_type: text("artifact_type"),
418 platforms: meta["platforms"]
419 .as_array()
420 .map(|list| list.iter().filter_map(|p| p.as_str().map(str::to_owned)).collect())
421 .unwrap_or_default(),
422 id: version.id,
423 version: version.version,
424 digest: version.digest,
425 size: version.size,
426 subject: version.subject,
427 published_by: version.published_by,
428 published_at: version.published_at,
npm on g1t.sh: publish and install @<workspace>/<name> with the npm CLI and a g1t token429 deprecated: version.deprecated,
Packages, with a container registry on g1t.sh; workspaces deleted whole and kept 30 days; Members for every member430 }
431 })
432 .collect();
npm on g1t.sh: publish and install @<workspace>/<name> with the npm CLI and a g1t token433 // The README its page shows: npm's, from the latest version.
434 let readme = match self.db.readme_digest(&row.package.id).await?.and_then(|d| digest::Digest::parse(&d)) {
435 Some(digest) => match self.db.blob(&digest).await? {
436 Some(blob) => self.store.read(&blob.object_key).await?.map(|b| String::from_utf8_lossy(&b).into_owned()),
437 None => None,
438 },
439 None => None,
440 };
Packages, with a container registry on g1t.sh; workspaces deleted whole and kept 30 days; Members for every member441 Ok(Outcome::Ok(PackageDetail {
npm on g1t.sh: publish and install @<workspace>/<name> with the npm CLI and a g1t token442 readme,
Packages, with a container registry on g1t.sh; workspaces deleted whole and kept 30 days; Members for every member443 package: self.summary(&row),
444 versions,
445 tags: tags
446 .into_iter()
447 .map(|tag| PackageTag { tag: tag.tag, digest: tag.digest, updated_at: tag.updated_at })
448 .collect(),
449 permissions,
450 }))
451 }
452
453 /// The package `actor` asks to change, if they may `action` it.
454 async fn for_change(&self, actor: &User, workspace: &str, ecosystem: Ecosystem, name: &str, action: Action) -> Result<Outcome<PackageRow>> {
455 let Some(package) = self.db.package(&workspace.to_lowercase(), ecosystem.as_str(), name).await?.filter(|p| !p.hidden()) else {
456 return Ok(not_found());
457 };
458 let target = TargetOf::package(&package);
459 let decision = access::decide(Some(actor), &target.view(), action);
460 if decision.allowed {
461 return Ok(Outcome::Ok(package));
462 }
463 if !access::decide(Some(actor), &target.view(), Action::Pull).allowed {
464 return Ok(not_found());
465 }
466 Ok(Outcome::fail(FailureCode::Forbidden, decision.reason.unwrap_or_else(|| "Not allowed.".to_owned())))
467 }
468
469 async fn delete_version(&self, a: DeleteVersionArgs) -> Result<Outcome<()>> {
470 let package = match self.for_change(&a.actor, &a.workspace, a.ecosystem, &a.name, Action::Delete).await? {
471 Outcome::Ok(package) => package,
472 Outcome::Fail(failure) => return Ok(Outcome::Fail(failure)),
473 };
474 let Some(version) = self.db.find_version(&package.id, &a.version).await? else {
475 return Ok(Outcome::fail(FailureCode::NotFound, "Version not found."));
476 };
477 let caller = Caller { actor: Some(AuditActor::of(&a.actor)) };
478 self.remove_version(&package, &version, &caller).await?;
479 Ok(Outcome::Ok(()))
480 }
481
482 async fn delete_package(&self, a: DeletePackageArgs) -> Result<Outcome<()>> {
483 let package = match self.for_change(&a.actor, &a.workspace, a.ecosystem, &a.name, Action::Delete).await? {
484 Outcome::Ok(package) => package,
485 Outcome::Fail(failure) => return Ok(Outcome::Fail(failure)),
486 };
487 self.db.delete_package(&package.id).await?;
488 self.db.measure(&package.workspace).await?;
489 let caller = Caller { actor: Some(AuditActor::of(&a.actor)) };
490 self.announce("package.deleted", &package, self.event_of(&package), &caller).await;
491 self.audit(&caller, "package.delete", &package, None, a.surface).await;
492 Ok(Outcome::Ok(()))
493 }
494
495 async fn set_package(&self, a: SetPackageArgs) -> Result<Outcome<PackageSummary>> {
496 let package = match self.for_change(&a.actor, &a.workspace, a.ecosystem, &a.name, Action::Delete).await? {
497 Outcome::Ok(package) => package,
498 Outcome::Fail(failure) => return Ok(Outcome::Fail(failure)),
499 };
500 let caller = Caller { actor: Some(AuditActor::of(&a.actor)) };
501 let now = now_ms();
502 let before = package.visibility.clone();
503 if a.unlink {
504 self.db.set_link(&package.id, None, "private", now).await?;
505 } else if let Some(link) = a.link.as_deref() {
506 let Some(repo) = self.repo_by_name(&package.workspace, link).await? else {
507 return Ok(Outcome::fail(FailureCode::NotFound, format!("There is no repository {}/{link}.", package.workspace)));
508 };
509 // Linking hands the package to the repository's roles: only
510 // someone who administers that repository may.
511 let role = g1t_contracts::access::permission(Some(&a.actor), &repo);
512 if role < Some(g1t_contracts::access::RepoRole::Admin) {
513 return Ok(Outcome::fail(
514 FailureCode::Forbidden,
515 format!("You need the Admin role on {}/{} to link a package to it.", repo.namespace, repo.name),
516 ));
517 }
518 let visibility = if repo.is_private { "private" } else { "public" };
519 self.db.set_link(&package.id, Some((&repo.id, &repo.name)), visibility, now).await?;
520 }
521 if let Some(visibility) = a.visibility {
522 let linked = if a.unlink { false } else { a.link.is_some() || package.repo_id.is_some() };
523 if linked {
524 return Ok(Outcome::fail(
525 FailureCode::Invalid,
526 "A linked package has its repository's visibility. Change the repository's, or unlink the package.",
527 ));
528 }
529 self.db.set_visibility(&package.id, visibility.as_str(), now).await?;
530 }
531 self.audit(&caller, "package.update", &package, None, a.surface).await;
532 let Some(row) = self.listed(&package.workspace, a.ecosystem, &package.name).await? else {
533 return Ok(not_found());
534 };
535 if row.package.visibility != before {
536 self.db.measure(&package.workspace).await?;
537 let event = PackageEvent { visibility: Some(row.package.visibility.clone()), ..self.event_of(&row.package) };
538 self.announce("package.visibility_changed", &row.package, event, &caller).await;
539 }
540 Ok(Outcome::Ok(self.summary(&row)))
541 }
542
543 /// Lets go of expired uploads, and of blobs no version has used for a day.
544 async fn sweep(&self) -> Result<(u32, u32)> {
545 let now = now_ms();
546 let mut uploads = 0;
547 for row in self.db.expired_uploads(now, SWEEP_BATCH).await? {
548 if let Some(progress) = row.progress()
549 && let Err(error) = upload::abort(&self.store, &progress).await
550 {
551 worker::console_error!("packages: upload {} not aborted: {error}", row.id);
552 }
553 self.db.delete_upload(&row.id).await?;
554 uploads += 1;
555 }
556 let mut blobs = 0;
557 for blob in self.db.unused_blobs(now, SWEEP_BATCH).await? {
558 self.store.delete(&blob.object_key).await?;
559 self.db.forget_blob(&blob.digest).await?;
560 blobs += 1;
561 }
562 Ok((uploads, blobs))
563 }
564
565 /// Follows what happens elsewhere: workspaces renamed and deleted,
566 /// repositories that change visibility, are renamed, move or go.
567 async fn on_event(&self, env: &Env, event: &Event) -> Result<()> {
568 let db = &self.db.db;
569 let protected = g1t_contracts::identity::protected_names(Some(&store::var(env, "PROTECTED_WORKSPACES")));
570 match workspace_mark(event, &protected, &g1t_contracts::time::rfc3339(now_ms())) {
571 Some(WorkspaceMark::Hide { workspace, at }) => return self.db.mark_workspace(&workspace, Some(&at)).await,
572 Some(WorkspaceMark::Show { workspace }) => return self.db.mark_workspace(&workspace, None).await,
573 None => {}
574 }
575 if g1t_kit::rename::on_event(env, db, event, &[
576 "UPDATE packages SET workspace = ?1 WHERE workspace = ?2",
577 "UPDATE uploads SET workspace = ?1 WHERE workspace = ?2",
578 "UPDATE workspace_blobs SET workspace = ?1 WHERE workspace = ?2",
579 ])
580 .await?
581 {
582 return Ok(());
583 }
584 if g1t_kit::deleted::on_event(db, event, &[
585 "DELETE FROM tags WHERE package_id IN (SELECT id FROM packages WHERE workspace = ?1)",
586 "DELETE FROM version_files WHERE version_id IN (SELECT v.id FROM versions v JOIN packages p ON p.id = v.package_id WHERE p.workspace = ?1)",
587 "DELETE FROM versions WHERE package_id IN (SELECT id FROM packages WHERE workspace = ?1)",
588 "DELETE FROM package_blobs WHERE package_id IN (SELECT id FROM packages WHERE workspace = ?1)",
589 "DELETE FROM uploads WHERE workspace = ?1",
590 "DELETE FROM workspace_blobs WHERE workspace = ?1",
591 "DELETE FROM packages WHERE workspace = ?1",
592 ])
593 .await?
594 {
595 return Ok(());
596 }
597 // A repository renamed keeps its packages linked under its new
598 // name; one moved to another workspace leaves them behind, unlinked.
599 if g1t_kit::transfer::on_event(env, db, event, &[
600 "UPDATE packages SET repo_name = ?6 WHERE repo_id = ?5 AND workspace = ?3",
601 "UPDATE packages SET repo_id = NULL, repo_name = NULL WHERE repo_id = ?5 AND workspace <> ?3",
602 ])
603 .await?
604 {
605 return Ok(());
606 }
607 if g1t_kit::lifecycle::on_purged(db, event, &["UPDATE packages SET repo_id = NULL, repo_name = NULL, visibility = 'private' WHERE repo_id = ?1"])
608 .await?
609 {
610 return Ok(());
611 }
612 if event.kind == "repo.visibility_changed" {
613 #[derive(serde::Deserialize)]
614 #[serde(rename_all = "camelCase")]
615 struct Changed {
616 repo_id: String,
617 is_private: bool,
618 }
619
620 let Ok(changed) = serde_json::from_value::<Changed>(event.data.clone()) else {
621 worker::console_error!("repo.visibility_changed {} could not be read", event.id);
622 return Ok(());
623 };
624 self.db.follow_visibility(&changed.repo_id, changed.is_private).await?;
625 for workspace in self.db.workspaces_linked_to(&changed.repo_id).await? {
626 self.db.measure(&workspace).await?;
627 }
628 }
629 Ok(())
630 }
631}
632
633#[event(fetch)]
634async fn fetch(mut request: Request, env: Env, ctx: Context) -> Result<Response> {
635 let packages = Packages::from_env(&env)?;
636 let Some(method) = rpc_method(&request) else {
npm on g1t.sh: publish and install @<workspace>/<name> with the npm CLI and a g1t token637 if request.path().starts_with("/-/npm/") || request.path() == "/-/npm" {
638 return packages.npm(request, &ctx).await;
639 }
Packages, with a container registry on g1t.sh; workspaces deleted whole and kept 30 days; Members for every member640 return packages.registry(request, &ctx).await;
641 };
642 let body: serde_json::Value = request.json().await?;
643 match method.as_str() {
644 "list_packages" => reply(&packages.list_packages(args(body)?).await?),
645 "get_package" => reply(&packages.get_package(args(body)?).await?),
646 "delete_version" => reply(&packages.delete_version(args(body)?).await?),
647 "delete_package" => reply(&packages.delete_package(args(body)?).await?),
648 "set_package" => reply(&packages.set_package(args(body)?).await?),
649 // For billing: what a workspace's packages hold.
650 "storage_all" => reply(&packages.db.storage_all().await?),
651 "storage" => {
652 let a: StorageArgs = args(body)?;
653 let (public_bytes, private_bytes) = packages.db.storage(&a.workspace.to_lowercase()).await?;
654 reply(&PackageStorage { public_bytes, private_bytes })
655 }
656 _ => Response::error("Unknown method", 404),
657 }
658}
659
660/// Every hour: expired uploads are let go, and blobs no version has used
661/// for a day are deleted from the store.
662#[event(scheduled)]
663async fn scheduled(_event: ScheduledEvent, env: Env, _ctx: ScheduleContext) {
664 let packages = match Packages::from_env(&env) {
665 Ok(packages) => packages,
666 Err(error) => {
667 worker::console_error!("packages: the sweep could not start: {error}");
668 return;
669 }
670 };
671 match packages.sweep().await {
672 Ok((0, 0)) => {}
673 Ok((uploads, blobs)) => worker::console_log!("packages: let go of {uploads} uploads and {blobs} blobs"),
674 Err(error) => worker::console_error!("packages: the sweep failed: {error}"),
675 }
676}
677
678#[event(queue)]
679async fn queue(batch: MessageBatch<Event>, env: Env, _ctx: Context) -> Result<()> {
680 let packages = Packages::from_env(&env)?;
681 for message in batch.messages()? {
682 packages.on_event(&env, message.body()).await?;
683 }
684 Ok(())
685}
686
687#[cfg(test)]
688mod tests {
689 use super::*;
690 use serde_json::json;
691
692 fn event(kind: &str, data: serde_json::Value) -> Event {
693 Event {
694 id: "evt_1".into(),
695 kind: kind.into(),
696 source: "identity".into(),
697 time: "2026-10-06T00:00:00.000Z".into(),
698 repo_id: None,
699 actor: None,
700 data,
701 }
702 }
703
704 #[test]
705 fn a_deleted_workspace_is_hidden_and_a_restored_one_shown() {
706 let protected = g1t_contracts::identity::protected_names(Some("wsp_keep"));
707 let now = "2026-10-06T12:00:00.000Z";
708 let deleting = event("workspace.deleting", json!({ "workspaceId": "wsp_1", "slug": "Acme", "by": "ana", "purgeAfter": "x" }));
709 assert_eq!(
710 workspace_mark(&deleting, &protected, now),
711 Some(WorkspaceMark::Hide { workspace: "acme".into(), at: now.into() })
712 );
713 let restored = event("workspace.restored", json!({ "workspaceId": "wsp_1", "slug": "acme" }));
714 assert_eq!(workspace_mark(&restored, &protected, now), Some(WorkspaceMark::Show { workspace: "acme".into() }));
715 }
716
717 #[test]
718 fn protected_workspaces_and_other_events_are_left_alone() {
719 let protected = g1t_contracts::identity::protected_names(Some("wsp_keep"));
720 let now = "2026-10-06T12:00:00.000Z";
721 let flagon = event("workspace.deleting", json!({ "workspaceId": "wsp_f", "slug": "flagon-io" }));
722 assert_eq!(workspace_mark(&flagon, &protected, now), None, "always protected");
723 let by_id = event("workspace.deleting", json!({ "workspaceId": "wsp_keep", "slug": "kept" }));
724 assert_eq!(workspace_mark(&by_id, &protected, now), None, "protected by id");
725 let purged = event("workspace.deleted", json!({ "workspaceId": "wsp_1", "slug": "acme" }));
726 assert_eq!(workspace_mark(&purged, &protected, now), None, "the purge is handled apart");
727 let nameless = event("workspace.deleting", json!({ "workspaceId": "wsp_1", "slug": "" }));
728 assert_eq!(workspace_mark(&nameless, &protected, now), None);
729 }
730}