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 let slug = self.slug.clone();
137 self.store.with(|state| {
138 let id = ChangeId((state.changes.len() + 1).to_string());
141 let url = format!("https://{DEFAULT_HOST}/{slug}/pull/{}", id.0);
142 let change = ChangeRequest {
143 head_sha: Sha(events::stable_sha(&[&slug, &req.head, &id.0])),
144 url: Url::parse(&url).map_err(|e| Error::Invalid {
145 reason: format!("{url:?} is not a URL: {e}"),
146 })?,
147 base: req.base.clone(),
148 id: id.clone(),
149 };
150 state.heads.insert(id, req.head.clone());
151 state.changes.push(change.clone());
152 Ok(change)
153 })
154 }
155
156 fn find_changes(&self, head: &str, base: &str) -> Result<Vec<ChangeRequest>> {
157 addressable(head, "the head branch")?;
158 addressable(base, "the base branch")?;
159 let state = self.store.snapshot()?;
160 Ok(state
161 .changes
162 .iter()
163 .filter(|change| {
164 change.base == base
165 && state.heads.get(&change.id).is_some_and(|from| from == head)
166 && !matches!(state.merges.get(&change.id), Some(MergeOutcome::Merged(_)))
169 })
170 .cloned()
171 .collect())
172 }
173
174 fn change_checks(&self, cr: &ChangeRequest) -> Result<Vec<Check>> {
175 Ok(self
176 .store
177 .snapshot()?
178 .checks
179 .get(&cr.id)
180 .cloned()
181 .unwrap_or_default())
182 }
183
184 fn check_log(&self, cr: &ChangeRequest, check: &Check) -> Result<ArtifactId> {
185 let log = self
186 .store
187 .snapshot()?
188 .check_logs
189 .get(&cr.id)
190 .and_then(|logs| logs.get(&check.name))
191 .cloned()
192 .unwrap_or_else(|| format!("the host log for check {}\n", check.name));
193 events::store_artifact(&artifact_id(&cr.id, &check.name), &log)
194 }
195
196 fn merge(&self, cr: &ChangeRequest, policy: MergePolicy) -> Result<MergeOutcome> {
197 self.store.with(|state| {
198 if let Some(decided) = state.merges.get(&cr.id) {
202 return Ok(decided.clone());
203 }
204 let landed = |state: &mut HostState| {
205 let sha = Sha(events::stable_sha(&["merge", &cr.id.0, cr.url.as_str()]));
206 state
207 .merges
208 .insert(cr.id.clone(), MergeOutcome::Merged(sha.clone()));
209 MergeOutcome::Merged(sha)
210 };
211 Ok(match policy {
212 MergePolicy::LocalDirect | MergePolicy::ChangeOpen => MergeOutcome::Open,
215 MergePolicy::ChangeAuto => {
216 if required_checks_green(state, &cr.id) {
217 landed(state)
218 } else {
219 state.merges.insert(cr.id.clone(), MergeOutcome::Queued);
222 MergeOutcome::Queued
223 }
224 }
225 MergePolicy::ChangeDirect => landed(state),
226 })
227 })
228 }
229}
230
231fn required_checks_green(state: &HostState, id: &ChangeId) -> bool {
236 let checks = match state.checks.get(id) {
237 Some(checks) => checks,
238 None => return false,
239 };
240 let required: Vec<&Check> = checks.iter().filter(|check| check.required).collect();
241 !required.is_empty() && required.iter().all(|check| check.green())
242}
243
244fn artifact_id(change: &ChangeId, check: &str) -> String {
250 let safe: String = check
251 .chars()
252 .map(|c| {
253 if c.is_ascii_alphanumeric() || c == '-' || c == '_' || c == '.' {
254 c
255 } else {
256 '-'
257 }
258 })
259 .collect();
260 let number: String = change
261 .0
262 .chars()
263 .filter(|c| c.is_ascii_alphanumeric())
264 .collect();
265 format!("a-testing-{number}-{safe}")
266}
267
268fn addressable(value: &str, what: &str) -> Result<()> {
275 if value.is_empty() || value.starts_with('-') || value.contains(char::is_whitespace) {
276 return Err(Error::Invalid {
277 reason: format!(
278 "{what} {value:?} cannot address anything on the host: it must be non-empty, \
279 must not begin with '-', and must carry no whitespace"
280 ),
281 });
282 }
283 Ok(())
284}
285
286fn named_repository(slug: &str) -> Result<String> {
288 let mut parts = slug.split('/');
289 let named = matches!(
290 (parts.next(), parts.next(), parts.next()),
291 (Some(owner), Some(name), None)
292 if !owner.is_empty()
293 && !name.is_empty()
294 && !slug.starts_with('-')
295 && !slug.contains(char::is_whitespace)
296 );
297 if !named {
298 return Err(Error::Invalid {
299 reason: format!("{slug:?} does not name one repository as owner/name"),
300 });
301 }
302 Ok(slug.to_owned())
303}