1use std::path::PathBuf;
11
12use url::Url;
13
14use onevcs::{
15 ArtifactId, ChangeId, ChangeRequest, ChangeSpec, Check, Error, Hosting, MergeOutcome,
16 MergePolicy, RemoteHost, Result, Sha,
17};
18
19use crate::events;
20use crate::state::HostState;
21use crate::store::{FileStore, MemoryStore, Store};
22
23pub const DEFAULT_HOST: &str = "github.com";
26
27pub const DEFAULT_SLUG: &str = "onevcs/testing";
31
32#[derive(Debug)]
39pub struct Host<T> {
40 store: T,
41 slug: String,
48}
49
50pub type MemoryHost = Host<MemoryStore<HostState>>;
52
53pub type FileHost = Host<FileStore<HostState>>;
56
57impl MemoryHost {
58 pub fn new() -> Self {
60 Self::seeded(HostState::default())
61 }
62
63 pub fn seeded(state: HostState) -> Self {
65 Self {
66 store: MemoryStore::new(state),
67 slug: DEFAULT_SLUG.to_owned(),
68 }
69 }
70
71 pub fn state(&self) -> HostState {
73 self.store
74 .snapshot()
75 .expect("an in-memory store always answers")
76 }
77}
78
79impl Default for MemoryHost {
80 fn default() -> Self {
81 Self::new()
82 }
83}
84
85impl FileHost {
86 pub fn create(path: impl Into<PathBuf>) -> Result<Self> {
92 Ok(Self {
93 store: FileStore::attach(path, &HostState::default())?,
94 slug: DEFAULT_SLUG.to_owned(),
95 })
96 }
97
98 pub fn seeded(path: impl Into<PathBuf>, state: HostState) -> Result<Self> {
101 Ok(Self {
102 store: FileStore::replace(path, &state)?,
103 slug: DEFAULT_SLUG.to_owned(),
104 })
105 }
106
107 pub fn state(&self) -> Result<HostState> {
109 self.store.snapshot()
110 }
111}
112
113impl<T: Store<HostState> + Clone + std::fmt::Debug + Send + Sync + 'static> Hosting for Host<T> {
114 fn for_repo(&self, slug: &str) -> Result<Box<dyn RemoteHost>> {
115 Ok(Box::new(Host {
116 store: self.store.clone(),
117 slug: named_repository(slug)?,
118 }))
119 }
120}
121
122impl<T: Store<HostState>> RemoteHost for Host<T> {
123 fn authenticated_user(&self) -> Result<String> {
124 let login = self.store.snapshot()?.authenticated_user;
125 if login.trim().is_empty() {
126 return Err(Error::Invalid {
127 reason: "the host reported no authenticated user".to_owned(),
128 });
129 }
130 Ok(login)
131 }
132
133 fn open_change(&self, req: ChangeSpec) -> Result<ChangeRequest> {
134 addressable(&req.head, "the head branch")?;
135 addressable(&req.base, "the base branch")?;
136 crate::state::titled(&req.title)?;
139 let slug = self.slug.clone();
140 self.store.with(|state| {
141 let id = ChangeId((state.changes.len() + 1).to_string());
144 let url = format!("https://{DEFAULT_HOST}/{slug}/pull/{}", id.0);
145 let change = ChangeRequest {
146 head_sha: Sha(events::stable_sha(&[&slug, &req.head, &id.0])),
147 url: Url::parse(&url).map_err(|e| Error::Invalid {
148 reason: format!("{url:?} is not a URL: {e}"),
149 })?,
150 base: req.base.clone(),
151 id: id.clone(),
152 };
153 state.heads.insert(id.clone(), req.head.clone());
154 state.titles.insert(id, req.title.clone());
155 state.changes.push(change.clone());
156 Ok(change)
157 })
158 }
159
160 fn find_changes(&self, head: &str, base: &str) -> Result<Vec<ChangeRequest>> {
161 addressable(head, "the head branch")?;
162 addressable(base, "the base branch")?;
163 let state = self.store.snapshot()?;
164 Ok(state
165 .changes
166 .iter()
167 .filter(|change| {
168 change.base == base
169 && state.heads.get(&change.id).is_some_and(|from| from == head)
170 && !matches!(state.merges.get(&change.id), Some(MergeOutcome::Merged(_)))
173 })
174 .cloned()
175 .collect())
176 }
177
178 fn change_checks(&self, cr: &ChangeRequest) -> Result<Vec<Check>> {
179 Ok(self
180 .store
181 .snapshot()?
182 .checks
183 .get(&cr.id)
184 .cloned()
185 .unwrap_or_default())
186 }
187
188 fn check_log(&self, cr: &ChangeRequest, check: &Check) -> Result<ArtifactId> {
189 let log = self
190 .store
191 .snapshot()?
192 .check_logs
193 .get(&cr.id)
194 .and_then(|logs| logs.get(&check.name))
195 .cloned()
196 .unwrap_or_else(|| format!("the host log for check {}\n", check.name));
197 events::store_artifact(&artifact_id(&cr.id, &check.name), &log)
198 }
199
200 fn merge(&self, cr: &ChangeRequest, policy: MergePolicy) -> Result<MergeOutcome> {
201 self.store.with(|state| {
202 if let Some(decided) = state.merges.get(&cr.id) {
206 return Ok(decided.clone());
207 }
208 let landed = |state: &mut HostState| {
209 let sha = Sha(events::stable_sha(&["merge", &cr.id.0, cr.url.as_str()]));
210 state
211 .merges
212 .insert(cr.id.clone(), MergeOutcome::Merged(sha.clone()));
213 MergeOutcome::Merged(sha)
214 };
215 Ok(match policy {
216 MergePolicy::LocalDirect | MergePolicy::ChangeOpen => MergeOutcome::Open,
219 MergePolicy::ChangeAuto => {
220 if required_checks_green(state, &cr.id) {
221 landed(state)
222 } else {
223 state.merges.insert(cr.id.clone(), MergeOutcome::Queued);
226 MergeOutcome::Queued
227 }
228 }
229 MergePolicy::ChangeDirect => landed(state),
230 })
231 })
232 }
233}
234
235fn required_checks_green(state: &HostState, id: &ChangeId) -> bool {
240 let checks = match state.checks.get(id) {
241 Some(checks) => checks,
242 None => return false,
243 };
244 let required: Vec<&Check> = checks.iter().filter(|check| check.required).collect();
245 !required.is_empty() && required.iter().all(|check| check.green())
246}
247
248fn artifact_id(change: &ChangeId, check: &str) -> String {
254 let safe: String = check
255 .chars()
256 .map(|c| {
257 if c.is_ascii_alphanumeric() || c == '-' || c == '_' || c == '.' {
258 c
259 } else {
260 '-'
261 }
262 })
263 .collect();
264 let number: String = change
265 .0
266 .chars()
267 .filter(|c| c.is_ascii_alphanumeric())
268 .collect();
269 format!("a-testing-{number}-{safe}")
270}
271
272fn addressable(value: &str, what: &str) -> Result<()> {
279 if value.is_empty() || value.starts_with('-') || value.contains(char::is_whitespace) {
280 return Err(Error::Invalid {
281 reason: format!(
282 "{what} {value:?} cannot address anything on the host: it must be non-empty, \
283 must not begin with '-', and must carry no whitespace"
284 ),
285 });
286 }
287 Ok(())
288}
289
290fn named_repository(slug: &str) -> Result<String> {
292 let mut parts = slug.split('/');
293 let named = matches!(
294 (parts.next(), parts.next(), parts.next()),
295 (Some(owner), Some(name), None)
296 if !owner.is_empty()
297 && !name.is_empty()
298 && !slug.starts_with('-')
299 && !slug.contains(char::is_whitespace)
300 );
301 if !named {
302 return Err(Error::Invalid {
303 reason: format!("{slug:?} does not name one repository as owner/name"),
304 });
305 }
306 Ok(slug.to_owned())
307}