flaxo/wynba

public

AI coding agent!!

Initial commit: coding-agent4915718a month ago333 lines · 9.4 KB

1	use std::io::Write;
2
3	const RESET: &str
4	const TEXT: &str
5	const ACCENT: &str
6	const GREEN: &str
7	const ORANGE: &str
8	const GRAY: &str
9	const DIM: &str
10	const LINK: &str
11
12	pub struct MarkdownStream
13	out: W
14	cols: usize
15	in_code: bool
16	line: String
17	}
18
19	impl<W: Write
20	pub fn new(
21	MarkdownStream {
22	out,
23	cols: cols.max
24	in_code: false,
25	line: String
26	}
27	}
28
29	pub fn feed(
30	for c in strip_ansi
31	match c {
32	'\n' => {
33	let line =
34	self.emit_line(&line
35	}
36	'\r' => {}
37	'\t' => self
38	c => self.
39	}
40	}
41	self.out.flush()
42	Ok(())
43	}
44
45	pub fn finish(
46	if !self.line.
47	let line =
48	self.emit_line(&line
49	}
50	self.out.flush()
51	Ok(())
52	}
53
54	pub fn text(
55	self.feed(text
56	self.finish()
57	}
58
59	pub fn output(
60	&self.out
61	}
62
63	pub fn output_mut(
64	&mut self.out
65	}
66
67	pub fn set_cols(
68	self.cols = cols
69	}
70
71	fn wln(&mut
72	write!(self.out
73	}
74
75	fn border(&mut
76	self.wln(&format!
77	}
78
79	fn emit_line(&mut
80	let line =
81	if self.in_code {
82	let t =
83	if t.starts_with(
84	self.in_code = false;
85	return self.border();
86	}
87	if line.is_empty()
88	return self.wln(
89	}
90	return self.wln(
91	}
92	let indent =
93	let t =
94	if t.starts_with(
95	self.in_code = true;
96	return self.border();
97	}
98	if is_hr(t)
99	return self.wln(
100	}
101	if let Some
102	return self.wln(
103	}
104	if let Some
105	return self.wln(
106	}
107	if let Some
108	return self.wln(
109	}
110	if let Some
111	return self.wln(
112	}
113	if line.is_empty()
114	return self.wln(
115	}
116	self.wln(&format!
117	}
118	}
119
120	pub(crate) fn strip_ansi
121	let mut out
122	let mut chars
123	while let Some
124	if c == '
125	for d in chars
126	if d.is_ascii_alphabetic()
127	break;
128	}
129	}
130	} else {
131	out.push(c
132	}
133	}
134	out
135	}
136
137	fn is_hr(t:
138	t.chars().count
139	}
140
141	fn heading(t:
142	let n =
143	if n == 0
144	return None;
145	}
146	let rest =
147	if !rest.starts_with
148	return None;
149	}
150	Some(rest.
151	}
152
153	fn bullet(t:
154	for p in [
155	if let Some
156	return Some(r
157	}
158	}
159	None
160	}
161
162	fn numbered(t:
163	let bytes =
164	let mut i
165	while i < bytes
166	i += 1;
167	}
168	if i == 0
169	return None;
170	}
171	Some((t[
172	}
173
174	fn quote(t:
175	if let Some
176	Some(r.
177	} else {
178	None
179	}
180	}
181
182	fn find(chars:
183	(from..chars.len
184	}
185
186	fn find_seq(chars:
187	if from + seq
188	return None;
189	}
190	(from..=chars.len
191	}
192
193	fn inline(s:
194	let chars: Vec
195	let n =
196	let mut out
197	let mut i
198	while i < n
199	let c =
200	if c == '
201	out.push(chars
202	i += 2;
203	continue;
204	}
205	if c == '
206	if let Some
207	let inner: String
208	out.push_str("\
209	out.push_str(&inline
210	out.push_str(RESET
211	i = j
212	continue;
213	}
214	out.push_str("**
215	i += 2;
216	continue;
217	}
218	if c == '
219	if let Some
220	let inner: String
221	out.push_str(ACCENT
222	out.push_str(&inner
223	out.push_str(RESET
224	i = j
225	continue;
226	}
227	out.push('`
228	i += 1;
229	continue;
230	}
231	if c == '
232	if let Some
233	let inner: String
234	out.push_str("\
235	out.push_str(&inline
236	out.push_str(RESET
237	i = j
238	continue;
239	}
240	out.push('*
241	i += 1;
242	continue;
243	}
244	if c == '
245	if let Some
246	&& close + 1
247	&& chars[close
248	&& let Some
249	{
250	let label: String
251	let url: String
252	out.push_str(LINK
253	out.push_str(&inline
254	out.push_str(RESET
255	out.push_str(DIM
256	out.push_str(&format!
257	out.push_str(RESET
258	i = end
259	continue;
260	}
261	out.push('[
262	i += 1;
263	continue;
264	}
265	out.push(c
266	i += 1;
267	}
268	out
269	}
270
271	#[cfg(test)]
272	mod tests {
273	use super::*;
274
275	fn render(text:
276	let mut buf
277	let mut md
278	md.text(text
279	String::from_utf8(
280	}
281
282	#[test]
283	fn strips_ansi_from_input() {
284	assert_eq!(strip_ansi(
285	}
286
287	#[test]
288	fn plain_text_passes_through() {
289	let out =
290	assert!(out.contains
291	}
292
293	#[test]
294	fn crlf_newlines() {
295	let out =
296	assert!(out.contains
297	assert!(!out.
298	}
299
300	#[test]
301	fn bold_and_code() {
302	let out =
303	assert!(out.contains
304	assert!(out.contains
305	}
306
307	#[test]
308	fn heading() {
309	let out =
310	assert!(out.contains
311	}
312
313	#[test]
314	fn code_block_bordered() {
315	let out =
316	assert_eq!(out.matches
317	assert!(out.contains
318	}
319
320	#[test]
321	fn bullet_and_numbered() {
322	let out =
323	assert!(out.contains
324	assert!(out.contains
325	}
326
327	#[test]
328	fn quote_and_hr() {
329	let out =
330	assert!(out.contains
331	assert!(out.contains
332	}
333	}