Skip to content
Sign in

flagon-io/g1t

public
0

g1t/services/repos/src/registry.rs

965 lines36,635 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.

Rust repos service with shipping; pull requests kept in the model1//! Repository metadata in D1.
2
Agents and memory, checks and conflicts, profiles, slug renames, custom domains3use std::cell::RefCell;
4use std::collections::HashMap;
5
Rust repos service with shipping; pull requests kept in the model6use g1t_contracts::Viewer;
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look7use g1t_contracts::access::{self, Capability, RepoRole};
Rust repos service with shipping; pull requests kept in the model8use g1t_contracts::repos::{Repo, RepoPath};
9use serde::Deserialize;
10use worker::wasm_bindgen::JsValue;
11use worker::{D1Database, Result};
12
13#[derive(Deserialize)]
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look14pub(crate) struct RepoRow {
Rust repos service with shipping; pull requests kept in the model15 id: String,
16 namespace: String,
17 name: String,
18 description: Option<String>,
19 is_private: u8,
20 owner_id: String,
21 default_branch: String,
22 fork_of: Option<String>,
Agents as a team: lifecycle, merge queue, billing and a new shell23 protected: u8,
RFC 3339 timestamps in identity and repos24 created_at: String,
Agents and memory, checks and conflicts, profiles, slug renames, custom domains25 /// Null only on rows written before the column existed and not yet
26 /// migrated; their key is the one worked out from the path.
27 #[serde(default)]
28 store: Option<String>,
Search across all of g1t, Explore, and a command palette29 /// JSON; absent on rows read before the column existed.
30 #[serde(default)]
31 topics: Option<String>,
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look32 #[serde(default)]
33 website: Option<String>,
34 #[serde(default)]
35 archived_at: Option<String>,
36 #[serde(default)]
37 deleted_at: Option<String>,
Mission control shows where you are needed and what agents landed without you; git answers in about 200ms38 /// Bumped by everything that changes the repository's refs; see
39 /// [`RefsState`]. Absent on rows read before the column existed.
40 #[serde(default)]
41 refs_version: Option<f64>,
42 #[serde(default)]
43 refs_open_until: Option<f64>,
Git storage hardened, pages in tens of milliseconds, honest security alerts, and costs reconciled daily44 /// A pull request working copy whose git data was removed, and the
45 /// head it had (forks.rs). Absent before the columns existed.
46 #[serde(default)]
47 retired_at: Option<String>,
48 #[serde(default)]
49 retired_head: Option<String>,
Merge branch 'worktree-agent-a2013627e5ea4ab13'50 /// Until when writes wait, and why: a move between namespaces
51 /// (moves.rs). Absent before the columns existed.
52 #[serde(default)]
53 writes_paused_until: Option<f64>,
54 #[serde(default)]
55 writes_paused_for: Option<String>,
56}
57
58thread_local! {
59 /// Repositories whose writes wait, by id: until when, and why. Filled
60 /// whenever a row is read.
61 static PAUSED: RefCell<HashMap<String, (u64, String)>> = RefCell::new(HashMap::new());
Mission control shows where you are needed and what agents landed without you; git answers in about 200ms62}
63
Merge branch 'worktree-agent-a2013627e5ea4ab13'64/// Records whether writes to the repository with this id wait, as its row says.
65pub fn note_paused(id: &str, until: Option<u64>, reason: Option<&str>) {
66 PAUSED.with(|paused| {
67 let mut paused = paused.borrow_mut();
68 match until {
69 Some(until) => {
70 paused.insert(id.to_owned(), (until, reason.unwrap_or("maintenance").to_owned()));
71 }
72 None => {
73 paused.remove(id);
74 }
75 }
76 });
77}
78
79/// Why writes to the repository with this id wait at `now`, if they do, as
80/// its row last read here said.
81pub fn paused(id: &str, now: u64) -> Option<String> {
82 PAUSED.with(|paused| paused.borrow().get(id).filter(|(until, _)| *until > now).map(|(_, reason)| reason.clone()))
83}
84
Mission control shows where you are needed and what agents landed without you; git answers in about 200ms85/// Where a repository's refs stand, as its row last said: `version` goes up
86/// with every change g1t makes to them, so an answer that lists them (see
87/// refs_cache.rs) is kept under the version it was made at, and a change
88/// leaves it behind. Until `open_until` (milliseconds) a credential that
89/// can change them is out of g1t's hands, and nothing is kept.
90#[derive(Clone, Copy, Debug, Default, PartialEq, Eq)]
91pub struct RefsState {
92 pub version: u64,
93 pub open_until: u64,
94}
95
96/// The newest [`RefsState`] this isolate has read or written, by
97/// repository id. A version only goes up, so an older read finishing late
98/// never takes a newer one back.
99#[derive(Default)]
100pub struct RefsStates {
101 states: HashMap<String, RefsState>,
Rust repos service with shipping; pull requests kept in the model102}
103
Mission control shows where you are needed and what agents landed without you; git answers in about 200ms104impl RefsStates {
105 pub fn note(&mut self, id: &str, state: RefsState) {
106 let kept = self.states.entry(id.to_owned()).or_default();
107 kept.version = kept.version.max(state.version);
108 kept.open_until = kept.open_until.max(state.open_until);
109 }
110
111 pub fn get(&self, id: &str) -> Option<RefsState> {
112 self.states.get(id).copied()
113 }
114}
115
Agents and memory, checks and conflicts, profiles, slug renames, custom domains116thread_local! {
Mission control shows where you are needed and what agents landed without you; git answers in about 200ms117 static REFS: RefCell<RefsStates> = RefCell::new(RefsStates::default());
118}
119
120/// Where the refs of the repository with this id stand, as this isolate
121/// last read them; `None` before the column existed or before its row was
122/// read here.
123pub fn refs_state(id: &str) -> Option<RefsState> {
124 REFS.with(|refs| refs.borrow().get(id))
125}
126
127fn note_refs(id: &str, version: Option<f64>, open_until: Option<f64>) {
128 if let Some(version) = version {
129 let state = RefsState {
130 version: version as u64,
131 open_until: open_until.unwrap_or(0.0) as u64,
132 };
133 REFS.with(|refs| refs.borrow_mut().note(id, state));
134 }
135}
136
137thread_local! {
Git storage hardened, pages in tens of milliseconds, honest security alerts, and costs reconciled daily138 /// Working copies whose git data was removed, by id, with the head
139 /// each had (forks.rs). Filled whenever a row is read.
140 static RETIRED: RefCell<HashMap<String, String>> = RefCell::new(HashMap::new());
141}
142
143/// The head a removed working copy had, if the repository with this id is one.
144pub fn retired(id: &str) -> Option<String> {
145 RETIRED.with(|retired| retired.borrow().get(id).cloned())
146}
147
148/// Records whether the repository with this id is a removed working copy.
149pub fn note_retired(id: &str, head: Option<&str>) {
150 RETIRED.with(|retired| {
151 let mut retired = retired.borrow_mut();
152 match head {
153 Some(head) => {
154 retired.insert(id.to_owned(), head.to_owned());
155 }
156 None => {
157 retired.remove(id);
158 }
159 }
160 });
161}
162
163thread_local! {
Agents and memory, checks and conflicts, profiles, slug renames, custom domains164 /// Store keys that differ from the one a repository's path gives: those
165 /// of repositories whose workspace was renamed after they were made.
166 /// Filled whenever a row is read or written, so every `Repo` this
Merge branch 'worktree-agent-a2013627e5ea4ab13'167 /// service holds has its key here. A key changes only when a move
168 /// between namespaces switches it (moves.rs), and every row read
169 /// after that brings the new one, so requests sharing the isolate can
170 /// share the map.
Agents and memory, checks and conflicts, profiles, slug renames, custom domains171 static MOVED: RefCell<HashMap<String, String>> = RefCell::new(HashMap::new());
172}
173
Mission control shows where you are needed and what agents landed without you; git answers in about 200ms174/// How long a fetch may go by a repository's row as it was read a moment
175/// ago: a clone is two or three requests in quick succession, and each
176/// would otherwise read the same row. Short enough that making a repository
177/// private, archiving or deleting it applies within seconds.
178pub const RECENT_MS: u64 = 5_000;
179
180/// Repositories read in the last [`RECENT_MS`], by path. Only rows that
181/// were found are kept, so a repository just made is never missed.
182#[derive(Default)]
183pub struct Recent {
184 rows: HashMap<(String, String), (Repo, u64)>,
185}
186
187impl Recent {
188 fn key(path: &RepoPath) -> (String, String) {
189 (path.namespace.to_lowercase(), path.name.to_lowercase())
190 }
191
192 pub fn get(&self, path: &RepoPath, now: u64) -> Option<Repo> {
193 self.rows
194 .get(&Self::key(path))
195 .filter(|(_, read)| now.saturating_sub(*read) < RECENT_MS)
196 .map(|(repo, _)| repo.clone())
197 }
198
199 pub fn keep(&mut self, path: &RepoPath, repo: &Repo, now: u64) {
200 self.rows.retain(|_, (_, read)| now.saturating_sub(*read) < RECENT_MS);
201 self.rows.insert(Self::key(path), (repo.clone(), now));
202 }
203}
204
205thread_local! {
206 static RECENT: RefCell<Recent> = RefCell::new(Recent::default());
207}
208
Agents and memory, checks and conflicts, profiles, slug renames, custom domains209/// The key a repository's path gives: what every repository was stored
210/// under before workspaces could be renamed.
211pub fn path_key(repo: &Repo) -> String {
212 format!("{}--{}", repo.namespace, repo.name)
213}
214
Merge branch 'worktree-agent-a2013627e5ea4ab13'215/// Records where a repository is stored, when its path does not say. A
216/// key changes when the repository moves between namespaces (moves.rs),
217/// so one that is the path's again is forgotten.
Agents and memory, checks and conflicts, profiles, slug renames, custom domains218pub fn remember_store(repo: &Repo, store: &str) {
Merge branch 'worktree-agent-a2013627e5ea4ab13'219 MOVED.with(|moved| {
220 let mut moved = moved.borrow_mut();
221 if store != path_key(repo) {
222 moved.insert(repo.id.clone(), store.to_owned());
223 } else {
224 moved.remove(&repo.id);
225 }
226 });
Agents and memory, checks and conflicts, profiles, slug renames, custom domains227}
228
Rust repos service with shipping; pull requests kept in the model229impl From<RepoRow> for Repo {
230 fn from(row: RepoRow) -> Self {
Agents and memory, checks and conflicts, profiles, slug renames, custom domains231 let repo = Repo {
Rust repos service with shipping; pull requests kept in the model232 id: row.id,
233 namespace: row.namespace,
234 name: row.name,
235 description: row.description,
236 is_private: row.is_private != 0,
237 owner_id: row.owner_id,
238 default_branch: row.default_branch,
239 fork_of: row.fork_of,
Agents as a team: lifecycle, merge queue, billing and a new shell240 protected: row.protected != 0,
RFC 3339 timestamps in identity and repos241 created_at: row.created_at,
Search across all of g1t, Explore, and a command palette242 topics: row
243 .topics
244 .as_deref()
245 .and_then(|topics| serde_json::from_str(topics).ok())
246 .unwrap_or_default(),
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look247 website: row.website,
248 archived_at: row.archived_at,
Agents and memory, checks and conflicts, profiles, slug renames, custom domains249 };
250 if let Some(store) = &row.store {
251 remember_store(&repo, store);
Rust repos service with shipping; pull requests kept in the model252 }
Mission control shows where you are needed and what agents landed without you; git answers in about 200ms253 note_refs(&repo.id, row.refs_version, row.refs_open_until);
Git storage hardened, pages in tens of milliseconds, honest security alerts, and costs reconciled daily254 note_retired(&repo.id, row.retired_at.as_ref().and(row.retired_head.as_deref()));
Merge branch 'worktree-agent-a2013627e5ea4ab13'255 note_paused(&repo.id, row.writes_paused_until.map(|until| until as u64), row.writes_paused_for.as_deref());
Agents and memory, checks and conflicts, profiles, slug renames, custom domains256 repo
Rust repos service with shipping; pull requests kept in the model257 }
258}
259
260/// The key a repo is stored under in the git store.
261pub fn store_key(repo: &Repo) -> String {
Git storage hardened, pages in tens of milliseconds, honest security alerts, and costs reconciled daily262 let key = MOVED
Agents and memory, checks and conflicts, profiles, slug renames, custom domains263 .with(|moved| moved.borrow().get(&repo.id).cloned())
Git storage hardened, pages in tens of milliseconds, honest security alerts, and costs reconciled daily264 .unwrap_or_else(|| path_key(repo));
265 // Its interactions with the store are metered for its workspace.
266 crate::meters::note_owner(&key, &repo.namespace);
267 key
Rust repos service with shipping; pull requests kept in the model268}
269
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look270/// The viewer's role on `repo` (see `g1t_contracts::access`): ownership of
271/// its workspace, the workspace's base permission, a direct grant, or
272/// Read on a public repository. A pull request's fork is its author's to
273/// write; whoever can read the repository it came from can read it too,
274/// which `Repos::may_read` checks.
275pub fn role(repo: &Repo, viewer: &Viewer) -> Option<RepoRole> {
276 if repo.fork_of.is_some() {
277 let author = viewer.as_ref().is_some_and(|user| user.id == repo.owner_id);
278 return if author {
279 Some(RepoRole::Write)
280 } else if repo.is_private {
281 None
282 } else {
283 Some(RepoRole::Read)
284 };
285 }
286 access::permission(viewer.as_ref(), repo)
287}
288
289/// Whether the viewer may read `repo`, going by the repository alone.
Rust repos service with shipping; pull requests kept in the model290pub fn can_read(repo: &Repo, viewer: &Viewer) -> bool {
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look291 role(repo, viewer).is_some()
Rust repos service with shipping; pull requests kept in the model292}
293
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look294/// Whether the viewer may push to `repo`: Write or higher, or the author
295/// of a pull request's fork.
Rust repos service with shipping; pull requests kept in the model296pub fn can_write(repo: &Repo, viewer: &Viewer) -> bool {
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look297 can(repo, viewer, Capability::Push)
Rust repos service with shipping; pull requests kept in the model298}
299
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look300/// Whether the viewer may do `capability` in `repo`. A fork has only its
301/// author's Write.
302pub fn can(repo: &Repo, viewer: &Viewer, capability: Capability) -> bool {
303 if repo.fork_of.is_some() {
304 return role(repo, viewer).is_some_and(|role| access::allows(role, capability))
305 && !access::OWNER_ONLY.contains(&capability);
306 }
307 access::can(viewer.as_ref(), repo, capability)
308}
309
Rust repos service with shipping; pull requests kept in the model310fn optional(value: &Option<String>) -> JsValue {
311 value.as_deref().map_or(JsValue::NULL, JsValue::from)
312}
313
314pub struct Registry {
315 pub db: D1Database,
316}
317
318impl Registry {
319 pub async fn by_path(&self, path: &RepoPath) -> Result<Option<Repo>> {
320 Ok(self
321 .db
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look322 .prepare("SELECT * FROM repos WHERE namespace = ? AND name = ? AND deleted_at IS NULL")
Rust repos service with shipping; pull requests kept in the model323 .bind(&[
324 path.namespace.to_lowercase().into(),
325 path.name.to_lowercase().into(),
326 ])?
327 .first::<RepoRow>(None)
328 .await?
329 .map(Repo::from))
330 }
331
Mission control shows where you are needed and what agents landed without you; git answers in about 200ms332 /// The repository at `path`, as read in the last few seconds if it was
333 /// (see [`RECENT_MS`]). For fetches only: a push always reads the row.
334 pub async fn by_path_recent(&self, path: &RepoPath) -> Result<Option<Repo>> {
335 let now = g1t_kit::now_ms();
336 if let Some(repo) = RECENT.with(|recent| recent.borrow().get(path, now)) {
337 return Ok(Some(repo));
338 }
339 let found = self.by_path(path).await?;
340 if let Some(repo) = &found {
341 RECENT.with(|recent| recent.borrow_mut().keep(path, repo, now));
342 }
343 Ok(found)
344 }
345
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look346 /// Its details; who can see it changes with `set_private`.
Agents as a team: lifecycle, merge queue, billing and a new shell347 pub async fn update(
348 &self,
349 id: &str,
350 description: Option<&str>,
351 protected: bool,
Search across all of g1t, Explore, and a command palette352 topics: &[String],
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look353 website: Option<&str>,
Agents as a team: lifecycle, merge queue, billing and a new shell354 ) -> Result<()> {
355 self.db
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look356 .prepare("UPDATE repos SET description = ?, protected = ?, topics = ?, website = ? WHERE id = ?")
Agents as a team: lifecycle, merge queue, billing and a new shell357 .bind(&[
358 description.map_or(JsValue::NULL, JsValue::from),
359 u32::from(protected).into(),
Search across all of g1t, Explore, and a command palette360 serde_json::to_string(topics)?.into(),
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look361 website.map_or(JsValue::NULL, JsValue::from),
Agents as a team: lifecycle, merge queue, billing and a new shell362 id.into(),
363 ])?
364 .run()
365 .await?;
366 Ok(())
367 }
368
Rust repos service with shipping; pull requests kept in the model369 pub async fn by_id(&self, id: &str) -> Result<Option<Repo>> {
370 Ok(self
371 .db
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look372 .prepare("SELECT * FROM repos WHERE id = ? AND deleted_at IS NULL")
Rust repos service with shipping; pull requests kept in the model373 .bind(&[id.into()])?
374 .first::<RepoRow>(None)
375 .await?
376 .map(Repo::from))
377 }
378
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look379 /// The repository at `path`, deleted or not: what holds the name.
380 pub async fn by_path_any(&self, path: &RepoPath) -> Result<Option<(Repo, Option<String>)>> {
381 Ok(self
382 .db
383 .prepare("SELECT * FROM repos WHERE namespace = ? AND name = ?")
384 .bind(&[
385 path.namespace.to_lowercase().into(),
386 path.name.to_lowercase().into(),
387 ])?
388 .first::<RepoRow>(None)
389 .await?
390 .map(|mut row| {
391 let deleted_at = row.deleted_at.take();
392 (Repo::from(row), deleted_at)
393 }))
394 }
395
Issues and pull requests replace intents and attempts396 /// Repos the viewer may see, newest first. Excludes pull request forks.
Workspaces own repositories397 /// With `member_only`, only repos in the viewer's own workspaces.
Rust repos service with shipping; pull requests kept in the model398 pub async fn list(
399 &self,
400 viewer: &Viewer,
401 query: Option<&str>,
402 namespace: Option<&str>,
Workspaces own repositories403 member_only: bool,
Rust repos service with shipping; pull requests kept in the model404 ) -> Result<Vec<Repo>> {
Workspaces own repositories405 let workspaces: Vec<&str> = viewer
406 .iter()
407 .flat_map(|user| &user.workspaces)
408 .map(|membership| membership.slug.as_str())
409 .collect();
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look410 // The workspaces whose private repositories the viewer reads all
411 // of (an owner, or a base permission other than none), and the
412 // repositories they were given a role on: see access.rs. A probe
413 // repository in each workspace stands for all of them.
414 let reading: Vec<&str> = viewer
415 .iter()
416 .flat_map(|user| {
417 user.workspaces.iter().filter(move |membership| {
418 let probe = access::RepoRef { id: "", namespace: &membership.slug, private: true };
419 access::granted(user, probe).is_some()
420 })
421 })
422 .map(|membership| membership.slug.as_str())
423 .collect();
424 let granted: Vec<&str> = viewer
425 .iter()
426 .flat_map(|user| &user.grants)
427 .map(|grant| grant.repo_id.as_str())
428 .collect();
429 let mut params: Vec<JsValue> = vec![
430 serde_json::to_string(&reading)?.into(),
431 serde_json::to_string(&granted)?.into(),
432 ];
433 let private_ok = "(namespace IN (SELECT value FROM json_each(?)) OR id IN (SELECT value FROM json_each(?)))";
Workspaces own repositories434 let mut conditions = vec![
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look435 "fork_of IS NULL AND deleted_at IS NULL".to_owned(),
436 format!("(is_private = 0 OR {private_ok})"),
Workspaces own repositories437 ];
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look438 if member_only {
439 conditions.push("namespace IN (SELECT value FROM json_each(?))".to_owned());
440 params.push(serde_json::to_string(&workspaces)?.into());
441 }
Rust repos service with shipping; pull requests kept in the model442 if let Some(namespace) = namespace {
Workspaces own repositories443 conditions.push("namespace = ?".to_owned());
Rust repos service with shipping; pull requests kept in the model444 params.push(namespace.to_lowercase().into());
445 }
446 if let Some(query) = query.map(str::trim).filter(|query| !query.is_empty()) {
Workspaces own repositories447 conditions
448 .push("(name LIKE ? ESCAPE '\\' OR description LIKE ? ESCAPE '\\')".to_owned());
Rust repos service with shipping; pull requests kept in the model449 // LIKE wildcards in the query are matched literally.
450 let escaped: String = query
451 .chars()
452 .flat_map(|c| match c {
453 '\\' | '%' | '_' => vec!['\\', c],
454 _ => vec![c],
455 })
456 .collect();
457 let pattern = format!("%{escaped}%");
458 params.push(pattern.as_str().into());
459 params.push(pattern.into());
460 }
461 let sql = format!(
462 "SELECT * FROM repos WHERE {} ORDER BY created_at DESC, id DESC LIMIT 50",
463 conditions.join(" AND ")
464 );
465 let rows = self
466 .db
467 .prepare(sql)
468 .bind(&params)?
469 .all()
470 .await?
471 .results::<RepoRow>()?;
472 Ok(rows.into_iter().map(Repo::from).collect())
473 }
474
Agents and memory, checks and conflicts, profiles, slug renames, custom domains475 /// Of these ids, the repositories (not forks) the viewer may read.
476 pub async fn readable(&self, ids: &[String], viewer: &Viewer) -> Result<Vec<Repo>> {
477 let ids: Vec<&String> = ids.iter().take(g1t_contracts::repos::MAX_READABLE).collect();
478 if ids.is_empty() {
479 return Ok(Vec::new());
480 }
481 // One parameter however many ids: D1 binds at most 100.
482 let rows = self
483 .db
484 .prepare(
485 "SELECT * FROM repos
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look486 WHERE id IN (SELECT value FROM json_each(?)) AND fork_of IS NULL AND deleted_at IS NULL",
Agents and memory, checks and conflicts, profiles, slug renames, custom domains487 )
488 .bind(&[serde_json::to_string(&ids)?.into()])?
489 .all()
490 .await?
491 .results::<RepoRow>()?;
492 Ok(rows
493 .into_iter()
494 .map(Repo::from)
495 .filter(|repo| can_read(repo, viewer))
496 .collect())
497 }
498
499 /// The workspaces in which this account made a public repository.
500 pub async fn public_namespaces(&self, owner_id: &str) -> Result<Vec<String>> {
501 #[derive(Deserialize)]
502 struct Row {
503 namespace: String,
504 }
505 Ok(self
506 .db
507 .prepare(
508 "SELECT DISTINCT namespace FROM repos
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look509 WHERE owner_id = ? AND is_private = 0 AND fork_of IS NULL AND deleted_at IS NULL
Agents and memory, checks and conflicts, profiles, slug renames, custom domains510 ORDER BY namespace",
511 )
512 .bind(&[owner_id.into()])?
513 .all()
514 .await?
515 .results::<Row>()?
516 .into_iter()
517 .map(|row| row.namespace)
518 .collect())
519 }
520
Search across all of g1t, Explore, and a command palette521 /// Repositories that are not forks, by id, a page at a time.
Merge membership: owners, org roles, GitHub's repo roles, privileges, 2FA522 /// Repositories that are not forks, with who created each, by id after
523 /// `after`.
524 pub async fn creators_after(&self, after: Option<&str>, limit: u32) -> Result<Vec<g1t_contracts::repos::RepoCreator>> {
525 self.db
526 .prepare(
527 "SELECT id, namespace, name, owner_id FROM repos
528 WHERE fork_of IS NULL AND deleted_at IS NULL AND id > ? ORDER BY id LIMIT ?",
529 )
530 .bind(&[after.unwrap_or("").into(), limit.into()])?
531 .all()
532 .await?
533 .results::<g1t_contracts::repos::RepoCreator>()
534 }
535
Search across all of g1t, Explore, and a command palette536 pub async fn ids_after(&self, after: Option<&str>, limit: u32) -> Result<Vec<String>> {
537 #[derive(Deserialize)]
538 struct Row {
539 id: String,
540 }
541 Ok(self
542 .db
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look543 .prepare("SELECT id FROM repos WHERE fork_of IS NULL AND deleted_at IS NULL AND id > ? ORDER BY id LIMIT ?")
Search across all of g1t, Explore, and a command palette544 .bind(&[after.unwrap_or("").into(), limit.into()])?
545 .all()
546 .await?
547 .results::<Row>()?
548 .into_iter()
549 .map(|row| row.id)
550 .collect())
551 }
552
Team plan, an open-source pool, monthly trials and honest metering; the sidebar for everyone; a workspace that stays put553 /// Adds a pushed pack's bytes to what the repository is counted as
554 /// holding: its own, or, for a pull request's working copy, the
555 /// repository it is a copy of, whose storage it is.
556 pub async fn add_stored_bytes(&self, repo: &Repo, bytes: u64) -> Result<()> {
557 if bytes == 0 {
558 return Ok(());
559 }
560 let root = repo.fork_of.as_deref().unwrap_or(&repo.id);
561 self.db
562 .prepare("UPDATE repos SET stored_bytes = stored_bytes + ? WHERE id = ?")
563 .bind(&[(bytes as f64).into(), root.into()])?
564 .run()
565 .await?;
566 Ok(())
567 }
568
569 /// Which of these `namespace/name` paths are private. A working copy
570 /// answers as its repository. Unknown paths are left out.
571 pub async fn visibility(&self, paths: &[String]) -> Result<Vec<g1t_contracts::repos::RepoVisibility>> {
572 let mut out = Vec::new();
573 for path in paths.iter().take(50) {
574 let Some((namespace, name)) = path.split_once('/') else { continue };
575 let Some(repo) = self
576 .by_path(&RepoPath { namespace: namespace.to_owned(), name: name.to_owned() })
577 .await?
578 else {
579 continue;
580 };
581 let is_private = match &repo.fork_of {
582 Some(parent) => self.by_id(parent).await?.map_or(repo.is_private, |parent| parent.is_private),
583 None => repo.is_private,
584 };
585 out.push(g1t_contracts::repos::RepoVisibility { path: path.clone(), is_private });
586 }
587 Ok(out)
588 }
589
590 /// What each workspace's repositories are counted as holding, private
591 /// and public apart. Working copies count toward their repository.
592 pub async fn storage(&self) -> Result<Vec<g1t_contracts::repos::WorkspaceStorage>> {
593 #[derive(Deserialize)]
594 struct Row {
595 namespace: String,
596 private_bytes: Option<f64>,
597 public_bytes: Option<f64>,
598 }
599 Ok(self
600 .db
601 .prepare(
602 "SELECT namespace,
603 SUM(CASE WHEN is_private = 1 THEN stored_bytes ELSE 0 END) AS private_bytes,
604 SUM(CASE WHEN is_private = 0 THEN stored_bytes ELSE 0 END) AS public_bytes
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look605 FROM repos WHERE fork_of IS NULL AND deleted_at IS NULL AND stored_bytes > 0 GROUP BY namespace",
Team plan, an open-source pool, monthly trials and honest metering; the sidebar for everyone; a workspace that stays put606 )
607 .all()
608 .await?
609 .results::<Row>()?
610 .into_iter()
611 .map(|row| g1t_contracts::repos::WorkspaceStorage {
612 namespace: row.namespace,
613 private_bytes: row.private_bytes.unwrap_or(0.0) as i64,
614 public_bytes: row.public_bytes.unwrap_or(0.0) as i64,
615 })
616 .collect())
617 }
618
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look619 /// What one workspace's private repositories are counted as holding.
620 pub async fn private_bytes(&self, namespace: &str) -> Result<i64> {
621 #[derive(Deserialize)]
622 struct Row {
623 bytes: Option<f64>,
624 }
625 Ok(self
626 .db
627 .prepare("SELECT SUM(stored_bytes) AS bytes FROM repos WHERE namespace = ? AND is_private = 1 AND fork_of IS NULL AND deleted_at IS NULL")
628 .bind(&[namespace.into()])?
629 .first::<Row>(None)
630 .await?
631 .and_then(|row| row.bytes)
632 .unwrap_or(0.0) as i64)
633 }
634
Agents as a team: lifecycle, merge queue, billing and a new shell635 /// Forgets a repository that could not be filled.
636 pub async fn remove(&self, id: &str) -> Result<()> {
637 self.db
638 .prepare("DELETE FROM repos WHERE id = ?")
639 .bind(&[id.into()])?
640 .run()
641 .await?;
642 Ok(())
643 }
644
Agents and memory, checks and conflicts, profiles, slug renames, custom domains645 /// Picks the store key for a repository about to be made, and
646 /// remembers it: the one its path gives, unless a repository already
647 /// holds that (one made in a workspace that has since been renamed,
648 /// whose old name this workspace now has), when its id.
Git storage hardened, pages in tens of milliseconds, honest security alerts, and costs reconciled daily649 ///
650 /// `namespace` is the git store namespace it goes in (shards.rs), or
651 /// `None` for the default, `default`. A name is taken in any of them.
652 pub async fn claim_store_key(&self, repo: &Repo, namespace: Option<&str>, default: &str) -> Result<String> {
Agents and memory, checks and conflicts, profiles, slug renames, custom domains653 let wanted = path_key(repo);
654 let held = self
655 .db
Git storage hardened, pages in tens of milliseconds, honest security alerts, and costs reconciled daily656 .prepare(
657 "SELECT 1 AS held FROM repos
Merge branch 'worktree-agent-a2013627e5ea4ab13'658 WHERE store = ?1 OR (instr(store, '/') > 0 AND substr(store, instr(store, '/') + 1) = ?1)
659 UNION ALL
660 SELECT 1 AS held FROM repo_move_copies WHERE name = ?1 AND cleaned_ms IS NULL",
Git storage hardened, pages in tens of milliseconds, honest security alerts, and costs reconciled daily661 )
Agents and memory, checks and conflicts, profiles, slug renames, custom domains662 .bind(&[wanted.as_str().into()])?
663 .first::<serde_json::Value>(None)
664 .await?
665 .is_some();
Git storage hardened, pages in tens of milliseconds, honest security alerts, and costs reconciled daily666 let name = if held { repo.id.clone() } else { wanted };
667 let key = crate::shards::compose(namespace, &name, default);
Agents and memory, checks and conflicts, profiles, slug renames, custom domains668 remember_store(repo, &key);
669 Ok(key)
670 }
671
Git storage hardened, pages in tens of milliseconds, honest security alerts, and costs reconciled daily672
Agents and memory, checks and conflicts, profiles, slug renames, custom domains673 /// Moves a renamed workspace's repositories to its current slug, from
674 /// any of `stale`. A repository whose name the current slug already has
675 /// (one pushed there in the moment before this ran) stays where it is;
676 /// returns how many did.
677 pub async fn rename_namespace(&self, stale: &[String], current: &str) -> Result<usize> {
678 if stale.is_empty() {
679 return Ok(0);
680 }
681 let marks = vec!["?"; stale.len()].join(", ");
682 let mut moved: Vec<JsValue> = vec![current.into()];
683 moved.extend(stale.iter().map(|slug| JsValue::from(slug.as_str())));
684 let left: Vec<JsValue> = stale.iter().map(|slug| JsValue::from(slug.as_str())).collect();
685 let results = self
686 .db
687 .batch(vec![
688 self.db
689 .prepare(format!(
690 "UPDATE OR IGNORE repos SET namespace = ? WHERE namespace IN ({marks})"
691 ))
692 .bind(&moved)?,
693 self.db
694 .prepare(format!(
695 "SELECT count(*) AS left FROM repos WHERE namespace IN ({marks})"
696 ))
697 .bind(&left)?,
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look698 // Git operations follow the workspace, added together.
699 self.db
700 .prepare(format!(
701 "INSERT INTO git_operations (namespace, hour, operations)
702 SELECT ?, hour, SUM(operations) FROM git_operations WHERE namespace IN ({marks}) GROUP BY hour
703 ON CONFLICT (namespace, hour) DO UPDATE SET operations = git_operations.operations + excluded.operations"
704 ))
705 .bind(&moved)?,
706 self.db
707 .prepare(format!("DELETE FROM git_operations WHERE namespace IN ({marks})"))
708 .bind(&left)?,
709 // Paths repositories were transferred away from follow the
710 // workspace too, so the old slug's redirect then finds them.
711 self.db
712 .prepare(format!(
713 "UPDATE OR IGNORE repo_redirects SET namespace = ? WHERE namespace IN ({marks})"
714 ))
715 .bind(&moved)?,
Agents and memory, checks and conflicts, profiles, slug renames, custom domains716 ])
717 .await?;
718 #[derive(Deserialize)]
719 struct Left {
720 left: usize,
721 }
722 Ok(results
723 .get(1)
724 .map(|result| result.results::<Left>())
725 .transpose()?
726 .and_then(|rows| rows.into_iter().next())
727 .map_or(0, |row| row.left))
728 }
729
Mission control shows where you are needed and what agents landed without you; git answers in about 200ms730 /// Records that the refs of the repository with this id changed, after
731 /// they did: what anything that lists them keeps goes stale.
732 pub async fn refs_moved(&self, id: &str) -> Result<()> {
733 self.bump_refs(
734 "UPDATE repos SET refs_version = refs_version + 1 WHERE id = ?
735 RETURNING refs_version, refs_open_until",
736 &[id.into()],
737 id,
738 )
739 .await
740 }
741
742 /// Records that a credential able to change the refs of the repository
743 /// with this id was handed out of g1t's hands, until `until`
744 /// (milliseconds): until then, nothing that lists them is kept.
745 pub async fn refs_open(&self, id: &str, until: u64) -> Result<()> {
746 self.bump_refs(
747 "UPDATE repos SET refs_version = refs_version + 1,
748 refs_open_until = max(coalesce(refs_open_until, 0), ?)
749 WHERE id = ? RETURNING refs_version, refs_open_until",
750 &[(until as f64).into(), id.into()],
751 id,
752 )
753 .await
754 }
755
756 async fn bump_refs(&self, sql: &str, params: &[JsValue], id: &str) -> Result<()> {
757 #[derive(Deserialize)]
758 struct Bumped {
759 refs_version: Option<f64>,
760 refs_open_until: Option<f64>,
761 }
762 let bumped = self
763 .db
764 .prepare(sql)
765 .bind(params)?
766 .first::<Bumped>(None)
767 .await?;
768 if let Some(bumped) = bumped {
769 note_refs(id, bumped.refs_version, bumped.refs_open_until);
770 }
771 Ok(())
772 }
773
Rust repos service with shipping; pull requests kept in the model774 pub async fn insert(&self, repo: &Repo) -> Result<()> {
775 self.db
776 .prepare(
777 "INSERT INTO repos
778 (id, namespace, name, description, is_private, owner_id,
Agents and memory, checks and conflicts, profiles, slug renames, custom domains779 default_branch, fork_of, created_at, store)
780 VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)",
Rust repos service with shipping; pull requests kept in the model781 )
782 .bind(&[
783 repo.id.as_str().into(),
784 repo.namespace.as_str().into(),
785 repo.name.as_str().into(),
786 optional(&repo.description),
787 (repo.is_private as u8).into(),
788 repo.owner_id.as_str().into(),
789 repo.default_branch.as_str().into(),
790 optional(&repo.fork_of),
RFC 3339 timestamps in identity and repos791 repo.created_at.as_str().into(),
Agents and memory, checks and conflicts, profiles, slug renames, custom domains792 store_key(repo).into(),
Rust repos service with shipping; pull requests kept in the model793 ])?
794 .run()
795 .await?;
796 Ok(())
797 }
798}
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look799
800#[cfg(test)]
801mod tests {
802 use super::*;
803 use g1t_contracts::access::{BasePermission, RepoGrant};
804 use g1t_contracts::{Membership, Role, User};
805
Mission control shows where you are needed and what agents landed without you; git answers in about 200ms806 #[test]
807 fn the_refs_state_kept_only_moves_forward() {
808 let mut states = RefsStates::default();
809 assert_eq!(states.get("rep_1"), None);
810 states.note("rep_1", RefsState { version: 3, open_until: 0 });
811 // A read that started before a bump and finished after it.
812 states.note("rep_1", RefsState { version: 2, open_until: 0 });
813 assert_eq!(states.get("rep_1").unwrap().version, 3);
814 states.note("rep_1", RefsState { version: 4, open_until: 9_000 });
815 states.note("rep_1", RefsState { version: 5, open_until: 0 });
816 assert_eq!(states.get("rep_1"), Some(RefsState { version: 5, open_until: 9_000 }));
817 assert_eq!(states.get("rep_2"), None);
818 }
819
820 #[test]
821 fn a_row_from_before_the_column_has_no_refs_state() {
822 let row = |version: Option<f64>| RepoRow {
823 id: format!("rep_row_{}", version.is_some()),
824 namespace: "acme".into(),
825 name: "rocket".into(),
826 description: None,
827 is_private: 0,
828 owner_id: "usr_owner".into(),
829 default_branch: "main".into(),
830 fork_of: None,
831 protected: 0,
832 created_at: String::new(),
833 store: None,
834 topics: None,
835 website: None,
836 archived_at: None,
837 deleted_at: None,
838 refs_version: version,
839 refs_open_until: None,
Git storage hardened, pages in tens of milliseconds, honest security alerts, and costs reconciled daily840 retired_at: None,
841 retired_head: None,
Merge branch 'worktree-agent-a2013627e5ea4ab13'842 writes_paused_until: None,
843 writes_paused_for: None,
Mission control shows where you are needed and what agents landed without you; git answers in about 200ms844 };
845 let old = Repo::from(row(None));
846 assert_eq!(refs_state(&old.id), None);
847 let new = Repo::from(row(Some(7.0)));
848 assert_eq!(refs_state(&new.id), Some(RefsState { version: 7, open_until: 0 }));
849 }
850
851 #[test]
Git storage hardened, pages in tens of milliseconds, honest security alerts, and costs reconciled daily852 fn a_removed_working_copy_is_known_by_its_row() {
853 note_retired("rep_fork", Some("abc"));
854 assert_eq!(retired("rep_fork").as_deref(), Some("abc"));
855 note_retired("rep_fork", None);
856 assert_eq!(retired("rep_fork"), None);
857 }
858
859 #[test]
Mission control shows where you are needed and what agents landed without you; git answers in about 200ms860 fn a_repository_read_a_moment_ago_is_reused_for_a_few_seconds() {
861 let mut recent = Recent::default();
862 let path = RepoPath {
863 namespace: "Acme".into(),
864 name: "Rocket".into(),
865 };
866 recent.keep(&path, &repo(false), 1_000);
867 // Paths are matched as the table matches them, ignoring case.
868 let lower = RepoPath {
869 namespace: "acme".into(),
870 name: "rocket".into(),
871 };
872 assert_eq!(recent.get(&lower, 1_000 + RECENT_MS - 1).unwrap().id, "rep_1");
873 assert!(recent.get(&lower, 1_000 + RECENT_MS).is_none());
874 let other = RepoPath {
875 namespace: "acme".into(),
876 name: "booster".into(),
877 };
878 assert!(recent.get(&other, 1_000).is_none());
879 // Keeping another later drops the stale row.
880 recent.keep(&other, &repo(true), 1_000 + RECENT_MS);
881 assert_eq!(recent.rows.len(), 1);
882 }
883
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look884 fn repo(private: bool) -> Repo {
885 Repo {
886 id: "rep_1".into(),
887 namespace: "acme".into(),
888 name: "rocket".into(),
889 description: None,
890 is_private: private,
891 owner_id: "usr_owner".into(),
892 default_branch: "main".into(),
893 fork_of: None,
894 protected: false,
895 created_at: String::new(),
896 topics: Vec::new(),
897 website: None,
898 archived_at: None,
899 }
900 }
901
902 fn person(id: &str, memberships: Vec<Membership>, grants: Vec<(&str, RepoRole)>) -> Viewer {
903 Some(User {
904 id: id.into(),
905 username: id.into(),
906 verified: true,
907 workspaces: memberships,
908 grants: grants
909 .into_iter()
Teams and CODEOWNERS, labels and milestones, dependency updates, the security suite, and a clearer top bar910 .map(|(repo_id, role)| RepoGrant { repo_id: repo_id.into(), workspace: "acme".into(), role, team: None })
Invite-only launch: sign in with GitHub, repository access and lifecycle, many emails, a new look911 .collect(),
912 ..User::default()
913 })
914 }
915
916 /// What git asks: clone and fetch need Read on a private repository,
917 /// push needs Write.
918 #[test]
919 fn git_reads_with_read_and_pushes_with_write() {
920 let private = repo(true);
921 let reader = person("usr_r", vec![], vec![("rep_1", RepoRole::Read)]);
922 assert!(can_read(&private, &reader));
923 assert!(!can_write(&private, &reader));
924 let writer = person("usr_w", vec![], vec![("rep_1", RepoRole::Write)]);
925 assert!(can_read(&private, &writer) && can_write(&private, &writer));
926 let stranger = person("usr_s", vec![], vec![("rep_2", RepoRole::Admin)]);
927 assert!(!can_read(&private, &stranger) && !can_write(&private, &stranger));
928 assert!(!can_read(&private, &None));
929 // A public repository: anyone clones, nobody without Write pushes.
930 let public = repo(false);
931 assert!(can_read(&public, &None) && !can_write(&public, &None));
932 assert!(can_read(&public, &stranger) && !can_write(&public, &stranger));
933 }
934
935 #[test]
936 fn members_follow_the_base_permission_and_owners_have_admin() {
937 let private = repo(true);
938 let default_member = person("usr_m", vec![Membership::member("acme")], vec![]);
939 assert!(can_write(&private, &default_member));
940 assert!(!can(&private, &default_member, Capability::ManageIntegrations));
941 let none = Membership { base_permission: Some(BasePermission::None), ..Membership::member("acme") };
942 let locked_out = person("usr_n", vec![none.clone()], vec![]);
943 assert!(!can_read(&private, &locked_out));
944 let given = person("usr_g", vec![none], vec![("rep_1", RepoRole::Triage)]);
945 assert!(can_read(&private, &given) && !can_write(&private, &given));
946 let owner = person("usr_o", vec![Membership { role: Role::Owner, ..Membership::member("acme") }], vec![]);
947 assert_eq!(role(&private, &owner), Some(RepoRole::Admin));
948 assert!(can(&private, &owner, Capability::Delete));
949 }
950
951 #[test]
952 fn a_pull_requests_fork_is_its_authors() {
953 let fork = Repo {
954 namespace: "pulls".into(),
955 fork_of: Some("rep_1".into()),
956 owner_id: "usr_a".into(),
957 ..repo(true)
958 };
959 let author = person("usr_a", vec![], vec![]);
960 assert!(can_write(&fork, &author));
961 assert!(!can(&fork, &author, Capability::ManageSettings));
962 let other = person("usr_b", vec![Membership::member("acme")], vec![]);
963 assert!(!can_write(&fork, &other));
964 }
965}

This file's history is long; its oldest lines are credited to the oldest commit read.