╭ patch.zig ────────────────────────────────────────────────────╮ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ╰───────────────────────────────────────────────────────────────╯
╭────╮ ╭ clone ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ╮ │http│ ssh │ │ ╰────╯ ╰ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ╯
1 const std = @import("std"); 2 const evt = @import("event.zig"); 3 const xit = @import("xit"); 4 const rp = xit.repo; 5 const hash = xit.hash; 6 const obj = xit.object; 7 const rf = xit.ref; 8 const mrg = xit.merge; 9 const find = @import("find.zig"); 10 const srch_cmmt = @import("search_commit.zig"); 11 const fork = @import("fork.zig"); 12 const serve_common = @import("serve_common.zig"); 13 const progress = @import("./progress.zig"); 14 15 // the action a patch merge records 16 pub const merge_undo_action = "haxy/merge"; 17 18 // the action a mergeability recheck records 19 pub const mergeability_undo_action = "haxy/mergeability"; 20 21 // whether a push moved this branch 22 fn branchUpdated(updates: []const xit.net_server_receive_pack.AppliedRefUpdate, branch: []const u8) bool { 23 const prefix = "refs/heads/"; 24 for (updates) |update| { 25 if (std.mem.startsWith(u8, update.ref_name, prefix) and std.mem.eql(u8, update.ref_name[prefix.len..], branch)) return true; 26 } 27 return false; 28 } 29 30 // capture an existing branch without creating a fork 31 pub fn writeBranchPatch( 32 host_kind: evt.HostKind, 33 comptime repo_kind: rp.RepoKind, 34 comptime repo_opts: rp.RepoOpts(repo_kind), 35 io: std.Io, 36 allocator: std.mem.Allocator, 37 repo: *rp.Repo(repo_kind, repo_opts), 38 id: [evt.event_id_size * 2]u8, 39 patch: evt.Patch, 40 expected_patch: ?evt.Patch, 41 author: evt.CommitAuthor, 42 ) !void { 43 var arena = std.heap.ArenaAllocator.init(allocator); 44 defer arena.deinit(); 45 46 // write git events through the local event database 47 if (repo_kind == .git) { 48 // reject edits based on an outdated patch 49 if (expected_patch) |expected| { 50 const current = (try evt.readFromRepo(evt.Patch, repo_kind, repo_opts, io, allocator, &arena, repo, &try evt.parseEventId(&id))) orelse return error.PatchOutOfDate; 51 if (current.removed or !evt.fieldEqual(evt.Patch, current.event, expected)) return error.PatchOutOfDate; 52 } 53 54 // capture the branch revision and consume its events 55 const events = try branchEvents(repo_kind, repo_opts, .{ .core = &repo.core, .extra = .{} }, io, &arena, id, patch, author); 56 try evt.consume(host_kind, .repo, repo_kind, repo_opts, io, allocator, repo, evt.events_ref, &events); 57 } else { 58 const DB = evt.EventDB(repo_opts.hash); 59 60 const first_parent = try eventsFirstParent(repo_kind, repo_opts, io, allocator, repo); 61 62 // commit the patch, mergeability, and undo record together 63 { 64 const Ctx = struct { 65 repo: *rp.Repo(repo_kind, repo_opts), 66 io: std.Io, 67 arena: *std.heap.ArenaAllocator, 68 id: [evt.event_id_size * 2]u8, 69 patch: evt.Patch, 70 expected_patch: ?evt.Patch, 71 author: evt.CommitAuthor, 72 checks: *MergeCheckInputs(repo_opts), 73 first_parent: ?[1][hash.hexLen(repo_opts.hash)]u8, 74 75 pub fn run(ctx: @This(), cursor: *DB.Cursor(.read_write)) !void { 76 var moment = try DB.HashMap(.read_write).init(cursor.*); 77 const state = rp.Repo(repo_kind, repo_opts).State(.read_write){ .core = &ctx.repo.core, .extra = .{ .moment = &moment } }; 78 79 try writeBranchPatchInTransaction(repo_opts, state, &moment, ctx.io, ctx.arena, ctx.id, ctx.patch, ctx.expected_patch, ctx.author, ctx.first_parent, ctx.checks); 80 81 // record the user action for undo 82 try xit.undo.write(repo_opts, state, std.Io.Timestamp.now(ctx.io, .real).toSeconds(), .{ .custom = .{ .action_kind = evt.undo_action } }); 83 } 84 }; 85 86 // a branch patch has no fork, so this only names the patch to check 87 var checks: MergeCheckInputs(repo_opts) = .{}; 88 defer checks.deinit(io, allocator); 89 if (host_kind == .server) try checks.add(io, allocator, &repo.core, try evt.parseEventId(&id), patch); 90 91 try repo.core.db_file.lock(io, .exclusive); 92 defer repo.core.db_file.unlock(io); 93 94 const history = try DB.ArrayList(.read_write).init(repo.core.db.rootCursor()); 95 try history.appendContext(.{ .slot = try history.getSlot(-1) }, Ctx{ .repo = repo, .io = io, .arena = &arena, .id = id, .patch = patch, .expected_patch = expected_patch, .author = author, .checks = &checks, .first_parent = first_parent }); 96 } 97 } 98 } 99 100 // write one branch patch's revision into a transaction the caller owns. the 101 // undo record is the caller's to write, since the transaction may be theirs. 102 // git repos never reach here: they write through the local event database 103 pub fn writeBranchPatchInTransaction( 104 comptime repo_opts: rp.RepoOpts(.xit), 105 state: rp.Repo(.xit, repo_opts).State(.read_write), 106 moment: *evt.EventDB(repo_opts.hash).HashMap(.read_write), 107 io: std.Io, 108 arena: *std.heap.ArenaAllocator, 109 id: [evt.event_id_size * 2]u8, 110 patch: evt.Patch, 111 expected_patch: ?evt.Patch, 112 author: evt.CommitAuthor, 113 first_parent: ?[1][hash.hexLen(repo_opts.hash)]u8, 114 checks: *MergeCheckInputs(repo_opts), 115 ) !void { 116 const DB = evt.EventDB(repo_opts.hash); 117 118 // recheck the expected patch under the repository lock 119 if (expected_patch) |expected| { 120 const events_moment = try evt.currentMomentFromRepoMoment(repo_opts.hash, moment.readOnly()); 121 const current = (try evt.Patch.readById(DB, repo_opts.hash, events_moment, arena, &try evt.parseEventId(&id))) orelse return error.PatchOutOfDate; 122 if (current.removed or !evt.fieldEqual(evt.Patch, current.event, expected)) return error.PatchOutOfDate; 123 } 124 125 // capture and consume the branch revision 126 const events = try branchEvents(.xit, repo_opts, state, io, arena, id, patch, author); 127 { 128 const parent = if (try rf.readRecur(.xit, repo_opts, state.readOnly(), io, .{ .ref = evt.events_ref }) == null) first_parent else null; 129 try evt.commitEvents(.xit, repo_opts, state, io, arena.child_allocator, evt.events_ref, &events, parent); 130 if (!try evt.consumeInTransaction(.repo, .xit, repo_opts, state, &state.core.db, moment, io, arena.child_allocator, evt.events_ref)) return error.CancelTransaction; 131 } 132 133 // save mergeability before finishing the patch transaction 134 _ = try checks.update(state, io, arena.child_allocator); 135 } 136 137 // continue from a remote event history when the local ref is absent, so a 138 // fresh clone's first event does not start a history of its own 139 fn eventsFirstParent( 140 comptime repo_kind: rp.RepoKind, 141 comptime repo_opts: rp.RepoOpts(repo_kind), 142 io: std.Io, 143 allocator: std.mem.Allocator, 144 repo: *rp.Repo(repo_kind, repo_opts), 145 ) !?[1][hash.hexLen(repo_opts.hash)]u8 { 146 if (try repo.readRef(io, evt.events_ref) != null) return null; 147 var remotes = try repo.listRemotes(io, allocator); 148 defer remotes.deinit(); 149 for (remotes.sections.keys()) |name| { 150 if (try repo.readRef(io, .{ .kind = .{ .remote = name }, .name = evt.events_ref.name })) |oid| return .{oid}; 151 } 152 return null; 153 } 154 155 fn branchEvents( 156 comptime repo_kind: rp.RepoKind, 157 comptime repo_opts: rp.RepoOpts(repo_kind), 158 state: rp.Repo(repo_kind, repo_opts).State(.read_write), 159 io: std.Io, 160 arena: *std.heap.ArenaAllocator, 161 id: [evt.event_id_size * 2]u8, 162 patch: evt.Patch, 163 author: evt.CommitAuthor, 164 ) ![2]evt.EventWithId { 165 // validate the patch fields and branch names 166 const branch = patch.source_branch orelse return error.InvalidSourceBranch; 167 if (!evt.Patch.fieldsValid(patch.title, patch.labels)) return error.InvalidFields; 168 if (!evt.Patch.branchValid(branch)) return error.InvalidSourceBranch; 169 if (!evt.Patch.branchValid(patch.target_branch)) return error.InvalidTargetBranch; 170 171 // resolve both tips and their common ancestor 172 const source = (try rf.readRecur(repo_kind, repo_opts, state.readOnly(), io, .{ .ref = .{ .kind = .head, .name = branch } })) orelse return error.InvalidSourceBranch; 173 const target = (try rf.readRecur(repo_kind, repo_opts, state.readOnly(), io, .{ .ref = .{ .kind = .head, .name = patch.target_branch } })) orelse return error.InvalidTargetBranch; 174 if (std.mem.eql(u8, branch, patch.target_branch)) return error.SameBranch; 175 const base = mrg.commonAncestor(repo_kind, repo_opts, state.readOnly(), io, arena.allocator(), &target, &source) catch |err| switch (err) { 176 error.NoCommonAncestor => return error.UnrelatedBranches, 177 else => return err, 178 }; 179 180 // prepare the revision and attach it to the patch event 181 const timestamp: u64 = @intCast(std.Io.Timestamp.now(io, .real).toSeconds()); 182 const prepared = try evt.PatchRev.prepare(repo_kind, repo_opts, state, io, arena, &base, &source, patch.title, author, timestamp); 183 var updated = patch; 184 updated.revision = prepared.revision; 185 return .{ 186 prepared.event, 187 .{ .id = id, .author = author, .timestamp = timestamp, .event = .{ .patch = updated } }, 188 }; 189 } 190 191 // the open patches whose source branch has moved past the revision they 192 // stored, mapped to the record the refresh rewrites. `updates` limits the 193 // search to the branches one push changed 194 fn collectOutdatedBranchPatches( 195 comptime repo_kind: rp.RepoKind, 196 comptime repo_opts: rp.RepoOpts(repo_kind), 197 io: std.Io, 198 allocator: std.mem.Allocator, 199 arena: *std.heap.ArenaAllocator, 200 state: rp.Repo(repo_kind, repo_opts).State(.read_only), 201 moment: evt.EventDB(repo_opts.hash).HashMap(.read_only), 202 updates: ?[]const xit.net_server_receive_pack.AppliedRefUpdate, 203 host_kind: evt.HostKind, 204 patches: *std.AutoArrayHashMapUnmanaged([evt.event_id_size]u8, evt.Patch.Record), 205 ) !void { 206 const DB = evt.EventDB(repo_opts.hash); 207 // find tracked source branches and skip conflicting patches 208 const source_cursor = try moment.getCursor(hash.hashInt(repo_opts.hash, evt.Patch.source_to_id_set_key)) orelse return; 209 const sources = try DB.SortedMap(.read_only).init(source_cursor); 210 const conflicts = if (try moment.getCursor(hash.hashInt(repo_opts.hash, evt.Patch.conflicts_key))) |cursor| try DB.SortedMap(.read_only).init(cursor) else null; 211 var iter = try sources.iterator(); 212 while (try iter.next()) |cursor| { 213 const pair = try cursor.readKeyValuePair(); 214 const branch = try pair.key_cursor.readBytesAlloc(arena.allocator(), null); 215 216 // limit source refreshes to branches changed by this push 217 if (updates) |changed| if (!branchUpdated(changed, branch)) continue; 218 219 // read the source tip before comparing stored revisions 220 const source = (try rf.readRecur(repo_kind, repo_opts, state, io, .{ .ref = .{ .kind = .head, .name = branch } })) orelse { 221 if (host_kind == .server) std.log.warn("patch source branch missing: {s}", .{branch}); 222 continue; 223 }; 224 225 // collect open patches whose stored source tip is outdated 226 const ids = try DB.CountedHashSet(.read_only).init(pair.value_cursor); 227 var ids_iter = try ids.iterator(); 228 while (try ids_iter.next()) |id_cursor| { 229 const id_pair = try id_cursor.readKeyValuePair(); 230 var id: [evt.event_id_size]u8 = undefined; 231 if ((try id_pair.key_cursor.readBytes(&id)).len != id.len) return error.InvalidPatch; 232 const record = (try evt.Patch.readById(DB, repo_opts.hash, moment, arena, &id)) orelse continue; 233 if (record.removed or record.event.status.kind() != .open) continue; 234 if (conflicts) |map| if (try map.getCursor(&evt.orderKeyDesc(record.created_order, &id)) != null) continue; 235 if (record.event.revision) |revision| if (std.mem.eql(u8, revision.source_oid, &source)) continue; 236 try patches.put(allocator, id, record); 237 } 238 } 239 } 240 241 // refresh the patches their source branch has outrun, inside the caller's 242 // transaction, so a push and the revisions it causes land together. reports 243 // whether any revision was written, so a caller that owns an otherwise empty 244 // transaction can cancel it. mergeability is only saved for branch patches: 245 // a fork's lock has to be taken before this transaction opened 246 pub fn refreshBranchesInTransaction( 247 host_kind: evt.HostKind, 248 comptime repo_opts: rp.RepoOpts(.xit), 249 state: rp.Repo(.xit, repo_opts).State(.read_write), 250 moment: *evt.EventDB(repo_opts.hash).HashMap(.read_write), 251 io: std.Io, 252 allocator: std.mem.Allocator, 253 updates: ?[]const xit.net_server_receive_pack.AppliedRefUpdate, 254 first_parent: ?[1][hash.hexLen(repo_opts.hash)]u8, 255 progress_ctx_maybe: ?repo_opts.ProgressCtx, 256 ) !bool { 257 var arena = std.heap.ArenaAllocator.init(allocator); 258 defer arena.deinit(); 259 var patches: std.AutoArrayHashMapUnmanaged([evt.event_id_size]u8, evt.Patch.Record) = .empty; 260 defer patches.deinit(allocator); 261 262 const events_moment = evt.currentMomentFromRepoMoment(repo_opts.hash, moment.readOnly()) catch |err| switch (err) { 263 error.NotFound => return false, 264 else => return err, 265 }; 266 try collectOutdatedBranchPatches(.xit, repo_opts, io, allocator, &arena, state.readOnly(), events_moment, updates, host_kind, &patches); 267 268 const total = patches.count(); 269 if (total > 0) { 270 progress.report(repo_opts, io, progress_ctx_maybe, .{ .text = "Updating patches" }); 271 progress.report(repo_opts, io, progress_ctx_maybe, .{ .start = .{ .kind = .writing_patch, .estimated_total_items = total } }); 272 } 273 274 var refreshed = false; 275 for (patches.keys(), patches.values()) |id, record| { 276 defer progress.report(repo_opts, io, progress_ctx_maybe, .{ .complete_one = .writing_patch }); 277 278 // a branch patch opens no fork, so its checks need no lock of their own 279 var checks: MergeCheckInputs(repo_opts) = .{}; 280 defer checks.deinit(io, allocator); 281 if (host_kind == .server) try checks.add(io, allocator, state.core, id, record.event); 282 283 const author = evt.CommitAuthor{ .name = "haxy", .email = record.author_email orelse "user@haxy" }; 284 writeBranchPatchInTransaction(repo_opts, state, moment, io, &arena, std.fmt.bytesToHex(id, .lower), record.event, record.event, author, first_parent, &checks) catch |err| { 285 std.log.warn("failed to refresh branch patch: {s}", .{@errorName(err)}); 286 continue; 287 }; 288 refreshed = true; 289 } 290 if (total > 0) progress.report(repo_opts, io, progress_ctx_maybe, .{ .end = .writing_patch }); 291 return refreshed; 292 } 293 294 // refresh open patches' tracked branches and, on the server, their 295 // mergeability. this runs after the caller's transaction, and on xit writes 296 // every revision in one transaction of its own 297 pub fn refreshBranches( 298 host_kind: evt.HostKind, 299 comptime repo_kind: rp.RepoKind, 300 comptime repo_opts: rp.RepoOpts(repo_kind), 301 io: std.Io, 302 allocator: std.mem.Allocator, 303 repo: *rp.Repo(repo_kind, repo_opts), 304 updates: ?[]const xit.net_server_receive_pack.AppliedRefUpdate, 305 progress_ctx_maybe: ?repo_opts.ProgressCtx, 306 ) !void { 307 if (repo_kind == .xit) { 308 const DB = evt.EventDB(repo_opts.hash); 309 310 // every revision shares one transaction, so the refresh is a single 311 // undo entry. the transaction collects the patches itself, and is 312 // cancelled when none of them moved 313 { 314 const Ctx = struct { 315 core: *rp.Repo(.xit, repo_opts).Core, 316 io: std.Io, 317 allocator: std.mem.Allocator, 318 host_kind: evt.HostKind, 319 updates: ?[]const xit.net_server_receive_pack.AppliedRefUpdate, 320 first_parent: ?[1][hash.hexLen(repo_opts.hash)]u8, 321 progress_ctx_maybe: ?repo_opts.ProgressCtx, 322 323 pub fn run(ctx: @This(), cursor: *DB.Cursor(.read_write)) !void { 324 var moment = try DB.HashMap(.read_write).init(cursor.*); 325 const state = rp.Repo(.xit, repo_opts).State(.read_write){ .core = ctx.core, .extra = .{ .moment = &moment } }; 326 if (!try refreshBranchesInTransaction(ctx.host_kind, repo_opts, state, &moment, ctx.io, ctx.allocator, ctx.updates, ctx.first_parent, ctx.progress_ctx_maybe)) return error.CancelTransaction; 327 328 // record the user action for undo 329 try xit.undo.write(repo_opts, state, std.Io.Timestamp.now(ctx.io, .real).toSeconds(), .{ .custom = .{ .action_kind = evt.undo_action } }); 330 } 331 }; 332 333 const first_parent = try eventsFirstParent(repo_kind, repo_opts, io, allocator, repo); 334 335 try repo.core.db_file.lock(io, .exclusive); 336 defer repo.core.db_file.unlock(io); 337 338 const history = try DB.ArrayList(.read_write).init(repo.core.db.rootCursor()); 339 history.appendContext(.{ .slot = try history.getSlot(-1) }, Ctx{ 340 .core = &repo.core, 341 .io = io, 342 .allocator = allocator, 343 .host_kind = host_kind, 344 .updates = updates, 345 .first_parent = first_parent, 346 .progress_ctx_maybe = progress_ctx_maybe, 347 }) catch |err| switch (err) { 348 error.CancelTransaction => {}, 349 else => return err, 350 }; 351 } 352 353 // every open patch is checked after the transaction, since a fork's 354 // lock has to be taken before the target's 355 if (host_kind == .server) refreshOpenMergeability(repo_opts, io, allocator, repo, updates, progress_ctx_maybe); 356 return; 357 } 358 359 // git repos have no transaction to share, so each revision is written on 360 // its own through the local event database. they report no progress, 361 // since only a push has a progress context to report to 362 var arena = std.heap.ArenaAllocator.init(allocator); 363 defer arena.deinit(); 364 var patches: std.AutoArrayHashMapUnmanaged([evt.event_id_size]u8, evt.Patch.Record) = .empty; 365 defer patches.deinit(allocator); 366 367 collect: { 368 var local_db = (try evt.LocalEventDB(repo_opts.hash).openReadOnly(io, allocator, repo.core.repo_dir)) orelse break :collect; 369 defer local_db.deinit(io, allocator); 370 const moment = evt.currentMomentFromDb(repo_opts.hash, local_db.db) catch |err| switch (err) { 371 error.NotFound => break :collect, 372 else => return err, 373 }; 374 try collectOutdatedBranchPatches(repo_kind, repo_opts, io, allocator, &arena, .{ .core = &repo.core, .extra = .{} }, moment, updates, host_kind, &patches); 375 } 376 377 for (patches.keys(), patches.values()) |id, record| { 378 const author = evt.CommitAuthor{ .name = "haxy", .email = record.author_email orelse "user@haxy" }; 379 writeBranchPatch(host_kind, repo_kind, repo_opts, io, allocator, repo, std.fmt.bytesToHex(id, .lower), record.event, record.event, author) catch |err| { 380 if (host_kind == .server) std.log.warn("failed to refresh branch patch: {s}", .{@errorName(err)}); 381 }; 382 } 383 } 384 385 pub const PublishInput = struct { 386 id: [evt.event_id_size * 2]u8, 387 user_id: [evt.event_id_size]u8, 388 repo_id: [evt.event_id_size]u8, 389 author: evt.CommitAuthor, 390 timestamp: u64, 391 }; 392 393 pub const EditDraftInput = struct { 394 id: [evt.event_id_size * 2]u8, 395 user_id: [evt.event_id_size]u8, 396 repo_id: [evt.event_id_size]u8, 397 title: []const u8, 398 labels: []const u8, 399 description: []const u8, 400 target_branch: []const u8, 401 author: evt.CommitAuthor, 402 timestamp: u64, 403 }; 404 405 pub const Mergeability = struct { 406 pub const Status = enum { clean, conflict, unknown }; 407 408 source: Status = .unknown, 409 squash: Status = .unknown, 410 411 pub fn get(self: Mergeability, revision: evt.Patch.MergeRevision) Status { 412 return switch (revision) { 413 .source => self.source, 414 .squash => self.squash, 415 }; 416 } 417 418 pub fn status(self: Mergeability) Status { 419 if (self.source == .clean or self.squash == .clean) return .clean; 420 if (self.source == .conflict and self.squash == .conflict) return .conflict; 421 return .unknown; 422 } 423 }; 424 425 const MergeCheck = struct { 426 revision: evt.Patch.Revision, 427 target_branch: []const u8, 428 target_oid: []const u8, 429 result: Mergeability = .{}, 430 431 fn matches(self: MergeCheck, patch: evt.Patch, target_oid: []const u8) bool { 432 return evt.fieldEqual(?evt.Patch.Revision, self.revision, patch.revision) and 433 std.mem.eql(u8, self.target_branch, patch.target_branch) and 434 std.mem.eql(u8, self.target_oid, target_oid); 435 } 436 }; 437 438 fn readMergeCheck( 439 comptime repo_opts: rp.RepoOpts(.xit), 440 haxy_moment: evt.EventDB(repo_opts.hash).HashMap(.read_only), 441 arena: *std.heap.ArenaAllocator, 442 id: *const [evt.event_id_size]u8, 443 ) !?MergeCheck { 444 const DB = evt.EventDB(repo_opts.hash); 445 446 // find the cached check for this patch 447 const map_cursor = try haxy_moment.getCursor(hash.hashInt(repo_opts.hash, evt.Patch.patch_id_to_mergeability_key)) orelse return null; 448 const map = try DB.HashMap(.read_only).init(map_cursor); 449 const cursor = try map.getCursor(hash.hashInt(repo_opts.hash, id)) orelse return null; 450 451 // treat unrecognized cached statuses as a cache miss 452 return evt.read(MergeCheck, DB, repo_opts.hash, arena, try DB.HashMap(.read_only).init(cursor)) catch |err| switch (err) { 453 error.InvalidEnumTag => return null, 454 else => return err, 455 }; 456 } 457 458 // reads never run a merge or refresh the cache 459 pub fn readMergeability( 460 comptime repo_opts: rp.RepoOpts(.xit), 461 state: rp.Repo(.xit, repo_opts).State(.read_only), 462 io: std.Io, 463 arena: *std.heap.ArenaAllocator, 464 id: *const [evt.event_id_size]u8, 465 patch: evt.Patch, 466 ) !Mergeability { 467 // reject checks for an outdated source revision 468 if (patch.source_branch != null and !try branchCurrent(repo_opts, state, io, arena, patch)) return .{}; 469 470 // use the cached result only when the target still matches 471 const target_oid = try rf.readRecur(.xit, repo_opts, state, io, .{ .ref = .{ .kind = .head, .name = patch.target_branch } }) orelse return .{}; 472 const haxy_moment = evt.currentMomentFromRepoMoment(repo_opts.hash, state.extra.moment.*) catch |err| switch (err) { 473 error.NotFound => return .{}, 474 else => return err, 475 }; 476 const cached = try readMergeCheck(repo_opts, haxy_moment, arena, id) orelse return .{}; 477 return if (cached.matches(patch, &target_oid)) cached.result else .{}; 478 } 479 480 // refresh after the caller's transaction and locks have finished. a check 481 // failure must not turn an accepted push or event into a failed operation. 482 pub fn refreshMergeability( 483 comptime repo_opts: rp.RepoOpts(.xit), 484 io: std.Io, 485 allocator: std.mem.Allocator, 486 target_repo: *rp.Repo(.xit, repo_opts), 487 id: [evt.event_id_size]u8, 488 ) void { 489 refreshMergeCheck(repo_opts, io, allocator, target_repo, &id) catch |err| { 490 std.log.warn("failed to refresh mergeability: {s}", .{@errorName(err)}); 491 }; 492 } 493 494 // check the open patches a push could have invalidated, reporting progress as 495 // each one finishes. `updates` limits them to the branches it moved 496 pub fn refreshOpenMergeability( 497 comptime repo_opts: rp.RepoOpts(.xit), 498 io: std.Io, 499 allocator: std.mem.Allocator, 500 target_repo: *rp.Repo(.xit, repo_opts), 501 updates: ?[]const xit.net_server_receive_pack.AppliedRefUpdate, 502 progress_ctx_maybe: ?repo_opts.ProgressCtx, 503 ) void { 504 refreshOpenMergeChecks(repo_opts, io, allocator, target_repo, updates, progress_ctx_maybe) catch |err| { 505 std.log.warn("failed to refresh mergeability: {s}", .{@errorName(err)}); 506 }; 507 } 508 509 fn refreshOpenMergeChecks( 510 comptime repo_opts: rp.RepoOpts(.xit), 511 io: std.Io, 512 allocator: std.mem.Allocator, 513 target_repo: *rp.Repo(.xit, repo_opts), 514 updates: ?[]const xit.net_server_receive_pack.AppliedRefUpdate, 515 progress_ctx_maybe: ?repo_opts.ProgressCtx, 516 ) !void { 517 // collect open patch ids before any refresh changes the database 518 const DB = evt.EventDB(repo_opts.hash); 519 var arena = std.heap.ArenaAllocator.init(allocator); 520 defer arena.deinit(); 521 var ids: std.ArrayList([evt.event_id_size]u8) = .empty; 522 defer ids.deinit(allocator); 523 collect: { 524 const moment = evt.currentMoment(repo_opts, target_repo) catch |err| switch (err) { 525 error.NotFound => break :collect, 526 else => return err, 527 }; 528 const statuses_cursor = try moment.getCursor(hash.hashInt(repo_opts.hash, evt.Patch.status_to_id_set_key)) orelse break :collect; 529 const statuses = try DB.SortedMap(.read_only).init(statuses_cursor); 530 const open_cursor = try statuses.getCursor("open") orelse break :collect; 531 const open = try DB.SortedSet(.read_only).init(open_cursor); 532 var iter = try open.iteratorFromIndex(0); 533 while (try iter.next()) |cursor| { 534 const id = try evt.readOrderKeyId(DB, cursor); 535 536 // a saved check only goes stale when the patch's revision or its 537 // target tip moves, so a push can skip the branches it left alone 538 if (updates) |changed| { 539 const record = (try evt.Patch.readById(DB, repo_opts.hash, moment, &arena, &id)) orelse continue; 540 const source_moved = if (record.event.source_branch) |branch| branchUpdated(changed, branch) else false; 541 if (!source_moved and !branchUpdated(changed, record.event.target_branch)) continue; 542 } 543 try ids.append(allocator, id); 544 } 545 } 546 547 // start progress reporting after the patch count is known 548 progress.report(repo_opts, io, progress_ctx_maybe, .{ .text = "Checking mergeability" }); 549 progress.report(repo_opts, io, progress_ctx_maybe, .{ .start = .{ .kind = .writing_patch, .estimated_total_items = ids.items.len } }); 550 551 // refresh each collected patch independently 552 for (ids.items) |id| { 553 refreshMergeability(repo_opts, io, allocator, target_repo, id); 554 progress.report(repo_opts, io, progress_ctx_maybe, .{ .complete_one = .writing_patch }); 555 } 556 557 // finish progress reporting after all patches have been visited 558 progress.report(repo_opts, io, progress_ctx_maybe, .{ .end = .writing_patch }); 559 } 560 561 fn refreshMergeCheck( 562 comptime repo_opts: rp.RepoOpts(.xit), 563 io: std.Io, 564 allocator: std.mem.Allocator, 565 target_repo: *rp.Repo(.xit, repo_opts), 566 id: *const [evt.event_id_size]u8, 567 ) !void { 568 const Repo = rp.Repo(.xit, repo_opts); 569 const DB = Repo.DB; 570 var arena = std.heap.ArenaAllocator.init(allocator); 571 defer arena.deinit(); 572 573 // skip patches that no longer need mergeability checks 574 const initial = (try evt.readFromRepo(evt.Patch, .xit, repo_opts, io, allocator, &arena, target_repo, id)) orelse return; 575 if (initial.removed or initial.event.status.kind() != .open) return; 576 577 // acquire fork locks before the target repository lock 578 var checks: MergeCheckInputs(repo_opts) = .{}; 579 defer checks.deinit(io, allocator); 580 try checks.add(io, allocator, &target_repo.core, id.*, initial.event); 581 582 // cancel the cache transaction when the result is unchanged 583 { 584 const Ctx = struct { 585 core: *Repo.Core, 586 checks: *MergeCheckInputs(repo_opts), 587 io: std.Io, 588 allocator: std.mem.Allocator, 589 590 pub fn run(ctx: @This(), cursor: *DB.Cursor(.read_write)) !void { 591 var moment = try DB.HashMap(.read_write).init(cursor.*); 592 const state = Repo.State(.read_write){ .core = ctx.core, .extra = .{ .moment = &moment } }; 593 if (!try ctx.checks.update(state, ctx.io, ctx.allocator)) return error.CancelTransaction; 594 595 // name the transaction, which no user action owns 596 try xit.undo.write(repo_opts, state, std.Io.Timestamp.now(ctx.io, .real).toSeconds(), .{ .custom = .{ .action_kind = mergeability_undo_action } }); 597 } 598 }; 599 600 try target_repo.core.db_file.lock(io, .exclusive); 601 defer target_repo.core.db_file.unlock(io); 602 603 const history = try DB.ArrayList(.read_write).init(target_repo.core.db.rootCursor()); 604 history.appendContext(.{ .slot = try history.getSlot(-1) }, Ctx{ 605 .core = &target_repo.core, 606 .checks = &checks, 607 .io = io, 608 .allocator = allocator, 609 }) catch |err| switch (err) { 610 error.CancelTransaction => {}, 611 else => return err, 612 }; 613 } 614 } 615 616 // prepare fork handles before taking the target lock. this preserves the 617 // fork-before-target lock order used by publishing and merge operations. 618 pub fn MergeCheckInputs(comptime repo_opts: rp.RepoOpts(.xit)) type { 619 const Repo = rp.Repo(.xit, repo_opts); 620 return struct { 621 forks: std.AutoArrayHashMapUnmanaged([evt.event_id_size]u8, ?Repo) = .empty, 622 const Self = @This(); 623 624 pub fn prepare(self: *Self, io: std.Io, allocator: std.mem.Allocator, repo: *Repo, events: []const evt.EventWithId) !void { 625 for (events) |event| { 626 if (event.event != .patch) continue; 627 const patch = event.event.patch orelse continue; 628 try self.add(io, allocator, &repo.core, try evt.parseEventId(&event.id), patch); 629 } 630 } 631 632 fn add(self: *Self, io: std.Io, allocator: std.mem.Allocator, core: *Repo.Core, id: [evt.event_id_size]u8, patch: evt.Patch) !void { 633 if (patch.source_branch != null) return self.addFork(io, allocator, id, patch, null); 634 635 // find the fork through the forker recorded in the target's config 636 var moment = try core.latestMoment(); 637 var config = try xit.config.Config(.xit, repo_opts).init(.{ .core = core, .extra = .{ .moment = &moment } }, io, allocator); 638 defer config.deinit(); 639 const forker_id = (try fork.readForkerId(&config.sections, &id)) orelse return self.addFork(io, allocator, id, patch, null); 640 641 // the target lives at <users dir>/<owner>/repos/<id> 642 const users_dir = std.fs.path.dirname(std.fs.path.dirname(std.fs.path.dirname(core.work_path) orelse ".") orelse ".") orelse "."; 643 const path = try fork.forkPath(allocator, users_dir, &forker_id, &id); 644 defer allocator.free(path); 645 try self.addFork(io, allocator, id, patch, path); 646 } 647 648 // track an open patch once, locking the fork at `fork_path` when it has one 649 fn addFork(self: *Self, io: std.Io, allocator: std.mem.Allocator, id: [evt.event_id_size]u8, patch: evt.Patch, fork_path: ?[]const u8) !void { 650 if (patch.status.kind() != .open) return; 651 const entry = try self.forks.getOrPut(allocator, id); 652 if (entry.found_existing) return; 653 entry.value_ptr.* = null; 654 const path = fork_path orelse return; 655 656 // open the fork before acquiring its shared lock 657 var opened = Repo.open(io, allocator, .{ .path = path, .require_repo_root = true }) catch |err| { 658 std.log.warn("mergeability fork unavailable: {s}", .{@errorName(err)}); 659 return; 660 }; 661 662 // transfer the locked fork to the collection for cleanup 663 errdefer opened.deinit(io, allocator); 664 try opened.core.db_file.lock(io, .shared); 665 entry.value_ptr.* = opened; 666 } 667 668 pub fn deinit(self: *Self, io: std.Io, allocator: std.mem.Allocator) void { 669 for (self.forks.values()) |*value| if (value.*) |*repo| { 670 repo.core.db_file.unlock(io); 671 repo.deinit(io, allocator); 672 }; 673 self.forks.deinit(allocator); 674 } 675 676 // optional checks may fail without rejecting a patch, but a failed 677 // restoration must abort the transaction so trial writes cannot be 678 // committed. reports whether any cached result changed. 679 pub fn update(self: *Self, state: Repo.State(.read_write), io: std.Io, allocator: std.mem.Allocator) !bool { 680 var arena = std.heap.ArenaAllocator.init(allocator); 681 defer arena.deinit(); 682 var changed = false; 683 for (self.forks.keys(), self.forks.values()) |id, *repo| { 684 const saved = updateMergeCheck(repo_opts, state, io, &arena, &id, if (repo.*) |*value| value else null) catch |err| switch (err) { 685 error.MergeCheckRestoreFailed => return err, 686 else => { 687 std.log.warn("failed to refresh mergeability: {s}", .{@errorName(err)}); 688 continue; 689 }, 690 }; 691 if (saved) changed = true; 692 } 693 return changed; 694 } 695 }; 696 } 697 698 // the caller holds the target's exclusive lock, and any fork's shared lock. 699 // this only saves the cache; it never creates a top-level history transaction. 700 fn updateMergeCheck( 701 comptime repo_opts: rp.RepoOpts(.xit), 702 state: rp.Repo(.xit, repo_opts).State(.read_write), 703 io: std.Io, 704 arena: *std.heap.ArenaAllocator, 705 id: *const [evt.event_id_size]u8, 706 fork_repo: ?*rp.Repo(.xit, repo_opts), 707 ) !bool { 708 const DB = rp.Repo(.xit, repo_opts).DB; 709 710 // read the patch from the current transaction 711 const target_events = evt.currentMomentFromRepoMoment(repo_opts.hash, state.extra.moment.readOnly()) catch |err| switch (err) { 712 error.NotFound => return false, 713 else => return err, 714 }; 715 const patch_record = (try evt.Patch.readById(DB, repo_opts.hash, target_events, arena, id)) orelse return false; 716 if (patch_record.removed or patch_record.event.status.kind() != .open) return false; 717 718 // find the newest fork revision when the patch uses a fork 719 var newest: ?evt.PatchRev.WithId = null; 720 if (fork_repo) |repo| { 721 if (evt.currentMoment(repo_opts, repo)) |events| { 722 newest = evt.PatchRev.readNewest(DB, repo_opts.hash, events, arena) catch |err| blk: { 723 std.log.warn("mergeability revision unavailable: {s}", .{@errorName(err)}); 724 break :blk null; 725 }; 726 } else |err| std.log.warn("mergeability revision unavailable: {s}", .{@errorName(err)}); 727 } 728 729 // compare the selected revision and target against the cache 730 const patch = patch_record.event; 731 const revision = patch.revision orelse return false; 732 const target_oid = try rf.readRecur(.xit, repo_opts, state.readOnly(), io, .{ .ref = .{ .kind = .head, .name = patch.target_branch } }); 733 var check = MergeCheck{ .revision = revision, .target_branch = patch.target_branch, .target_oid = if (target_oid) |*oid| oid else "" }; 734 const cached = try readMergeCheck(repo_opts, target_events, arena, id); 735 const current = if (patch.source_branch != null) 736 try branchCurrent(repo_opts, state.readOnly(), io, arena, patch) 737 else if (newest) |latest| 738 std.mem.eql(u8, &std.fmt.bytesToHex(latest.id, .lower), &revision.id) and revision.matches(latest.record) 739 else 740 false; 741 742 // run trial merges only for a current revision with an unresolved check 743 if (current and target_oid != null) { 744 if (cached) |value| if (value.matches(patch, check.target_oid) and value.result.source != .unknown and value.result.squash != .unknown) return false; 745 check.result = try trialMergeCheck(repo_opts, state, io, arena.child_allocator, &check, fork_repo); 746 } 747 748 // persist the result only when it differs from the cache 749 if (cached) |value| if (evt.fieldEqual(MergeCheck, value, check)) return false; 750 try saveMergeCheck(repo_opts, state.extra.moment.*, id, check); 751 return true; 752 } 753 754 fn trialMergeCheck( 755 comptime repo_opts: rp.RepoOpts(.xit), 756 state: rp.Repo(.xit, repo_opts).State(.read_write), 757 io: std.Io, 758 allocator: std.mem.Allocator, 759 check: *const MergeCheck, 760 fork_repo: ?*rp.Repo(.xit, repo_opts), 761 ) !Mergeability { 762 // freeze the parent state so trial writes copy its nodes 763 const moment = state.extra.moment; 764 const snapshot = moment.slot(); 765 // TODO: use an xitdb savepoint to also reclaim trial bytes on rollback. 766 // restoring the root leaves the appended data unused until gc. 767 try state.core.db.freeze(); 768 moment.* = rp.Repo(.xit, repo_opts).DB.HashMap(.read_write).init(moment.cursor) catch return error.MergeCheckRestoreFailed; 769 770 // keep the trial result in memory while discarding its state changes 771 const result = calculateMergeCheck(repo_opts, state, io, allocator, check, fork_repo); 772 773 // restore the parent state before handling any trial error 774 moment.cursor.write(.{ .slot = snapshot }) catch return error.MergeCheckRestoreFailed; 775 moment.* = rp.Repo(.xit, repo_opts).DB.HashMap(.read_write).init(moment.cursor) catch return error.MergeCheckRestoreFailed; 776 777 // report failed trials as unknown after the parent state is restored 778 return result catch |err| { 779 std.log.warn("mergeability check failed: {s}", .{@errorName(err)}); 780 return .{}; 781 }; 782 } 783 784 fn calculateMergeCheck( 785 comptime repo_opts: rp.RepoOpts(.xit), 786 state: rp.Repo(.xit, repo_opts).State(.read_write), 787 io: std.Io, 788 allocator: std.mem.Allocator, 789 check: *const MergeCheck, 790 fork_repo: ?*rp.Repo(.xit, repo_opts), 791 ) !Mergeability { 792 const source = try oidArray(repo_opts.hash, check.revision.source_oid); 793 const squash = try oidArray(repo_opts.hash, check.revision.squash_oid); 794 795 // import fork objects into the temporary trial state 796 if (fork_repo) |repo| { 797 var fork_moment = try repo.core.latestMoment(); 798 const fork_state = rp.Repo(.xit, repo_opts).State(.read_only){ .core = &repo.core, .extra = .{ .moment = &fork_moment } }; 799 var objects = try obj.ObjectIterator(.xit, repo_opts).init(fork_state, io, allocator, .{ .kind = .all }); 800 defer objects.deinit(); 801 try objects.include(&source); 802 try objects.include(&squash); 803 try obj.copyFromObjectIterator(.xit, repo_opts, state, .xit, repo_opts, &objects, io, null); 804 } 805 806 // check source and squash merges against the same target 807 const target = rf.Ref{ .kind = .head, .name = check.target_branch }; 808 return .{ 809 .source = checkCommit(repo_opts, state, io, allocator, &source, target), 810 .squash = checkCommit(repo_opts, state, io, allocator, &squash, target), 811 }; 812 } 813 814 fn saveMergeCheck(comptime repo_opts: rp.RepoOpts(.xit), moment: rp.Repo(.xit, repo_opts).DB.HashMap(.read_write), id: *const [evt.event_id_size]u8, check: MergeCheck) !void { 815 const DB = rp.Repo(.xit, repo_opts).DB; 816 817 // store the check under the current event history moment 818 const last_object_id_cursor = try moment.getCursor(hash.hashInt(repo_opts.hash, evt.last_object_id_key)) orelse return error.NotFound; 819 var last_object_id: [hash.byteLen(repo_opts.hash)]u8 = undefined; 820 _ = try last_object_id_cursor.readBytes(&last_object_id); 821 const haxy_history = try DB.ArrayList(.read_write).init(try moment.putCursor(hash.hashInt(repo_opts.hash, evt.history_key))); 822 const haxy_moments = try DB.HashMap(.read_write).init(try haxy_history.putCursor(-1)); 823 const haxy_moment = try DB.HashMap(.read_write).init(try haxy_moments.putCursor(hash.bytesToInt(repo_opts.hash, &last_object_id))); 824 const map = try DB.HashMap(.read_write).init(try haxy_moment.putCursor(hash.hashInt(repo_opts.hash, evt.Patch.patch_id_to_mergeability_key))); 825 const entry = try DB.HashMap(.read_write).init(try map.putCursor(hash.hashInt(repo_opts.hash, id))); 826 try evt.upsert(MergeCheck, DB, repo_opts.hash, entry, check); 827 } 828 829 fn branchCurrent(comptime repo_opts: rp.RepoOpts(.xit), state: rp.Repo(.xit, repo_opts).State(.read_only), io: std.Io, arena: *std.heap.ArenaAllocator, patch: evt.Patch) !bool { 830 // check that the source branch still points to the selected commit 831 const branch = patch.source_branch orelse return false; 832 const selected = patch.revision orelse return false; 833 const oid = (try rf.readRecur(.xit, repo_opts, state, io, .{ .ref = .{ .kind = .head, .name = branch } })) orelse return false; 834 if (!std.mem.eql(u8, &oid, selected.source_oid)) return false; 835 836 // verify that the stored revision metadata also matches 837 const events = try evt.currentMomentFromRepoMoment(repo_opts.hash, state.extra.moment.*); 838 const revision = (try evt.PatchRev.readById(evt.EventDB(repo_opts.hash), repo_opts.hash, events, arena, &try evt.parseEventId(&selected.id))) orelse return false; 839 return selected.matches(revision); 840 } 841 842 fn oidArray(comptime hash_kind: hash.HashKind, oid: []const u8) ![hash.hexLen(hash_kind)]u8 { 843 try evt.PatchRev.validateOid(hash_kind, oid); 844 var result: [hash.hexLen(hash_kind)]u8 = undefined; 845 @memcpy(&result, oid); 846 return result; 847 } 848 849 fn checkCommit( 850 comptime repo_opts: rp.RepoOpts(.xit), 851 state: rp.Repo(.xit, repo_opts).State(.read_write), 852 io: std.Io, 853 allocator: std.mem.Allocator, 854 oid: *const [hash.hexLen(repo_opts.hash)]u8, 855 target: rf.Ref, 856 ) Mergeability.Status { 857 var result = mrg.Merge(.xit, repo_opts).init(state, io, allocator, .{ 858 .kind = .full, 859 .action = .{ .new = .{ .source = &.{.{ .oid = oid }}, .algo = .patch } }, 860 .dry_run = true, 861 }, target, null) catch |err| { 862 std.log.warn("mergeability check failed: {s}", .{@errorName(err)}); 863 return .unknown; 864 }; 865 defer result.deinit(); 866 867 // translate the dry-run outcome into a cached status 868 return switch (result.result) { 869 .conflict => .conflict, 870 .clean, .success, .nothing, .fast_forward => .clean, 871 }; 872 } 873 874 pub const MergeInput = struct { 875 id: [evt.event_id_size * 2]u8, 876 revision: evt.Patch.MergeRevision, 877 author: evt.CommitAuthor, 878 timestamp: u64, 879 }; 880 881 pub fn publish( 882 comptime repo_opts: rp.RepoOpts(.xit), 883 io: std.Io, 884 allocator: std.mem.Allocator, 885 users_dir: []const u8, 886 admin_repo: *rp.Repo(.xit, evt.admin_repo_opts), 887 target_repo: *rp.Repo(.xit, repo_opts), 888 input: PublishInput, 889 ) !void { 890 const patch_id = try evt.parseEventId(&input.id); 891 var arena = std.heap.ArenaAllocator.init(allocator); 892 defer arena.deinit(); 893 894 // validate the fork ownership and publishing identity 895 var user_repo = (try evt.openUserRepo(io, allocator, users_dir, &input.user_id)) orelse return error.InvalidPatchDraft; 896 defer user_repo.deinit(io, allocator); 897 const admin_moment = try evt.currentMoment(evt.admin_repo_opts, admin_repo); 898 const user_moment = (try evt.userMoment(&user_repo)) orelse return error.InvalidPatchDraft; 899 const fork_record = (try evt.Fork.readById(evt.UserDB, evt.user_repo_opts.hash, user_moment, &arena, &patch_id)) orelse return error.InvalidPatchDraft; 900 if (fork_record.removed or !std.mem.eql(u8, fork_record.event.repo_id, &input.repo_id)) return error.InvalidPatchDraft; 901 const user = (try evt.User.readById(evt.AdminDB, evt.admin_repo_opts.hash, admin_moment, &arena, &input.user_id)) orelse return error.InvalidPatchDraft; 902 if (user.removed or 903 !std.mem.eql(u8, user.event.name, input.author.name) or 904 !std.mem.eql(u8, user.event.email, input.author.email)) return error.InvalidPatchDraft; 905 906 // read the patch staged in the fork 907 const fork_path = try fork.forkPath(arena.allocator(), users_dir, &input.user_id, &patch_id); 908 var fork_repo = rp.Repo(.xit, repo_opts).open(io, allocator, .{ .path = fork_path }) catch return error.PatchDataUnavailable; 909 defer fork_repo.deinit(io, allocator); 910 const fork_moment = evt.currentMoment(repo_opts, &fork_repo) catch return error.PatchDataUnavailable; 911 const local_patch = (try evt.Patch.readById(evt.EventDB(repo_opts.hash), repo_opts.hash, fork_moment, &arena, &patch_id)) orelse return error.PatchDataUnavailable; 912 913 // check whether this patch has already been published 914 const target_moment = evt.currentMoment(repo_opts, target_repo) catch |err| switch (err) { 915 error.NotFound => null, 916 else => |other| return other, 917 }; 918 const existing = if (target_moment) |moment| 919 try evt.Patch.readById(evt.EventDB(repo_opts.hash), repo_opts.hash, moment, &arena, &patch_id) 920 else 921 null; 922 if (existing) |record| { 923 if (record.removed or !std.mem.eql(u8, record.author_email orelse "", input.author.email)) return error.InvalidPatch; 924 } 925 926 // publish the selected revision unless the draft was already consumed 927 if (local_patch.removed) { 928 if (existing == null) return error.PatchDataUnavailable; 929 } else { 930 var patch = (existing orelse local_patch).event; 931 if (local_patch.event.revision) |selected| { 932 const revision_id = try evt.parseEventId(&selected.id); 933 const revision = (try evt.PatchRev.readById(evt.EventDB(repo_opts.hash), repo_opts.hash, fork_moment, &arena, &revision_id)) orelse return error.PatchNotPushed; 934 if (!selected.matches(revision)) return error.PatchNotPushed; 935 patch.revision = selected; 936 } 937 if (existing == null) patch.status = .open; 938 939 // write the published patch and its forker to the target repository 940 { 941 var checks: MergeCheckInputs(repo_opts) = .{}; 942 defer checks.deinit(io, allocator); 943 try checks.addFork(io, allocator, patch_id, patch, fork_path); 944 const first_parent = try eventsFirstParent(.xit, repo_opts, io, allocator, target_repo); 945 946 const DB = rp.Repo(.xit, repo_opts).DB; 947 const State = rp.Repo(.xit, repo_opts).State; 948 const Ctx = struct { 949 core: *rp.Repo(.xit, repo_opts).Core, 950 io: std.Io, 951 allocator: std.mem.Allocator, 952 events: []const evt.EventWithId, 953 patch_id: [evt.event_id_size]u8, 954 forker_id: [evt.event_id_size * 2]u8, 955 checks: *MergeCheckInputs(repo_opts), 956 first_parent: ?[1][hash.hexLen(repo_opts.hash)]u8, 957 958 pub fn run(ctx: @This(), cursor: *DB.Cursor(.read_write)) !void { 959 var moment = try DB.HashMap(.read_write).init(cursor.*); 960 const state = State(.read_write){ .core = ctx.core, .extra = .{ .moment = &moment } }; 961 962 // code that starts from the target finds the fork through this entry 963 var config = try xit.config.Config(.xit, repo_opts).init(state.readOnly(), ctx.io, ctx.allocator); 964 defer config.deinit(); 965 try config.add(state, ctx.io, .{ .name = &fork.forkerConfigName(&ctx.patch_id), .value = &ctx.forker_id }); 966 967 const parent = if (try rf.readRecur(.xit, repo_opts, state.readOnly(), ctx.io, .{ .ref = evt.events_ref }) == null) ctx.first_parent else null; 968 try evt.commitEvents(.xit, repo_opts, state, ctx.io, ctx.allocator, evt.events_ref, ctx.events, parent); 969 if (!try evt.consumeInTransaction(.repo, .xit, repo_opts, state, &ctx.core.db, &moment, ctx.io, ctx.allocator, evt.events_ref)) return error.CancelTransaction; 970 _ = try ctx.checks.update(state, ctx.io, ctx.allocator); 971 try xit.undo.write(repo_opts, state, std.Io.Timestamp.now(ctx.io, .real).toSeconds(), .{ .custom = .{ .action_kind = evt.undo_action } }); 972 } 973 }; 974 975 try target_repo.core.db_file.lock(io, .exclusive); 976 defer target_repo.core.db_file.unlock(io); 977 978 const history = try DB.ArrayList(.read_write).init(target_repo.core.db.rootCursor()); 979 history.appendContext(.{ .slot = try history.getSlot(-1) }, Ctx{ 980 .core = &target_repo.core, 981 .io = io, 982 .allocator = allocator, 983 .events = &.{.{ 984 .id = input.id, 985 .timestamp = input.timestamp, 986 .author = input.author, 987 .event = .{ .patch = patch }, 988 }}, 989 .patch_id = patch_id, 990 .forker_id = std.fmt.bytesToHex(input.user_id, .lower), 991 .checks = &checks, 992 .first_parent = first_parent, 993 }) catch |err| switch (err) { 994 error.CancelTransaction => {}, 995 else => |e| return e, 996 }; 997 } 998 999 // remove the consumed draft from the fork 1000 try evt.consume(.server, .fork, .xit, repo_opts, io, allocator, &fork_repo, evt.events_ref, &.{.{ 1001 .id = input.id, 1002 .timestamp = input.timestamp, 1003 .author = input.author, 1004 .event = .{ .patch = null }, 1005 }}); 1006 } 1007 1008 // mark the fork as published in the forker's user repo 1009 if (fork_record.event.stage == .draft) { 1010 var published = fork_record.event; 1011 published.stage = .publish; 1012 try evt.consume(.server, .user, .xit, evt.user_repo_opts, io, allocator, &user_repo, evt.events_ref, &.{.{ 1013 .id = input.id, 1014 .timestamp = input.timestamp, 1015 .author = input.author, 1016 .event = .{ .fork = published }, 1017 }}); 1018 } 1019 } 1020 1021 pub fn editDraft( 1022 comptime repo_opts: rp.RepoOpts(.xit), 1023 io: std.Io, 1024 allocator: std.mem.Allocator, 1025 users_dir: []const u8, 1026 admin_repo: *rp.Repo(.xit, evt.admin_repo_opts), 1027 input: EditDraftInput, 1028 ) !bool { 1029 // validate the submitted fields before opening repositories 1030 if (!evt.Patch.fieldsValid(input.title, input.labels)) return error.InvalidFields; 1031 if (!evt.Patch.branchValid(input.target_branch)) return error.InvalidTargetBranch; 1032 const patch_id = try evt.parseEventId(&input.id); 1033 var arena = std.heap.ArenaAllocator.init(allocator); 1034 defer arena.deinit(); 1035 1036 // validate the draft ownership and editing identity 1037 { 1038 var user_repo = (try evt.openUserRepo(io, allocator, users_dir, &input.user_id)) orelse return false; 1039 defer user_repo.deinit(io, allocator); 1040 const admin_moment = try evt.currentMoment(evt.admin_repo_opts, admin_repo); 1041 const user_moment = (try evt.userMoment(&user_repo)) orelse return false; 1042 const fork_record = (try evt.Fork.readById(evt.UserDB, evt.user_repo_opts.hash, user_moment, &arena, &patch_id)) orelse return false; 1043 if (fork_record.event.stage != .draft) return false; 1044 if (fork_record.removed or !std.mem.eql(u8, fork_record.event.repo_id, &input.repo_id)) return error.InvalidPatchDraft; 1045 const user = (try evt.User.readById(evt.AdminDB, evt.admin_repo_opts.hash, admin_moment, &arena, &input.user_id)) orelse return error.InvalidPatchDraft; 1046 if (user.removed or 1047 !std.mem.eql(u8, user.event.name, input.author.name) or 1048 !std.mem.eql(u8, user.event.email, input.author.email)) return error.InvalidPatchDraft; 1049 } 1050 1051 // read the draft patch from its fork 1052 const fork_path = try fork.forkPath(arena.allocator(), users_dir, &input.user_id, &patch_id); 1053 var fork_repo = rp.Repo(.xit, repo_opts).open(io, allocator, .{ .path = fork_path, .require_repo_root = true }) catch return error.PatchDataUnavailable; 1054 defer fork_repo.deinit(io, allocator); 1055 const fork_moment = evt.currentMoment(repo_opts, &fork_repo) catch return error.PatchDataUnavailable; 1056 const record = (try evt.Patch.readById(evt.EventDB(repo_opts.hash), repo_opts.hash, fork_moment, &arena, &patch_id)) orelse return error.PatchDataUnavailable; 1057 if (record.removed) return error.PatchDataUnavailable; 1058 1059 // apply the edits and invalidate a revision targeting another branch 1060 var patch = record.event; 1061 patch.title = input.title; 1062 patch.labels = input.labels; 1063 patch.description = input.description; 1064 if (!std.mem.eql(u8, patch.target_branch, input.target_branch)) patch.revision = null; 1065 patch.target_branch = input.target_branch; 1066 1067 // commit the updated draft metadata 1068 try evt.consume(.server, .fork, .xit, repo_opts, io, allocator, &fork_repo, evt.events_ref, &.{.{ 1069 .id = input.id, 1070 .timestamp = input.timestamp, 1071 .author = input.author, 1072 .event = .{ .patch = patch }, 1073 }}); 1074 return true; 1075 } 1076 1077 // merge the selected patch revision into its target branch 1078 pub fn merge( 1079 comptime repo_opts: rp.RepoOpts(.xit), 1080 io: std.Io, 1081 allocator: std.mem.Allocator, 1082 users_dir: []const u8, 1083 target_repo: *rp.Repo(.xit, repo_opts), 1084 input: MergeInput, 1085 ) !void { 1086 const patch_id = try evt.parseEventId(&input.id); 1087 errdefer |err| { 1088 if (err == error.PatchDataUnavailable or err == error.PatchOutOfDate) { 1089 refreshMergeability(repo_opts, io, allocator, target_repo, patch_id); 1090 } 1091 } 1092 var arena = std.heap.ArenaAllocator.init(allocator); 1093 defer arena.deinit(); 1094 1095 // validate the target patch 1096 const target_moment = try evt.currentMoment(repo_opts, target_repo); 1097 const patch_record = (try evt.Patch.readById(evt.EventDB(repo_opts.hash), repo_opts.hash, target_moment, &arena, &patch_id)) orelse return error.InvalidPatch; 1098 if (patch_record.removed) return error.InvalidPatch; 1099 switch (patch_record.event.status) { 1100 .open => {}, 1101 .closed => return error.PatchClosed, 1102 .merged => return error.PatchAlreadyMerged, 1103 } 1104 const selected = patch_record.event.revision orelse return error.PatchNotPushed; 1105 const target_ref = rf.Ref{ .kind = .head, .name = patch_record.event.target_branch }; 1106 1107 // open the fork 1108 var fork_repo_maybe: ?rp.Repo(.xit, repo_opts) = if (patch_record.event.source_branch != null) null else blk: { 1109 var config = try target_repo.listConfig(io, allocator); 1110 defer config.deinit(); 1111 const forker_id = (try fork.readForkerId(&config.sections, &patch_id)) orelse return error.PatchDataUnavailable; 1112 const fork_path = try fork.forkPath(arena.allocator(), users_dir, &forker_id, &patch_id); 1113 break :blk rp.Repo(.xit, repo_opts).open(io, allocator, .{ .path = fork_path, .require_repo_root = true }) catch return error.PatchDataUnavailable; 1114 }; 1115 defer if (fork_repo_maybe) |*repo| repo.deinit(io, allocator); 1116 1117 // select the commit and merge identity 1118 const selected_oid = oidArray(repo_opts.hash, switch (input.revision) { 1119 .squash => selected.squash_oid, 1120 .source => selected.source_oid, 1121 }) catch return error.InvalidPatch; 1122 const identity = try std.fmt.allocPrint(arena.allocator(), "{s} <{s}>", .{ input.author.name, input.author.email }); 1123 1124 // merge the code and events in one target transaction 1125 { 1126 const DB = rp.Repo(.xit, repo_opts).DB; 1127 const State = rp.Repo(.xit, repo_opts).State; 1128 const Ctx = struct { 1129 core: *rp.Repo(.xit, repo_opts).Core, 1130 fork_repo: ?*rp.Repo(.xit, repo_opts), 1131 io: std.Io, 1132 allocator: std.mem.Allocator, 1133 patch_id: [evt.event_id_size]u8, 1134 expected_patch: evt.Patch, 1135 target_ref: rf.Ref, 1136 selected_oid: [hash.hexLen(repo_opts.hash)]u8, 1137 revision: evt.Patch.MergeRevision, 1138 identity: []const u8, 1139 author: evt.CommitAuthor, 1140 timestamp: u64, 1141 1142 pub fn run(ctx: @This(), cursor: *DB.Cursor(.read_write)) !void { 1143 var moment = try DB.HashMap(.read_write).init(cursor.*); 1144 const state = State(.read_write){ .core = ctx.core, .extra = .{ .moment = &moment } }; 1145 1146 var check_arena = std.heap.ArenaAllocator.init(ctx.allocator); 1147 defer check_arena.deinit(); 1148 1149 // require the fork's newest revision while both repos are locked 1150 const selected_revision = ctx.expected_patch.revision orelse unreachable; 1151 const revision_id = try evt.parseEventId(&selected_revision.id); 1152 if (ctx.fork_repo) |repo| { 1153 const fork_moment = try evt.currentMoment(repo_opts, repo); 1154 const newest = (try evt.PatchRev.readNewest(evt.EventDB(repo_opts.hash), repo_opts.hash, fork_moment, &check_arena)) orelse return error.PatchDataUnavailable; 1155 if (!std.mem.eql(u8, &newest.id, &revision_id) or !selected_revision.matches(newest.record)) return error.PatchOutOfDate; 1156 } else if (!try branchCurrent(repo_opts, state.readOnly(), ctx.io, &check_arena, ctx.expected_patch)) return error.PatchOutOfDate; 1157 1158 // require a clean cached result for the selected merge style. 1159 { 1160 const availability = try readMergeability(repo_opts, state.readOnly(), ctx.io, &check_arena, &ctx.patch_id, ctx.expected_patch); 1161 switch (availability.get(ctx.revision)) { 1162 .clean => {}, 1163 .conflict => return error.MergeConflict, 1164 .unknown => return error.MergeCheckUnavailable, 1165 } 1166 } 1167 1168 // copy the selected commit and its dependencies 1169 if (ctx.fork_repo) |repo| { 1170 var fork_repo_moment = try repo.core.latestMoment(); 1171 const fork_state = State(.read_only){ .core = &repo.core, .extra = .{ .moment = &fork_repo_moment } }; 1172 var objects = try obj.ObjectIterator(.xit, repo_opts).init(fork_state, ctx.io, ctx.allocator, .{ .kind = .all }); 1173 defer objects.deinit(); 1174 try objects.include(&ctx.selected_oid); 1175 try obj.copyFromObjectIterator(.xit, repo_opts, state, .xit, repo_opts, &objects, ctx.io, null); 1176 } 1177 1178 // merge the selected commit 1179 const before_oid = (try rf.readRecur(.xit, repo_opts, state.readOnly(), ctx.io, .{ .ref = ctx.target_ref })) orelse return error.InvalidTargetBranch; 1180 { 1181 var merge_result = try mrg.Merge(.xit, repo_opts).init(state, ctx.io, ctx.allocator, .{ 1182 .kind = .full, 1183 .action = .{ .new = .{ .source = &.{.{ .oid = &ctx.selected_oid }}, .algo = .patch } }, 1184 .commit_metadata = .{ 1185 .author = ctx.identity, 1186 .committer = ctx.identity, 1187 .message = ctx.expected_patch.title, 1188 .timestamp = ctx.timestamp, 1189 }, 1190 }, ctx.target_ref, null); 1191 defer merge_result.deinit(); 1192 switch (merge_result.result) { 1193 .conflict => return error.MergeConflict, 1194 .success, .fast_forward, .nothing => {}, 1195 .clean => unreachable, 1196 } 1197 } 1198 const after_oid = (try rf.readRecur(.xit, repo_opts, state.readOnly(), ctx.io, .{ .ref = ctx.target_ref })) orelse return error.InvalidTargetBranch; 1199 1200 // retain the submitted revision and snapshot the merge result 1201 if (!try importMergedRevision(repo_opts, state, &ctx.core.db, &moment, ctx.io, ctx.allocator, ctx.fork_repo, &ctx.patch_id, .{ .revision = ctx.revision, .before_oid = &before_oid, .after_oid = &after_oid }, ctx.expected_patch, ctx.author, ctx.timestamp)) return error.PatchOutOfDate; 1202 1203 // the merge moved the target branch, so its indexes follow, 1204 // the commit index deriving from the old tip's version 1205 _ = try find.refreshInTransaction(repo_opts, state, &moment, ctx.io, ctx.allocator, null); 1206 _ = try srch_cmmt.refreshInTransaction(repo_opts, state, &moment, ctx.io, ctx.allocator, &.{.{ .old_oid = &before_oid, .new_oid = &after_oid }}, null); 1207 1208 // record the merge as one undoable action. 1209 try xit.undo.write(repo_opts, state, std.Io.Timestamp.now(ctx.io, .real).toSeconds(), .{ .custom = .{ .action_kind = merge_undo_action } }); 1210 } 1211 }; 1212 1213 // lock the fork before the target for the full merge transaction. 1214 if (fork_repo_maybe) |*repo| try repo.core.db_file.lock(io, .shared); 1215 defer if (fork_repo_maybe) |*repo| repo.core.db_file.unlock(io); 1216 try target_repo.core.db_file.lock(io, .exclusive); 1217 defer target_repo.core.db_file.unlock(io); 1218 1219 const history = try DB.ArrayList(.read_write).init(target_repo.core.db.rootCursor()); 1220 try history.appendContext(.{ .slot = try history.getSlot(-1) }, Ctx{ 1221 .core = &target_repo.core, 1222 .fork_repo = if (fork_repo_maybe) |*repo| repo else null, 1223 .io = io, 1224 .allocator = allocator, 1225 .patch_id = patch_id, 1226 .expected_patch = patch_record.event, 1227 .target_ref = target_ref, 1228 .selected_oid = selected_oid, 1229 .revision = input.revision, 1230 .identity = identity, 1231 .author = input.author, 1232 .timestamp = input.timestamp, 1233 }); 1234 } 1235 1236 // refresh other patches affected by the updated target branch. 1237 refreshOpenMergeability(repo_opts, io, allocator, target_repo, null, null); 1238 } 1239 1240 // merge a published patch and remove its fork 1241 pub fn mergeAndRemoveFork( 1242 comptime repo_opts: rp.RepoOpts(.xit), 1243 io: std.Io, 1244 allocator: std.mem.Allocator, 1245 users_dir: []const u8, 1246 target_repo: *rp.Repo(.xit, repo_opts), 1247 input: MergeInput, 1248 ) !void { 1249 try merge(repo_opts, io, allocator, users_dir, target_repo, input); 1250 var arena = std.heap.ArenaAllocator.init(allocator); 1251 defer arena.deinit(); 1252 1253 // remove fork data only for patches backed by a fork 1254 const patch_id = try evt.parseEventId(&input.id); 1255 const record = (try evt.readFromRepo(evt.Patch, .xit, repo_opts, io, allocator, &arena, target_repo, &patch_id)) orelse return error.InvalidPatch; 1256 if (record.event.source_branch != null) return; 1257 // a patch from another instance has no fork here 1258 const forker_id = blk: { 1259 var config = try target_repo.listConfig(io, allocator); 1260 defer config.deinit(); 1261 break :blk (try fork.readForkerId(&config.sections, &patch_id)) orelse return; 1262 }; 1263 try fork.remove(io, allocator, users_dir, &forker_id, &input.id, input.author); 1264 } 1265 1266 fn commitAuthor(line: []const u8) !evt.CommitAuthor { 1267 const open = std.mem.lastIndexOfScalar(u8, line, '<') orelse return error.AuthorNotFound; 1268 const close = std.mem.indexOfScalarPos(u8, line, open + 1, '>') orelse return error.AuthorNotFound; 1269 const name = std.mem.trimEnd(u8, line[0..open], " "); 1270 if (name.len == 0 or close == open + 1) return error.AuthorNotFound; 1271 return .{ .name = name, .email = line[open + 1 .. close] }; 1272 } 1273 1274 // import a revision after detecting a client-side merge 1275 fn importMerged( 1276 comptime repo_opts: rp.RepoOpts(.xit), 1277 state: rp.Repo(.xit, repo_opts).State(.read_write), 1278 db: *evt.EventDB(repo_opts.hash), 1279 moment: *evt.EventDB(repo_opts.hash).HashMap(.read_write), 1280 io: std.Io, 1281 allocator: std.mem.Allocator, 1282 users_dir: []const u8, 1283 patch_id: *const [evt.event_id_size]u8, 1284 matched_oid: *const [hash.hexLen(repo_opts.hash)]u8, 1285 before_oid: *const [hash.hexLen(repo_opts.hash)]u8, 1286 after_oid: *const [hash.hexLen(repo_opts.hash)]u8, 1287 ) !void { 1288 var arena = std.heap.ArenaAllocator.init(allocator); 1289 defer arena.deinit(); 1290 1291 // identify which selected revision was merged by the client. 1292 const events = try evt.currentMomentFromRepoMoment(repo_opts.hash, moment.readOnly()); 1293 const record = (try evt.Patch.readById(evt.EventDB(repo_opts.hash), repo_opts.hash, events, &arena, patch_id)) orelse return; 1294 const selected = record.event.revision orelse return; 1295 const revision: evt.Patch.MergeRevision = if (std.mem.eql(u8, matched_oid, selected.source_oid)) .source else if (std.mem.eql(u8, matched_oid, selected.squash_oid)) .squash else return; 1296 1297 // open the fork when the revision is not stored in the target repository. 1298 var fork_repo_maybe: ?rp.Repo(.xit, repo_opts) = if (record.event.source_branch != null) null else blk: { 1299 var config = try xit.config.Config(.xit, repo_opts).init(state.readOnly(), io, allocator); 1300 defer config.deinit(); 1301 const forker_id = (try fork.readForkerId(&config.sections, patch_id)) orelse return error.PatchDataUnavailable; 1302 const fork_path = try fork.forkPath(arena.allocator(), users_dir, &forker_id, patch_id); 1303 break :blk try rp.Repo(.xit, repo_opts).open(io, allocator, .{ .path = fork_path, .require_repo_root = true }); 1304 }; 1305 defer if (fork_repo_maybe) |*repo| repo.deinit(io, allocator); 1306 1307 // import the revision and record the client-side merge. 1308 _ = try importMergedRevision(repo_opts, state, db, moment, io, allocator, if (fork_repo_maybe) |*repo| repo else null, patch_id, .{ .revision = revision, .before_oid = before_oid, .after_oid = after_oid }, null, null, 0); 1309 } 1310 1311 // import the revision and mark its patch merged in the current transaction 1312 fn importMergedRevision( 1313 comptime repo_opts: rp.RepoOpts(.xit), 1314 state: rp.Repo(.xit, repo_opts).State(.read_write), 1315 db: *evt.EventDB(repo_opts.hash), 1316 moment: *evt.EventDB(repo_opts.hash).HashMap(.read_write), 1317 io: std.Io, 1318 allocator: std.mem.Allocator, 1319 fork_repo: ?*rp.Repo(.xit, repo_opts), 1320 patch_id: *const [evt.event_id_size]u8, 1321 merged: struct { 1322 revision: evt.Patch.MergeRevision, 1323 before_oid: *const [hash.hexLen(repo_opts.hash)]u8, 1324 after_oid: *const [hash.hexLen(repo_opts.hash)]u8, 1325 }, 1326 expected_patch: ?evt.Patch, 1327 merged_author: ?evt.CommitAuthor, 1328 merged_timestamp: u64, 1329 ) !bool { 1330 const DB = evt.EventDB(repo_opts.hash); 1331 var arena = std.heap.ArenaAllocator.init(allocator); 1332 defer arena.deinit(); 1333 1334 // read the patch and reject outdated or conflicting merge requests. 1335 const haxy_moment = try evt.currentMomentFromRepoMoment(repo_opts.hash, moment.readOnly()); 1336 const patch_record = (try evt.Patch.readById(DB, repo_opts.hash, haxy_moment, &arena, patch_id)) orelse return false; 1337 const selected = patch_record.event.revision orelse return false; 1338 if (patch_record.removed or patch_record.event.status.kind() == .merged) return false; 1339 if (expected_patch) |expected| { 1340 if (!evt.fieldEqual(evt.Patch, expected, patch_record.event)) return false; 1341 if (try haxy_moment.getCursor(hash.hashInt(repo_opts.hash, evt.Patch.conflicts_key))) |cursor| { 1342 const conflicts = try DB.SortedMap(.read_only).init(cursor); 1343 if (try conflicts.getCursor(&evt.orderKeyDesc(patch_record.created_order, patch_id)) != null) return false; 1344 } 1345 } 1346 const revision_id = try evt.parseEventId(&selected.id); 1347 1348 // resolve the selected revision from the target or its fork. 1349 const current_revision = try evt.PatchRev.readById(DB, repo_opts.hash, haxy_moment, &arena, &revision_id); 1350 const revision = (if (fork_repo) |repo| 1351 try evt.PatchRev.readById(DB, repo_opts.hash, try evt.currentMoment(repo_opts, repo), &arena, &revision_id) 1352 else 1353 current_revision) orelse return false; 1354 if (!selected.matches(revision)) return false; 1355 1356 // read the original revision commit and author. 1357 var fork_repo_moment = if (fork_repo) |repo| try repo.core.latestMoment() else moment.readOnly(); 1358 const fork_state = rp.Repo(.xit, repo_opts).State(.read_only){ .core = if (fork_repo) |repo| &repo.core else state.core, .extra = .{ .moment = &fork_repo_moment } }; 1359 var event_oid: [hash.hexLen(repo_opts.hash)]u8 = undefined; 1360 if (revision.event_oid.len != event_oid.len) return false; 1361 @memcpy(&event_oid, revision.event_oid); 1362 var event_object = try obj.Object(.xit, repo_opts).initCommit(fork_state, io, allocator, &event_oid); 1363 defer event_object.deinit(); 1364 const original_author = try commitAuthor(event_object.content.commit.metadata.author orelse return false); 1365 1366 // retain the submitted revision if the target has not imported it yet. 1367 var events_buffer: [3]evt.EventWithId = undefined; 1368 var events: std.ArrayList(evt.EventWithId) = .initBuffer(&events_buffer); 1369 var tree_entries: [2]evt.EventTreeEntry = undefined; 1370 const needs_import = if (current_revision) |existing| existing.removed else true; 1371 if (needs_import) { 1372 tree_entries = .{ 1373 .{ .tree = .{ .name = "base", .oid = revision.base_tree_oid } }, 1374 .{ .tree = .{ .name = "head", .oid = revision.head_tree_oid } }, 1375 }; 1376 events.appendAssumeCapacity(.{ 1377 .id = selected.id, 1378 .timestamp = event_object.content.commit.metadata.timestamp, 1379 .author = original_author, 1380 .tree_entries = &tree_entries, 1381 .event = .{ .patchrev = revision.event }, 1382 }); 1383 } else { 1384 const existing = current_revision orelse unreachable; 1385 if (!selected.matches(existing)) return error.InvalidPatch; 1386 } 1387 1388 // snapshot the merge result and mark the patch as merged. 1389 { 1390 const author = merged_author orelse original_author; 1391 const timestamp = if (merged_author != null) merged_timestamp else @as(u64, @intCast(std.Io.Timestamp.now(io, .real).toSeconds())); 1392 var patch = patch_record.event; 1393 const result = try evt.PatchRev.prepare(.xit, repo_opts, state, io, &arena, merged.before_oid, merged.after_oid, patch.title, author, timestamp); 1394 events.appendAssumeCapacity(result.event); 1395 patch.status = .{ .merged = .{ .revision = merged.revision, .patchrev_id = result.event.id } }; 1396 events.appendAssumeCapacity(.{ 1397 .id = std.fmt.bytesToHex(patch_id.*, .lower), 1398 .timestamp = timestamp, 1399 .author = author, 1400 .event = .{ .patch = patch }, 1401 }); 1402 } 1403 1404 // commit and consume the revision and patch events together. 1405 try evt.commitEvents(.xit, repo_opts, state, io, allocator, evt.events_ref, events.items, null); 1406 if (!try evt.consumeInTransaction(.repo, .xit, repo_opts, state, db, moment, io, allocator, evt.events_ref)) return error.CancelTransaction; 1407 return true; 1408 } 1409 1410 // find patches merged by commits received in a push 1411 pub fn detectMerged( 1412 comptime repo_opts: rp.RepoOpts(.xit), 1413 state: rp.Repo(.xit, repo_opts).State(.read_write), 1414 db: *evt.EventDB(repo_opts.hash), 1415 moment: *evt.EventDB(repo_opts.hash).HashMap(.read_write), 1416 io: std.Io, 1417 allocator: std.mem.Allocator, 1418 users_dir: []const u8, 1419 updates: []const xit.net_server_receive_pack.AppliedRefUpdate, 1420 error_writer: *std.Io.Writer, 1421 ) !void { 1422 const DB = evt.EventDB(repo_opts.hash); 1423 var arena = std.heap.ArenaAllocator.init(allocator); 1424 defer arena.deinit(); 1425 1426 // read the revision index used to identify merged patches. 1427 const haxy_moment = try evt.currentMomentFromRepoMoment(repo_opts.hash, moment.readOnly()); 1428 const revisions_cursor = try haxy_moment.getCursor(hash.hashInt(repo_opts.hash, evt.Patch.revision_to_id_set_key)) orelse return; 1429 const revisions = try DB.HashMap(.read_only).init(revisions_cursor); 1430 1431 // ignore event refs and inspect only changed branch tips. 1432 var events_ref_buffer: [rf.MAX_REF_CONTENT_SIZE]u8 = undefined; 1433 const events_ref_path = try evt.events_ref.toPath(&events_ref_buffer); 1434 const heads_prefix = "refs/heads/"; 1435 for (updates) |update| { 1436 if (!std.mem.startsWith(u8, update.ref_name, heads_prefix) or 1437 std.mem.eql(u8, update.ref_name, events_ref_path) or 1438 std.mem.eql(u8, update.old_oid, update.new_oid)) continue; 1439 1440 // validate the ref tips and skip branch deletions. 1441 if (update.old_oid.len != hash.hexLen(repo_opts.hash) or 1442 update.new_oid.len != hash.hexLen(repo_opts.hash)) return error.InvalidOid; 1443 var old_oid: [hash.hexLen(repo_opts.hash)]u8 = undefined; 1444 var new_oid: [hash.hexLen(repo_opts.hash)]u8 = undefined; 1445 @memcpy(&old_oid, update.old_oid); 1446 @memcpy(&new_oid, update.new_oid); 1447 if (std.mem.allEqual(u8, &new_oid, '0')) continue; 1448 1449 // traverse commits introduced by this branch update. 1450 var commits = try obj.ObjectIterator(.xit, repo_opts).init(state.readOnly(), io, arena.allocator(), .{ .kind = .commit }); 1451 defer commits.deinit(); 1452 if (!std.mem.allEqual(u8, &old_oid, '0')) try commits.exclude(&old_oid); 1453 var targets: std.ArrayList(struct { 1454 before_oid: [hash.hexLen(repo_opts.hash)]u8, 1455 after_oid: [hash.hexLen(repo_opts.hash)]u8, 1456 }) = .empty; 1457 1458 // collect merge boundaries along the target's first-parent history. 1459 { 1460 var first_parents = try obj.ObjectIterator(.xit, repo_opts).init(state.readOnly(), io, arena.allocator(), .{ .kind = .commit, .first_parent = true }); 1461 defer first_parents.deinit(); 1462 if (!std.mem.allEqual(u8, &old_oid, '0')) try first_parents.exclude(&old_oid); 1463 try first_parents.include(&new_oid); 1464 while (try first_parents.next(arena.allocator())) |commit| { 1465 defer commit.deinit(); 1466 try targets.append(arena.allocator(), .{ 1467 .before_oid = (commit.content.commit.metadata.firstParent() orelse &old_oid).*, 1468 .after_oid = commit.oid, 1469 }); 1470 } 1471 } 1472 1473 // visit target commits oldest first, attributing each revision to the 1474 // commit that introduced it; the shared iterator visits each ancestor once 1475 var revision_key_buffer: [rf.MAX_REF_CONTENT_SIZE + 1 + hash.hexLen(repo_opts.hash)]u8 = undefined; 1476 while (targets.pop()) |target| { 1477 try commits.include(&target.after_oid); 1478 while (try commits.next(arena.allocator())) |commit| { 1479 defer commit.deinit(); 1480 1481 // find patches whose selected revision matches this commit. 1482 const key = try std.fmt.bufPrint(&revision_key_buffer, "{s}\x00{s}", .{ update.ref_name[heads_prefix.len..], &commit.oid }); 1483 const ids_cursor = try revisions.getCursor(hash.hashInt(repo_opts.hash, key)) orelse continue; 1484 const ids = try DB.CountedHashSet(.read_only).init(ids_cursor); 1485 var ids_iter = try ids.iterator(); 1486 while (try ids_iter.next()) |cursor| { 1487 const pair = try cursor.readKeyValuePair(); 1488 var patch_id: [evt.event_id_size]u8 = undefined; 1489 if ((try pair.key_cursor.readBytes(&patch_id)).len != patch_id.len) return error.InvalidPatch; 1490 const patch_hex = std.fmt.bytesToHex(patch_id, .lower); 1491 1492 // a fast-forward has no recorded merge boundary; use the old ref tip 1493 const before_oid = if (std.mem.eql(u8, &commit.oid, &target.after_oid)) &old_oid else &target.before_oid; 1494 importMerged(repo_opts, state, db, moment, io, allocator, users_dir, &patch_id, &commit.oid, before_oid, &target.after_oid) catch |err| { 1495 serve_common.logError(io, error_writer, "failed to mark patch {s} as merged: {s}\n", .{ &patch_hex, @errorName(err) }); 1496 continue; 1497 }; 1498 } 1499 } 1500 } 1501 } 1502 } 1503