Sign in

flagon-io/g1t

public

g1t/services/packages/src/lib.rs

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