feat: expose merge requests as workspace resources

This commit is contained in:
2026-08-21 11:31:45 +09:00
parent ccb9d03865
commit 0cd5ca11cc
17 changed files with 880 additions and 148 deletions
+118
View File
@@ -27,6 +27,14 @@ pub enum MergeRequestState {
Closed,
}
impl MergeRequestState {
fn as_str(self) -> &'static str {
match self {
Self::Open => "open",
Self::Merged => "merged",
Self::Closed => "closed",
}
}
fn parse(v: &str) -> Result<Self, MergeRequestError> {
match v {
"draft" | "open" => Ok(Self::Open),
@@ -216,6 +224,23 @@ impl MergeRequest {
}
}
#[derive(Clone, Debug, Default)]
pub struct MergeRequestListQuery {
pub state: Option<MergeRequestState>,
pub repository_id: Option<String>,
pub ticket_id: Option<String>,
pub selector_from: Option<String>,
pub selector_to: Option<String>,
pub cursor: Option<String>,
pub limit: usize,
}
#[derive(Clone, Debug)]
pub struct MergeRequestListPage {
pub items: Vec<MergeRequest>,
pub next_cursor: Option<String>,
}
#[derive(Debug, Clone)]
pub struct OpenMergeRequest {
pub merge_request_id: String,
@@ -935,6 +960,80 @@ impl MergeRequestStore {
None => Err(MergeRequestError::NotFound),
}
}
pub fn get_by_id(
&self,
workspace_id: &str,
merge_request_id: &str,
) -> Result<MergeRequest, MergeRequestError> {
let c = self.lock()?;
load_mr(&c, workspace_id, merge_request_id)?.ok_or(MergeRequestError::NotFound)
}
pub fn list(
&self,
workspace_id: &str,
query: &MergeRequestListQuery,
) -> Result<MergeRequestListPage, MergeRequestError> {
let c = self.lock()?;
let limit = query.limit.clamp(1, 100);
let cursor_position = match query.cursor.as_deref() {
Some(cursor) => Some(
c.query_row(
"SELECT updated_at,merge_request_id FROM merge_requests WHERE workspace_id=?1 AND merge_request_id=?2",
params![workspace_id, cursor],
|row| Ok((row.get::<_, String>(0)?, row.get::<_, String>(1)?)),
)
.optional()?
.ok_or_else(|| MergeRequestError::Validation("invalid merge request cursor".into()))?,
),
None => None,
};
let cursor_updated_at = cursor_position
.as_ref()
.map(|(updated_at, _)| updated_at.as_str());
let cursor_id = cursor_position.as_ref().map(|(_, id)| id.as_str());
let state = query.state.map(MergeRequestState::as_str);
let mut statement = c.prepare(
"SELECT mr.merge_request_id
FROM merge_requests mr
WHERE mr.workspace_id=?1
AND (?2 IS NULL OR mr.state=?2)
AND (?3 IS NULL OR mr.repository_id=?3)
AND (?4 IS NULL OR mr.selector_from=?4)
AND (?5 IS NULL OR mr.selector_to=?5)
AND (?6 IS NULL OR EXISTS (
SELECT 1 FROM merge_request_ticket_relations relation
WHERE relation.workspace_id=mr.workspace_id
AND relation.merge_request_id=mr.merge_request_id
AND relation.ticket_id=?6
))
AND (?7 IS NULL OR mr.updated_at<?7 OR (mr.updated_at=?7 AND mr.merge_request_id>?8))
ORDER BY mr.updated_at DESC,mr.merge_request_id ASC
LIMIT ?9",
)?;
let rows = statement.query_map(
params![
workspace_id,
state,
query.repository_id.as_deref(),
query.selector_from.as_deref(),
query.selector_to.as_deref(),
query.ticket_id.as_deref(),
cursor_updated_at,
cursor_id,
(limit + 1) as i64,
],
|row| row.get::<_, String>(0),
)?;
let mut ids = rows.collect::<Result<Vec<_>, _>>()?;
let has_more = ids.len() > limit;
ids.truncate(limit);
let next_cursor = has_more.then(|| ids.last().cloned()).flatten();
let items = ids
.iter()
.map(|id| load_mr(&c, workspace_id, id)?.ok_or(MergeRequestError::NotFound))
.collect::<Result<Vec<_>, _>>()?;
Ok(MergeRequestListPage { items, next_cursor })
}
pub fn thread_page(
&self,
ws: &str,
@@ -946,6 +1045,25 @@ impl MergeRequestStore {
let c = self.lock()?;
load_thread(&c, ws, &mr.merge_request_id, after, limit.clamp(1, 200))
}
pub fn thread_page_by_id(
&self,
workspace_id: &str,
merge_request_id: &str,
after: Option<u64>,
limit: usize,
) -> Result<Vec<MergeRequestThreadEvent>, MergeRequestError> {
let c = self.lock()?;
if load_mr(&c, workspace_id, merge_request_id)?.is_none() {
return Err(MergeRequestError::NotFound);
}
load_thread(
&c,
workspace_id,
merge_request_id,
after,
limit.clamp(1, 200),
)
}
fn assigned(&self, a: &MergeRequestAuth, t: &str, r: &str) -> Result<(), MergeRequestError> {
self.repo(a, r)?;
let x = self
+69
View File
@@ -436,6 +436,75 @@ fn selector_repair_rejects_unapproved_resolved_subject() {
assert!(matches!(result, Err(MergeRequestError::NotReady(_))));
}
#[test]
fn first_class_list_and_detail_are_workspace_scoped_and_cursor_bounded() {
let (dir, store) = fixture();
open(&store);
Connection::open(dir.path().join("db"))
.unwrap()
.execute(
"UPDATE merge_requests SET state='closed' WHERE workspace_id='W' AND merge_request_id='MR'",
[],
)
.unwrap();
store
.open_merge_request(OpenMergeRequest {
merge_request_id: "MR-2".into(),
ticket_id: "T".into(),
repository_id: "R".into(),
selector_from: "work/t-2".into(),
selector_to: "develop".into(),
summary: "second".into(),
auth: auth(),
now: at(8),
})
.unwrap();
let first = store
.list(
"W",
&MergeRequestListQuery {
ticket_id: Some("T".into()),
limit: 1,
..Default::default()
},
)
.unwrap();
assert_eq!(first.items[0].merge_request_id, "MR-2");
assert_eq!(first.next_cursor.as_deref(), Some("MR-2"));
let second = store
.list(
"W",
&MergeRequestListQuery {
ticket_id: Some("T".into()),
cursor: first.next_cursor,
limit: 1,
..Default::default()
},
)
.unwrap();
assert_eq!(second.items[0].merge_request_id, "MR");
assert!(second.next_cursor.is_none());
let closed = store
.list(
"W",
&MergeRequestListQuery {
state: Some(MergeRequestState::Closed),
limit: 10,
..Default::default()
},
)
.unwrap();
assert_eq!(closed.items.len(), 1);
assert_eq!(store.get_by_id("W", "MR").unwrap().merge_request_id, "MR");
assert!(matches!(
store.get_by_id("other", "MR"),
Err(MergeRequestError::NotFound)
));
}
#[test]
fn transactional_completion_rejects_assignment_changed_in_control_plane_db() {
let (dir, store) = fixture();