You are a constitutional council ranking individual git commits for ownership allocation. Compare these two commits. Decide which contributed more lasting value to the project. Judge substance, not spectacle: - Prefer correct, lasting design and real bugfixes over churn, formatting, renames, or generated noise. - Prefer clarity and necessity over sheer line count. A small precise change can beat a large diffuse one. - Do not favor a side merely because its patch is longer or noisier. - Weight what the change does for the project, not the contributor's name. Return ONLY a JSON object: {"winner": "A" or "B", "ratio": "N:M", "explanation": "..."} The explanation must cite concrete differences in the patches (1-3 sentences). Side A — contributor: tommy-mor Side A — commit message: [529cc941] Replace autocomplete parser with paste-and-go navigate. The keystroke transition graph was unreliable; a textarea plus Go button now parses pasted Reddit URLs and redirects to the subreddit ranking scope. Co-authored-by: Cursor Side A — unified diff (full patch): diff --git a/AGENTS.md b/AGENTS.md index 77f2e31d4e860a92a77255ca5106c8b6c4510ee7..36ee4c0ec700bffbe226ee775ba9cf59cf35c770 100644 --- a/AGENTS.md +++ b/AGENTS.md @@ -11,7 +11,6 @@ Single Rust web app **`sorter2-server`**: pairwise voting, rank-centrality ranki - **Rust 1.88+** is required (some transitive crates need a recent Cargo). The image may ship older `/usr/local/cargo` (1.83); use **rustup** and `rustup default 1.88.0` before building. - **System packages** for builds: `pkg-config`, `libssl-dev` (for `reqwest` / OpenSSL in integration tests and release builds). - **Clojure CLI 1.12.0.1530** (optional but used in CI): install from https://clojure.org/guides/install_clojure — needed for `./scripts/clj-test.sh` / Kaocha tests. -- **Playwright browser** for the spel browser test (`test/parser_race.clj`): install once with `clojure -M -e "(com.microsoft.playwright.CLI/main (into-array String [\"install\" \"chromium\" \"--with-deps\"]))"`. The browser binary is cached under `~/.cache/ms-playwright`. ### Commands (see also `TEST.sh`) diff --git a/server/src/api/ui_html.rs b/server/src/api/ui_html.rs index 1339048c955c0a3eb5120381aaf031424cbed540..c4ab9d65c7b3cd42a5b4d093ba429993c101e9a8 100644 --- a/server/src/api/ui_html.rs +++ b/server/src/api/ui_html.rs @@ -8,7 +8,7 @@ use std::collections::HashMap; use crate::{ html::{js_string_literal, ranking_panel, JsBuilder}, parser::parse_reddit_url, - parser_render::parser_panel_morph, + parser_render::navigate_panel, state::AppState, ui_action::{parse_html_ui_from_form, HtmlUiAction}, }; @@ -57,18 +57,23 @@ pub async fn post_ui_html( .morph_selector("#ranking-panel", panel) .into_response() } - HtmlUiAction::ParseQuery { query } => { - let action = parse_reddit_url(&query); - let panel = parser_panel_morph(&query, &action); - let mut js = JsBuilder::new().morph_selector("#parser-panel", panel); - if let Some(comp) = action.primary_completion() { - js = js.raw(&format!( - "var __pi=document.getElementById('parser-input'); if(__pi){{__pi.dataset.completion={};}}", - js_string_literal(comp) - )); + HtmlUiAction::ParseQuery { query } => match parse_reddit_url(&query) { + Ok(subreddit) => { + let dest = format!("/?sub={subreddit}"); + JsBuilder::new() + .raw(&format!( + "window.location.href={};", + js_string_literal(&dest) + )) + .into_response() } - js.into_response() - } + Err(message) => { + let panel = navigate_panel(&query, Some(&message)); + JsBuilder::new() + .morph_selector("#parser-panel", panel) + .into_response() + } + }, } } diff --git a/server/src/html/mod.rs b/server/src/html/mod.rs index 94a5cfb561d1c8464bd2782f7ffd4e0965ccf95d..9650d333d29c4ac94ceb407aee3ee00399c7f40b 100644 --- a/server/src/html/mod.rs +++ b/server/src/html/mod.rs @@ -11,8 +11,7 @@ use serde::Deserialize; use crate::{ form_template::template_json_compact, - parser_action::ParserAction, - parser_render::parser_panel, + parser_render::navigate_panel, ranking::{top_bottom, RankedItem}, reducer::GroupState, state::{normalize_scope, AppState}, @@ -336,10 +335,9 @@ pub async fn home( let empty = GroupState::new(); let group = groups.get(&scope).unwrap_or(&empty); - let empty_action = ParserAction::suggest(String::new(), None); let body = html! { h1 { "sorter2" } - (parser_panel("", &empty_action)) + (navigate_panel("", None)) (vote_panel(&scope)) (ranking_panel(&scope, group)) }; diff --git a/server/src/lib.rs b/server/src/lib.rs index fa423640d598f4ba97a5885d228e78d7b97f7a22..de8bca48cbf689cad22337883e7791966e7c4919 100644 --- a/server/src/lib.rs +++ b/server/src/lib.rs @@ -4,7 +4,6 @@ pub mod events; pub mod form_template; pub mod html; pub mod parser; -pub mod parser_action; pub mod parser_render; pub mod path_types; pub mod ranking; diff --git a/server/src/parser.rs b/server/src/parser.rs index ea437c4434045b44cba04a2b3416df850db518e7..50571a59f0d3ece1e2538f88e00ef46ec40ea545 100644 --- a/server/src/parser.rs +++ b/server/src/parser.rs @@ -1,1811 +1,87 @@ -use std::collections::HashMap; -use std::sync::OnceLock; +//! Extract a subreddit name from a pasted Reddit URL or path. -use crate::parser_action::{GuideOption, ParserAction, ScrollingSuggestion, Suggestion}; - -// --- Core Abstractions --- - -/// Unique identifier for nodes in the graph -type NodeId = &'static str; - -/// Pattern matching for edges -#[derive(Debug, Clone)] -pub enum EdgePattern { - /// Matches exact literal string - Literal(&'static str), - - /// Matches any prefix of a string and suggests the full string - /// e.g., PrefixOf("reddit.com") matches "r", "re", "red", "reddit", "reddit.com" - PrefixOf(&'static str), - - /// Captures a variable segment (e.g., subreddit name, username) - Variable(&'static str), - - /// Matches any string (wildcard) - Any, -} - -impl EdgePattern { - /// Try to match this pattern against input, return (consumed_chars, captured_value) - fn matches(&self, input: &str) -> Option<(usize, Option)> { - match self { - EdgePattern::Literal(lit) => { - if input.starts_with(lit) { - Some((lit.len(), None)) - } else { - None - } - } - EdgePattern::PrefixOf(target) => { - // Check if input is a prefix of target - if target.starts_with(input) && !input.is_empty() { - // It's a valid prefix - Some((input.len(), None)) - } else if input.starts_with(target) { - // Full match - Some((target.len(), None)) - } else { - None - } - } - EdgePattern::Variable(var_name) => { - // Consume until next '/' or end of string - let end = input.find('/').unwrap_or(input.len()); - if end > 0 { - let captured = input[..end].to_string(); - // Validate based on variable type - if is_valid_variable(var_name, &captured) { - Some((end, Some(captured))) - } else { - None - } - } else { - None - } - } - EdgePattern::Any => { - // Match everything until next '/' or end - let end = input.find('/').unwrap_or(input.len()); - if end > 0 { - Some((end, Some(input[..end].to_string()))) - } else { - None - } - } - } - } - - /// Get the completion suggestion for this pattern - fn completion(&self, partial: &str) -> Option { - match self { - EdgePattern::PrefixOf(target) => { - if target.starts_with(partial) && partial != *target { - Some(target.to_string()) - } else { - None - } - } - _ => None, - } +pub fn parse_reddit_url(query: &str) -> Result { + let q = query.trim(); + if q.is_empty() { + return Err("Paste a Reddit URL or r/subreddit path".into()); } -} -/// Edge in the graph -pub struct Edge { - pattern: EdgePattern, - target: NodeId, - /// Optional description for autocomplete - description: Option<&'static str>, -} - -/// Handler function for generating UI actions (Send + Sync so the graph can live in `OnceLock`). -type Handler = Box) -> ParserAction + Send + Sync>; - -/// Node in the graph -pub struct Node { - #[allow(dead_code)] - id: NodeId, - edges: Vec, - handler: Option, -} - -/// The composable parser graph (immutable after `build`). -pub struct Graph { - nodes: HashMap, - root: NodeId, -} - -// --- Graph Builder (Fluent API) --- - -pub struct GraphBuilder { - nodes: HashMap, - current_node: Option, - root: NodeId, -} - -impl GraphBuilder { - pub fn new() -> Self { - let mut nodes = HashMap::new(); - nodes.insert( - "root", - Node { - id: "root", - edges: Vec::new(), - handler: None, - }, - ); - - GraphBuilder { - nodes, - current_node: Some("root"), - root: "root", - } + if let Some(sub) = subreddit_after_prefix(q, "r/") { + return Ok(sub); } - /// Select a node to add edges to - pub fn at(mut self, node_id: NodeId) -> Self { - self.nodes.entry(node_id).or_insert_with(|| Node { - id: node_id, - edges: Vec::new(), - handler: None, - }); - self.current_node = Some(node_id); - self + if let Some(sub) = subreddit_from_path_segment(q, "/r/") { + return Ok(sub); } - - /// Add an edge from the current node - pub fn edge(self, pattern: EdgePattern, target: NodeId) -> Self { - self.edge_with_desc(pattern, target, None) - } - - /// Add an edge with description - pub fn edge_with_desc( - mut self, - pattern: EdgePattern, - target: NodeId, - desc: Option<&'static str>, - ) -> Self { - let current = self.current_node.expect("No current node selected"); - - self.nodes.entry(target).or_insert_with(|| Node { - id: target, - edges: Vec::new(), - handler: None, - }); - - if let Some(node) = self.nodes.get_mut(current) { - node.edges.push(Edge { - pattern, - target, - description: desc, - }); - } - self - } - - /// Set handler for current node - pub fn handler(mut self, handler: F) -> Self - where - F: Fn(&str, &str, &HashMap) -> ParserAction + Send + Sync + 'static, - { - let current = self.current_node.expect("No current node selected"); - if let Some(node) = self.nodes.get_mut(current) { - node.handler = Some(Box::new(handler)); - } - self - } - - /// Build the final graph - pub fn build(self) -> Graph { - Graph { - nodes: self.nodes, - root: self.root, - } - } + Err("Could not find a subreddit in that URL".into()) } -// --- Parser Implementation --- - -impl Graph { - pub fn parse(&self, input: &str) -> ParserAction { - let normalized = input.trim().to_lowercase(); - let mut state = ParserState { - input: &normalized, - cursor: 0, - current_node_id: self.root, - context: HashMap::new(), - original_query: input.to_string(), - current_prefix: String::new(), - }; - - self.parse_recursive(&mut state) - } - - fn parse_recursive(&self, state: &mut ParserState) -> ParserAction { - let node = self - .nodes - .get(state.current_node_id) - .expect("Node not found in graph"); - - // If we've consumed all input, check for handler or suggestions - if state.cursor >= state.input.len() { - if let Some(handler) = &node.handler { - return handler(&state.original_query, &state.current_prefix, &state.context); - } - - // No handler, try to suggest based on available edges - return self.suggest_from_edges(node, state); - } - - let remaining = &state.input[state.cursor..]; - - // Try to match each edge - for edge in &node.edges { - if let Some((consumed, captured)) = edge.pattern.matches(remaining) { - // Save state for potential backtracking - let saved_cursor = state.cursor; - let saved_node = state.current_node_id; - let saved_prefix = state.current_prefix.clone(); - - // Update state - state.cursor += consumed; - state.current_node_id = edge.target; - state.current_prefix.push_str(&remaining[..consumed]); - - // Store captured variable if any - if let Some(value) = captured { - if let EdgePattern::Variable(var_name) = &edge.pattern { - state.context.insert(var_name.to_string(), value); - } - } - - // Check if this is a partial match that needs completion - if state.cursor == state.input.len() { - if let Some(completion_suffix) = edge.pattern.completion(remaining) { - // Use the current_prefix plus the completion suffix - let full_completion = format!("{}{}", - state.current_prefix, - completion_suffix.strip_prefix(remaining).unwrap_or(&completion_suffix) - ); - return ParserAction::suggest( - state.original_query.clone(), - Some(Suggestion { - text: full_completion.clone(), - completion: full_completion, - description: edge.description.map(|d| d.to_string()), - score: 1.0, - }) - ); - } - } - - // Continue parsing from the target node - let result = self.parse_recursive(state); - - // If we got a valid response, return it - if !matches!(result, ParserAction::ShowError(_)) { - return result; - } - - // Otherwise, restore state and try next edge - state.cursor = saved_cursor; - state.current_node_id = saved_node; - state.current_prefix = saved_prefix; - } - } - - // No edges matched - try to provide suggestions - self.suggest_from_edges(node, state) - } - - fn suggest_from_edges(&self, node: &Node, state: &ParserState) -> ParserAction { - let remaining = &state.input[state.cursor..]; - - // Find edges that could match with more input - for edge in &node.edges { - match &edge.pattern { - EdgePattern::PrefixOf(target) => { - if target.starts_with(remaining) && !remaining.is_empty() { - // Use current_prefix instead of rebuilding from input - let full_completion = format!("{}{}", state.current_prefix, target); - return ParserAction::suggest( - state.original_query.clone(), - Some(Suggestion { - text: full_completion.clone(), - completion: full_completion, - description: edge.description.map(|d| d.to_string()), - score: 1.0, - }) - ); - } - } - EdgePattern::Literal(lit) => { - if lit.starts_with(remaining) && !remaining.is_empty() { - let full_completion = format!("{}{}", state.current_prefix, lit); - return ParserAction::suggest( - state.original_query.clone(), - Some(Suggestion { - text: full_completion.clone(), - completion: full_completion, - description: edge.description.map(|d| d.to_string()), - score: 1.0, - }) - ); - } - } - _ => {} - } - } - - ParserAction::error( - "InvalidPath".to_string(), - format!("'{}' doesn't match any known pattern", state.original_query) - ) - } +fn subreddit_after_prefix(text: &str, prefix: &str) -> Option { + let rest = text.strip_prefix(prefix)?; + let sub = rest.split(['/', '?', '#']).next()?.trim(); + valid_subreddit(sub) } -struct ParserState<'a> { - input: &'a str, - cursor: usize, - current_node_id: NodeId, - context: HashMap, - original_query: String, - current_prefix: String, +fn subreddit_from_path_segment(text: &str, needle: &str) -> Option { + let idx = text.find(needle)?; + let rest = &text[idx + needle.len()..]; + let sub = rest.split(['/', '?', '#']).next()?.trim(); + valid_subreddit(sub) } -// --- Helper Functions --- - -fn is_valid_variable(var_name: &str, value: &str) -> bool { - match var_name { - "subreddit" => { - !value.is_empty() && - value.len() <= 21 && - value.chars().all(|c| c.is_alphanumeric() || c == '_') - } - "username" => { - !value.is_empty() && - value.len() <= 20 && - value.chars().all(|c| c.is_alphanumeric() || c == '_' || c == '-') - } - "post_id" => { - !value.is_empty() && - value.len() <= 10 && - value.chars().all(|c| c.is_alphanumeric()) - } - _ => true, // Allow any value for unknown variables +fn valid_subreddit(name: &str) -> Option { + if name.is_empty() { + return None; + } + if name + .chars() + .all(|c| c.is_ascii_alphanumeric() || c == '_') + { + Some(name.to_ascii_lowercase()) + } else { + None } -} - -// --- Define the Reddit Graph --- - -pub fn build_reddit_graph() -> Graph { - GraphBuilder::new() - // === ROOT LEVEL: Direct aliases and domain/protocol patterns === - .at("root") - // Direct aliases to subreddit and user selection - .edge_with_desc( - EdgePattern::PrefixOf("r/"), - "subreddit_selection", - Some("Browse subreddits (e.g., r/programming)") - ) - .edge_with_desc( - EdgePattern::PrefixOf("u/"), - "user_selection", - Some("Browse users (e.g., u/spez)") - ) - - // Reddit shortcuts - one pattern handles ALL prefixes! - .edge_with_desc( - EdgePattern::PrefixOf("reddit.com"), - "reddit_domain", - Some("Go to Reddit") - ) - - // Protocol patterns - "h" can suggest "https://" - .edge_with_desc( - EdgePattern::PrefixOf("https://"), - "https_protocol", - Some("HTTPS protocol") - ) - .edge_with_desc( - EdgePattern::PrefixOf("http://"), - "http_protocol", - Some("HTTP protocol") - ) - .edge_with_desc( - EdgePattern::PrefixOf("www."), - "www_prefix", - Some("World Wide Web") - ) - - - - // === HTTPS PROTOCOL: Can go to any domain === - .at("https_protocol") - .edge_with_desc( - EdgePattern::PrefixOf("reddit.com"), - "reddit_domain", - Some("Reddit (HTTPS)") - ) - .edge_with_desc( - EdgePattern::PrefixOf("www."), - "https_www", - Some("WWW prefix") - ) - - // === HTTP PROTOCOL: Similar to HTTPS === - .at("http_protocol") - .edge_with_desc( - EdgePattern::PrefixOf("reddit.com"), - "reddit_domain", - Some("Reddit (HTTP)") - ) - .edge_with_desc( - EdgePattern::PrefixOf("www."), - "http_www", - Some("WWW prefix") - ) - - // === HTTPS + WWW === - .at("https_www") - .edge_with_desc( - EdgePattern::PrefixOf("reddit.com"), - "reddit_domain", - Some("Reddit") - ) - - // === HTTP + WWW === - .at("http_www") - .edge_with_desc( - EdgePattern::PrefixOf("reddit.com"), - "reddit_domain", - Some("Reddit") - ) - - // === WWW PREFIX (without protocol) === - .at("www_prefix") - .edge_with_desc( - EdgePattern::PrefixOf("reddit.com"), - "reddit_domain", - Some("Reddit") - ) - - // === REDDIT DOMAIN: Expect "/" === - .at("reddit_domain") - .edge(EdgePattern::Literal("/"), "reddit_root") - .handler(|query, prefix, _ctx| { - // If someone just types "reddit.com" (or with protocol) without slash - // Suggest adding the slash using the current prefix - let completion = format!("{}/", prefix); - - ParserAction::suggest( - query.to_string(), - Some(Suggestion { - text: completion.clone(), - completion, - description: Some("Continue to Reddit homepage".to_string()), - score: 1.0, - }) - ) - }) - - // === REDDIT ROOT: The main Reddit navigation === - .at("reddit_root") - .edge(EdgePattern::Literal("r/"), "subreddit_selection") - .edge(EdgePattern::Literal("u/"), "user_selection") - .handler(|query, prefix, _ctx| { - ParserAction::multiple(vec![ - ParserAction::scrolling_suggestions( - query.to_string(), - vec![ - ScrollingSuggestion { - completion: format!("{}r/", prefix), - }, - ScrollingSuggestion { - completion: format!("{}u/", prefix), - }, - ], - 1400, // 1.4 second interval (slower) - true // loop through - ), - ParserAction::guide( - query.to_string(), - "Welcome to Sorter for Reddit".to_string(), - "Where would you like to start?".to_string(), - vec![ - GuideOption { - key: "r".to_string(), - label: "Sort a Subreddit".to_string(), - description: "Find the best posts in a community.".to_string(), - completion: format!("{}r/", prefix) - }, - GuideOption { - key: "u".to_string(), - label: "Sort User Content".to_string(), - description: "Explore and rank a user's posts and comments.".to_string(), - completion: format!("{}u/", prefix) - }, - ] - ), - ]) - }) - - - - // === SUBREDDIT SELECTION: THE UNIFIED NODE === - // This node is now reached from `r/` OR `reddit.com/r/` - .at("subreddit_selection") - .edge(EdgePattern::Variable("subreddit"), "subreddit_page") - .handler(|query, prefix, _ctx| { - ParserAction::multiple(vec![ - ParserAction::scrolling_suggestions( - query.to_string(), - vec![ - ScrollingSuggestion { - completion: format!("{}programming", prefix), - }, - ScrollingSuggestion { - completion: format!("{}askreddit", prefix), - }, - ScrollingSuggestion { - completion: format!("{}aww", prefix), - }, - ScrollingSuggestion { - completion: format!("{}rust", prefix), - }, - ScrollingSuggestion { - completion: format!("{}webdev", prefix), - }, - ], - 1600, // 1.6 second interval (slower) - true // loop through - ), - // Live DB-backed suggestions for subreddits as the user types - ParserAction::SuggestSubredditsFromDb { partial: query.to_string(), prefix: prefix.to_string() } - ]) - }) - - // === Specific subreddit page === - .at("subreddit_page") - .edge(EdgePattern::Literal("/"), "subreddit_slash") - .handler(|_query, prefix, ctx| { - // Use the new unified subreddit resolution logic - let subreddit = ctx.get("subreddit").cloned().unwrap_or_default(); - ParserAction::ResolveAndDisplaySubreddit { - subreddit, - prefix: prefix.to_string(), - } - }) - - .at("subreddit_slash") - .edge(EdgePattern::Literal("hot"), "subreddit_hot") - .edge(EdgePattern::Literal("top"), "subreddit_top") - .edge(EdgePattern::Literal("new"), "subreddit_new") - .edge(EdgePattern::Literal("comments"), "subreddit_comments") - .handler(|query, prefix, ctx| { - let subreddit = ctx.get("subreddit").cloned().unwrap_or_default(); - ParserAction::guide( - query.to_string(), - format!("What to sort in r/{}?", subreddit), - "Choose a category to begin sorting.".to_string(), - vec![ - GuideOption { - key: "hot".to_string(), - label: "Hot Posts".to_string(), - description: "Import and sort posts currently on the front page.".to_string(), - completion: format!("{}hot", prefix) - }, - GuideOption { - key: "top".to_string(), - label: "Top Posts".to_string(), - description: "Import and sort the highest-rated posts.".to_string(), - completion: format!("{}top", prefix) - }, - GuideOption { - key: "new".to_string(), - label: "New Posts".to_string(), - description: "Import and sort the newest posts.".to_string(), - completion: format!("{}new", prefix) - }, - GuideOption { - key: "comments".to_string(), - label: "All Comments".to_string(), - description: "Find the best comment across all imported threads.".to_string(), - completion: format!("{}comments/", prefix) - }, - ] - ) - }) - - - - .at("user_selection") - .edge(EdgePattern::Variable("username"), "user_profile") - .handler(|query, prefix, _ctx| { - ParserAction::guide( - query.to_string(), - "User Profile Sorting".to_string(), - "Enter a Reddit username to sort their content.".to_string(), - vec![ - GuideOption { - key: "popular".to_string(), - label: "Popular Users".to_string(), - description: "Browse well-known Reddit users.".to_string(), - completion: prefix.to_string(), - }, - ] - ) - }) - - // === Subreddit sort types === - .at("subreddit_hot") - .handler(|_query, _prefix, ctx| { - let subreddit = ctx.get("subreddit").cloned().unwrap_or_default(); - ParserAction::RenderEntityView { - ns: "reddit.subreddit".to_string(), - pk: subreddit, - } - }) - - .at("subreddit_top") - .handler(|_query, _prefix, ctx| { - let subreddit = ctx.get("subreddit").cloned().unwrap_or_default(); - ParserAction::RenderEntityView { - ns: "reddit.subreddit".to_string(), - pk: subreddit, - } - }) - - .at("subreddit_new") - .handler(|_query, _prefix, ctx| { - let subreddit = ctx.get("subreddit").cloned().unwrap_or_default(); - ParserAction::RenderEntityView { - ns: "reddit.subreddit".to_string(), - pk: subreddit, - } - }) - - .at("subreddit_comments") - .handler(|_query, _prefix, ctx| { - let subreddit = ctx.get("subreddit").cloned().unwrap_or_default(); - ParserAction::RenderEntityView { - ns: "reddit.subreddit".to_string(), - pk: subreddit, - } - }) - - // === User profile === - .at("user_profile") - .handler(|_query, _prefix, ctx| { - let username = ctx.get("username").cloned().unwrap_or_default(); - ParserAction::RenderEntityView { - ns: "reddit.user".to_string(), - pk: username, - } - }) - - // === Build the graph === - .build() -} - -// --- Public API --- - -static REDDIT_GRAPH: OnceLock = OnceLock::new(); - -fn reddit_graph() -> &'static Graph { - REDDIT_GRAPH.get_or_init(build_reddit_graph) -} - -/// Parse a query string and return a UI action -pub fn parse_reddit_url(query: &str) -> ParserAction { - reddit_graph().parse(query) } #[cfg(test)] mod tests { use super::*; - /// Represents a keystroke action - #[derive(Debug, Clone, PartialEq)] - enum KeyAction { - Type(String), // Type characters - Tab, // Press tab (accept completion) - - } - - /// Expected state after a keystroke - #[derive(Debug, Clone)] - enum ExpectedAction { - Suggestion { completion: String }, - ScrollingSuggestions { completions: Vec }, - Guide { title_contains: String }, - - RenderSubreddit { subreddit: String, sort: Option }, - RenderSubredditComments { subreddit: String }, - RenderUser { username: String }, - ResolveSubreddit { subreddit: String, prefix: String }, // New unified subreddit resolution - Error { error_type: String }, - Multiple { expected_actions: Vec }, // Multiple responses with specific expectations - MultipleAny, // Multiple responses (any sub-actions - legacy) - DbSuggestions { partial: String, prefix: String }, // Database-backed suggestions - - } - - impl ExpectedAction { - fn matches(&self, action: &ParserAction) -> bool { - match (self, action) { - (ExpectedAction::Suggestion { completion }, ParserAction::ShowSuggestions(data)) => { - data.suggestion.as_ref() - .map(|s| s.completion == *completion) - .unwrap_or(false) - } - (ExpectedAction::ScrollingSuggestions { completions }, ParserAction::ShowScrollingSuggestions { suggestions, .. }) => { - let actual_completions: Vec = suggestions.iter().map(|s| s.completion.clone()).collect(); - *completions == actual_completions - } - (ExpectedAction::Guide { title_contains }, ParserAction::ShowStaticGuide { title, .. }) => { - title.contains(title_contains) - } - (ExpectedAction::RenderSubreddit { subreddit, sort: _ }, - ParserAction::RenderEntityView { ns, pk }) => { - ns == "reddit.subreddit" && pk == subreddit - } - (ExpectedAction::RenderSubredditComments { subreddit }, - ParserAction::RenderEntityView { ns, pk }) => { - ns == "reddit.subreddit" && pk == subreddit - } - (ExpectedAction::RenderUser { username }, ParserAction::RenderEntityView { ns, pk }) => { - ns == "reddit.user" && pk == username - } - (ExpectedAction::ResolveSubreddit { subreddit, prefix }, - ParserAction::ResolveAndDisplaySubreddit { subreddit: s, prefix: p }) => { - s == subreddit && p == prefix - } - (ExpectedAction::Error { error_type }, ParserAction::ShowError(data)) => { - data.error_type == *error_type - } - (ExpectedAction::Multiple { expected_actions }, ParserAction::ShowMultiple { actions }) => { - // Check that all expected actions are present - if expected_actions.len() != actions.len() { - return false; - } - expected_actions.iter().zip(actions.iter()).all(|(expected, actual)| { - expected.matches(actual) - }) - } - (ExpectedAction::MultipleAny, ParserAction::ShowMultiple { .. }) => true, - (ExpectedAction::DbSuggestions { partial, prefix }, - ParserAction::SuggestSubredditsFromDb { partial: p, prefix: pr }) => { - p == partial && pr == prefix - } - - _ => false, - } - } - } - - /// Test helper to simulate a sequence of keystrokes - fn simulate_keystrokes(actions: Vec) -> Vec<(String, ParserAction)> { - let graph = build_reddit_graph(); - let mut current_text = String::new(); - let mut results = Vec::new(); - - for action in actions { - match action { - KeyAction::Type(text) => { - current_text.push_str(&text); - let result = graph.parse(¤t_text); - results.push((current_text.clone(), result)); - } - KeyAction::Tab => { - // Tab accepts the current suggestion if there is one - let result = graph.parse(¤t_text); - if let ParserAction::ShowSuggestions(ref data) = result { - if let Some(ref suggestion) = data.suggestion { - current_text = suggestion.completion.clone(); - let new_result = graph.parse(¤t_text); - results.push((current_text.clone(), new_result)); - } - } - } - - } - } - - results - } - - /// Test a flow using declarative (KeyAction, ExpectedAction) tuples - fn test_flow(name: &str, flow: Vec<(KeyAction, ExpectedAction)>) { - let graph = build_reddit_graph(); - let mut current_text = String::new(); - - println!("\n=== Flow: {} ===", name); - - for (i, (key_action, expected)) in flow.iter().enumerate() { - // Perform the keystroke - match key_action { - KeyAction::Type(text) => { - current_text.push_str(text); - } - KeyAction::Tab => { - // Tab accepts the current suggestion - let result = graph.parse(¤t_text); - if let ParserAction::ShowSuggestions(data) = result { - if let Some(suggestion) = &data.suggestion { - current_text = suggestion.completion.clone(); - } - } - } - } - - // Check the result - let actual_action = graph.parse(¤t_text); - - println!(" Step {}: {:?} -> '{}' -> {:?}", - i + 1, key_action, current_text, actual_action); - - assert!( - expected.matches(&actual_action), - "Flow '{}' failed at step {}\n Expected: {:?}\n Actual: {:?}\n Text: '{}'", - name, i + 1, expected, actual_action, current_text - ); - } - - println!("✓ Flow '{}' passed!", name); - } - - /// Legacy helper for backward compatibility (will be removed) - fn assert_flow( - name: &str, - actions: Vec, - expected_checks: Vec bool>>, - ) { - let results = simulate_keystrokes(actions); - - println!("\n=== Flow: {} ===", name); - for (i, (text, action)) in results.iter().enumerate() { - println!(" Step {}: '{}' -> {:?}", i + 1, text, action); - - if i < expected_checks.len() { - let check = &expected_checks[i]; - assert!( - check(text, action), - "Flow '{}' failed at step {} with text '{}' and action {:?}", - name, i + 1, text, action - ); - } - } - println!("✓ Flow '{}' passed!", name); - } - - #[test] - fn test_https_reddit_tab_flow() { - // Test: typing "https://reddit.com" and pressing tab should give "https://reddit.com/" - assert_flow( - "HTTPS Reddit with Tab", - vec![ - KeyAction::Type("https://reddit.com".to_string()), - KeyAction::Tab, - ], - vec![ - Box::new(|text, action| { - // After typing "https://reddit.com", should get a suggestion - text == "https://reddit.com" && matches!(action, ParserAction::ShowSuggestions(data) if - data.suggestion.as_ref().map(|s| s.completion == "https://reddit.com/").unwrap_or(false) - ) - }), - Box::new(|text, action| { - // After tab, should have "https://reddit.com/" and show guide - text == "https://reddit.com/" && matches!(action, ParserAction::ShowMultiple { .. }) - }), - ], - ); - } - - #[test] - fn test_quick_subreddit_flow() { - // Test: "r" -> TAB -> "rust" (direct alias flow) - assert_flow( - "Quick Subreddit Access", - vec![ - KeyAction::Type("r".to_string()), - KeyAction::Tab, - KeyAction::Type("rust".to_string()), - ], - vec![ - Box::new(|text, action| { - // "r" should suggest "r/" - text == "r" && matches!(action, ParserAction::ShowSuggestions(data) if - data.suggestion.as_ref().map(|s| s.completion == "r/").unwrap_or(false) - ) - }), - Box::new(|text, action| { - // After tab, should have "r/" and show subreddit selection - text == "r/" && matches!(action, ParserAction::ShowMultiple { .. }) - }), - Box::new(|text, action| { - // "r/rust" should resolve the subreddit - text == "r/rust" && matches!(action, ParserAction::ResolveAndDisplaySubreddit { subreddit, prefix } if subreddit == "rust" && prefix == "r/rust") - }), - ], - ); - } - - #[test] - fn test_progressive_completion_flow() { - // Test progressive typing: "r" -> "re" -> "red" -> "redd" -> "reddit" -> TAB - let progressive_actions = vec![ - KeyAction::Type("r".to_string()), - KeyAction::Type("e".to_string()), - KeyAction::Type("d".to_string()), - KeyAction::Type("d".to_string()), - KeyAction::Type("i".to_string()), - KeyAction::Type("t".to_string()), - KeyAction::Tab, - ]; - - let results = simulate_keystrokes(progressive_actions); - - println!("\n=== Progressive Completion Flow ==="); - for (i, (text, action)) in results.iter().enumerate() { - println!(" '{}' -> {:?}", text, action); - - // First step: "r" should suggest "r/" - if i == 0 && text == "r" { - match action { - ParserAction::ShowSuggestions(data) => { - assert_eq!( - data.suggestion.as_ref().unwrap().completion, - "r/", - "Should suggest r/ at 'r'" - ); - } - _ => panic!("Expected suggestion at 'r'"), - } - } - // Other steps before tab should suggest "reddit.com" - else if i > 0 && i < results.len() - 1 { - match action { - ParserAction::ShowSuggestions(data) => { - assert_eq!( - data.suggestion.as_ref().unwrap().completion, - "reddit.com", - "Should suggest reddit.com at '{}'", text - ); - } - _ => panic!("Expected suggestion at '{}'", text), - } - } - } - - // After tab, should have "reddit.com" - let (final_text, _) = results.last().unwrap(); - assert_eq!(final_text, "reddit.com"); - println!("✓ Progressive completion flow passed!"); - } - #[test] - fn test_subreddit_sort_flow() { - // Test navigating to a subreddit and choosing a sort option - assert_flow( - "Subreddit Sort Navigation", - vec![ - KeyAction::Type("reddit.com/r/programming/hot".to_string()), - ], - vec![ - Box::new(|text, action| { - text == "reddit.com/r/programming/hot" && - matches!(action, ParserAction::RenderEntityView { ns, pk } - if ns == "reddit.subreddit" && pk == "programming") - }), - ], - ); + fn parses_short_path() { + assert_eq!(parse_reddit_url("r/rust").unwrap(), "rust"); } #[test] - fn test_user_profile_flow() { - // Test navigating to a user profile - assert_flow( - "User Profile Navigation", - vec![ - KeyAction::Type("reddit.com/u/spez".to_string()), - ], - vec![ - Box::new(|text, action| { - text == "reddit.com/u/spez" && - matches!(action, ParserAction::RenderEntityView { ns, pk } if ns == "reddit.user" && pk == "spez") - }), - ], - ); + fn parses_path_with_trailing_slash() { + assert_eq!(parse_reddit_url("r/rust/").unwrap(), "rust"); } #[test] - fn test_alias_shortcut_flow() { - // Test using the "r/" shortcut - now correctly goes directly to subreddit - assert_flow( - "Alias Shortcut", - vec![ - KeyAction::Type("r/".to_string()), - KeyAction::Type("technology".to_string()), - ], - vec![ - Box::new(|text, action| { - text == "r/" && matches!(action, ParserAction::ShowMultiple { .. }) - }), - Box::new(|text, action| { - text == "r/technology" && - matches!(action, ParserAction::ResolveAndDisplaySubreddit { subreddit, prefix } if subreddit == "technology" && prefix == "r/technology") - }), - ], + fn parses_full_url() { + assert_eq!( + parse_reddit_url("https://www.reddit.com/r/programming/hot").unwrap(), + "programming" ); } #[test] - fn test_www_prefix_flow() { - // Test with www prefix - assert_flow( - "WWW Prefix", - vec![ - KeyAction::Type("www.reddit.com".to_string()), - KeyAction::Tab, - ], - vec![ - Box::new(|text, action| { - // Should suggest adding slash - text == "www.reddit.com" && matches!(action, ParserAction::ShowSuggestions(data) if - data.suggestion.as_ref().map(|s| s.completion == "www.reddit.com/").unwrap_or(false) - ) - }), - Box::new(|text, action| { - // After tab, should show guide - text == "www.reddit.com/" && matches!(action, ParserAction::ShowMultiple { .. }) - }), - ], + fn parses_url_without_scheme() { + assert_eq!( + parse_reddit_url("reddit.com/r/AskReddit").unwrap(), + "askreddit" ); } #[test] - fn test_invalid_path_handling() { - // Test that invalid paths show errors - assert_flow( - "Invalid Path", - vec![ - KeyAction::Type("reddit.com/invalid/path".to_string()), - ], - vec![ - Box::new(|text, action| { - text == "reddit.com/invalid/path" && - matches!(action, ParserAction::ShowError(_)) - }), - ], - ); - } - - #[test] - fn test_complete_user_journey() { - // Test a complete user journey: type partial URL, tab complete, navigate to subreddit - assert_flow( - "Complete User Journey", - vec![ - KeyAction::Type("http".to_string()), - KeyAction::Type("s://r".to_string()), - KeyAction::Tab, - KeyAction::Type("/".to_string()), - KeyAction::Type("r/".to_string()), - KeyAction::Type("programming".to_string()), - KeyAction::Type("/".to_string()), - KeyAction::Type("top".to_string()), - ], - vec![ - Box::new(|text, action| { - // "http" should suggest "https://" - text == "http" && matches!(action, ParserAction::ShowSuggestions(data) if - data.suggestion.as_ref().map(|s| s.completion == "https://").unwrap_or(false) - ) - }), - Box::new(|text, action| { - // "https://r" should suggest "https://reddit.com" - text == "https://r" && matches!(action, ParserAction::ShowSuggestions(data) if - data.suggestion.as_ref().map(|s| s.completion == "https://reddit.com").unwrap_or(false) - ) - }), - Box::new(|text, action| { - // After tab, should have "https://reddit.com" - text == "https://reddit.com" && matches!(action, ParserAction::ShowSuggestions(_)) - }), - Box::new(|text, action| { - // "https://reddit.com/" should show guide - text == "https://reddit.com/" && matches!(action, ParserAction::ShowMultiple { .. }) - }), - Box::new(|text, action| { - // "https://reddit.com/r/" should show subreddit selection - text == "https://reddit.com/r/" && matches!(action, ParserAction::ShowMultiple { .. }) - }), - Box::new(|text, action| { - // "https://reddit.com/r/programming" should resolve subreddit - text == "https://reddit.com/r/programming" && - matches!(action, ParserAction::ResolveAndDisplaySubreddit { subreddit, prefix } - if subreddit == "programming" && prefix == "https://reddit.com/r/programming") - }), - Box::new(|text, action| { - // "https://reddit.com/r/programming/" should show sort options - text == "https://reddit.com/r/programming/" && - matches!(action, ParserAction::ShowStaticGuide { .. }) - }), - Box::new(|text, action| { - // "https://reddit.com/r/programming/top" should render entity view - text == "https://reddit.com/r/programming/top" && - matches!(action, ParserAction::RenderEntityView { ns, pk } - if ns == "reddit.subreddit" && pk == "programming") - }), - ], - ); - } - - #[test] - fn test_multiple_tab_completions() { - // Test multiple tab completions in sequence - let actions = vec![ - KeyAction::Type("h".to_string()), - KeyAction::Tab, // Complete to "https://" - KeyAction::Type("r".to_string()), - KeyAction::Tab, // Complete to "https://reddit.com" - KeyAction::Type("/".to_string()), - ]; - - let results = simulate_keystrokes(actions); - - println!("\n=== Multiple Tab Completions ==="); - for (i, (text, _action)) in results.iter().enumerate() { - println!(" Step {}: '{}'", i + 1, text); - } - - // Verify the final state - assert_eq!(results[1].0, "https://"); // After first tab - assert_eq!(results[3].0, "https://reddit.com"); // After second tab - assert_eq!(results[4].0, "https://reddit.com/"); // After typing / - - println!("✓ Multiple tab completions work correctly!"); - } - - // === CONVENIENCE MACROS FOR CLEANER TESTS === - - macro_rules! flow { - ($(($key:expr, $expected:expr)),* $(,)?) => { - vec![$(($key, $expected)),*] - }; - } - - macro_rules! type_text { - ($text:expr) => { - KeyAction::Type($text.to_string()) - }; - } - - macro_rules! suggests { - ($completion:expr) => { - ExpectedAction::Suggestion { completion: $completion.to_string() } - }; - } - - macro_rules! renders_subreddit { - ($subreddit:expr) => { - ExpectedAction::RenderSubreddit { subreddit: $subreddit.to_string(), sort: None } - }; - ($subreddit:expr, $sort:expr) => { - ExpectedAction::RenderSubreddit { - subreddit: $subreddit.to_string(), - sort: Some($sort.to_string()) - } - }; - } - - macro_rules! renders_subreddit_comments { - ($subreddit:expr) => { - ExpectedAction::RenderSubredditComments { subreddit: $subreddit.to_string() } - }; - } - - macro_rules! resolves_subreddit { - ($subreddit:expr, $prefix:expr) => { - ExpectedAction::ResolveSubreddit { - subreddit: $subreddit.to_string(), - prefix: $prefix.to_string() - } - }; - } - - macro_rules! shows_guide { - ($title_contains:expr) => { - ExpectedAction::Guide { title_contains: $title_contains.to_string() } - }; - } - - macro_rules! multiple { - ($($action:expr),* $(,)?) => { - ExpectedAction::Multiple { expected_actions: vec![$($action),*] } - }; - } - - macro_rules! scrolling_suggestions { - ($($completion:expr),* $(,)?) => { - ExpectedAction::ScrollingSuggestions { completions: vec![$($completion.to_string()),*] } - }; - } - - macro_rules! db_suggestions { - ($partial:expr, $prefix:expr) => { - ExpectedAction::DbSuggestions { partial: $partial.to_string(), prefix: $prefix.to_string() } - }; - } - - // === NEW DECLARATIVE TESTS === - - #[test] - fn test_declarative_https_tab_flow() { - test_flow("HTTPS Tab Completion", vec![ - (KeyAction::Type("https://reddit.com".to_string()), - ExpectedAction::Suggestion { completion: "https://reddit.com/".to_string() }), - (KeyAction::Tab, - multiple![ - scrolling_suggestions!("https://reddit.com/r/", "https://reddit.com/u/"), - shows_guide!("Welcome to Sorter") - ]), - ]); - } - - #[test] - fn test_declarative_quick_subreddit_flow() { - test_flow("Quick Subreddit Flow", vec![ - (KeyAction::Type("r".to_string()), - ExpectedAction::Suggestion { completion: "r/".to_string() }), - (KeyAction::Tab, - ExpectedAction::MultipleAny), // r/ shows subreddit selection - (KeyAction::Type("rust".to_string()), - ExpectedAction::ResolveSubreddit { subreddit: "rust".to_string(), prefix: "r/rust".to_string() }), - ]); - } - - #[test] - fn test_declarative_complete_journey() { - test_flow("Complete User Journey", vec![ - (KeyAction::Type("http".to_string()), - ExpectedAction::Suggestion { completion: "https://".to_string() }), - (KeyAction::Type("s://r".to_string()), - ExpectedAction::Suggestion { completion: "https://reddit.com".to_string() }), - (KeyAction::Tab, - ExpectedAction::Suggestion { completion: "https://reddit.com/".to_string() }), - (KeyAction::Type("/r/programming/top".to_string()), - ExpectedAction::RenderSubreddit { - subreddit: "programming".to_string(), - sort: Some("top".to_string()) - }), - ]); - } - - #[test] - fn test_declarative_user_profile() { - test_flow("User Profile Navigation", vec![ - (KeyAction::Type("reddit.com/u/spez".to_string()), - ExpectedAction::RenderUser { username: "spez".to_string() }), - ]); - } - - #[test] - fn test_declarative_error_handling() { - test_flow("Error Handling", vec![ - (KeyAction::Type("reddit.com/invalid/path".to_string()), - ExpectedAction::Error { error_type: "InvalidPath".to_string() }), - ]); - } - - #[test] - fn test_declarative_progressive_completion() { - test_flow("Progressive Completion", vec![ - (KeyAction::Type("r".to_string()), - ExpectedAction::Suggestion { completion: "r/".to_string() }), - (KeyAction::Type("e".to_string()), - ExpectedAction::Suggestion { completion: "reddit.com".to_string() }), - (KeyAction::Type("d".to_string()), - ExpectedAction::Suggestion { completion: "reddit.com".to_string() }), - (KeyAction::Type("dit".to_string()), - ExpectedAction::Suggestion { completion: "reddit.com".to_string() }), - (KeyAction::Tab, - ExpectedAction::Suggestion { completion: "reddit.com/".to_string() }), - ]); - } - - #[test] - fn test_declarative_subreddit_guide() { - test_flow("Subreddit Guide", vec![ - (KeyAction::Type("reddit.com/r/programming/".to_string()), - ExpectedAction::Guide { title_contains: "What to sort".to_string() }), - ]); - } - - #[test] - fn test_clean_macro_example() { - // This is what the tests can look like with macros! - test_flow("Clean Macro Example", flow![ - (type_text!("r"), suggests!("r/")), - (KeyAction::Tab, ExpectedAction::MultipleAny), - (type_text!("rust/hot"), renders_subreddit!("rust", "hot")), - ]); - } - - #[test] - fn test_unified_subreddit_resolution_flow() { - // This test demonstrates the new unified behavior: - // - r/programming (exact) → tries exact match first - // - r/pro (partial) → tries exact match, then suggestions + TAB completion - - test_flow("Unified Resolution: Exact Match", flow![ - (type_text!("r/programming"), resolves_subreddit!("programming", "r/programming")), - ]); - - test_flow("Unified Resolution: Partial Match", flow![ - (type_text!("r/pro"), resolves_subreddit!("pro", "r/pro")), - ]); - - // Both go through the same action type, but dispatcher handles them differently: - // - If "programming" exists in DB → shows EntityView immediately - // - If "pro" doesn't exist in DB → shows Multiple with: - // 1. Suggestions (for TAB completion to best match) - // 2. Selection (for clickable options including import) - } - - #[test] - fn test_tab_completion_workflow_unified() { - // This demonstrates the desired TAB completion behavior: - // r/pro + TAB → r/programming (if "programming" is the best DB match) - - // Note: This test shows the PARSER behavior. The actual TAB completion - // happens in the frontend when it receives the Multiple response containing - // both Suggestions (for TAB) and Selection (for click options). - - let graph = build_reddit_graph(); - - // 1. Parser generates unified action for partial input - match graph.parse("r/pro") { - ParserAction::ResolveAndDisplaySubreddit { subreddit, prefix } => { - assert_eq!(subreddit, "pro"); - assert_eq!(prefix, "r/pro"); - println!("✓ Parser correctly identifies 'r/pro' as subreddit resolution"); - } - _ => panic!("Expected ResolveAndDisplaySubreddit for 'r/pro'"), - } - - // 2. When dispatcher runs (in real app), it will return Multiple response with: - // - Suggestions: { completion: "r/programming" } (for TAB) - // - Selection: [ "Import r/pro", "r/programming", ... ] (for clicks) - - println!("✓ TAB completion workflow: r/pro → ResolveAndDisplaySubreddit → Multiple(Suggestions + Selection)"); - } - - #[test] - fn test_ultra_clean_user_journey() { - test_flow("Ultra Clean User Journey", flow![ - (type_text!("h"), suggests!("https://")), - (type_text!("ttps://r"), suggests!("https://reddit.com")), - (KeyAction::Tab, suggests!("https://reddit.com/")), - (type_text!("/u/spez"), ExpectedAction::RenderUser { username: "spez".to_string() }), - ]); - } - - // === TESTS FROM parser.tdsl === - - #[test] - fn test_tdsl_basic_reddit_progression() { - // Tests from parser.tdsl: r -> r/, then re -> reddit.com with all intermediate steps - test_flow("TDSL Basic Reddit Progression", flow![ - (type_text!("r"), suggests!("r/")), - (type_text!("e"), suggests!("reddit.com")), - (type_text!("d"), suggests!("reddit.com")), - (type_text!("d"), suggests!("reddit.com")), - (type_text!("i"), suggests!("reddit.com")), - (type_text!("t"), suggests!("reddit.com")), - (type_text!("."), suggests!("reddit.com")), - (type_text!("c"), suggests!("reddit.com")), - (type_text!("o"), suggests!("reddit.com")), - (KeyAction::Tab, suggests!("reddit.com/")), - ]); - } - - #[test] - fn test_tdsl_reddit_com_slash_infographic() { - // reddit.com/ -> {show infographic explaining that u (sort user posts) and r (sort subreddit posts)} - test_flow("TDSL Reddit.com/ Infographic", flow![ - (type_text!("reddit.com/"), ExpectedAction::MultipleAny), - ]); - } - - #[test] - fn test_tdsl_subreddit_selection() { - // reddit.com/r/ -> reddit.com/r/{randomly chose sub from list} - test_flow("TDSL Subreddit Selection", flow![ - (type_text!("reddit.com/r/"), ExpectedAction::MultipleAny), - ]); - } - - #[test] - fn test_tdsl_subreddit_view() { - // reddit.com/r/{sub} -> {resolve subreddit (exact match or suggestions)} - test_flow("TDSL Subreddit View", flow![ - (type_text!("reddit.com/r/programming"), resolves_subreddit!("programming", "reddit.com/r/programming")), - ]); - } - - #[test] - fn test_tdsl_subreddit_slash_infographic() { - // reddit.com/r/{sub}/ -> {show infographic or something} - test_flow("TDSL Subreddit Slash Infographic", flow![ - (type_text!("reddit.com/r/programming/"), shows_guide!("What to sort")), - ]); - } - - #[test] - fn test_tdsl_subreddit_comments() { - // reddit.com/r/{sub}/comments/{randomly chose comment from sql} - test_flow("TDSL Subreddit Comments", flow![ - (type_text!("reddit.com/r/programming/comments"), renders_subreddit_comments!("programming")), - ]); - } - - #[test] - fn test_tdsl_h_to_https() { - // h->https:// (show supported domains) - test_flow("TDSL H to HTTPS", flow![ - (type_text!("h"), suggests!("https://")), - ]); - } - - #[test] - fn test_tdsl_composable_https_reddit() { - // https://r->https://reddit.com/ - test_flow("TDSL Composable HTTPS Reddit", flow![ - (type_text!("https://r"), suggests!("https://reddit.com")), - ]); - } - - #[test] - fn test_tdsl_composable_https_www() { - // https://w->https://www. - test_flow("TDSL Composable HTTPS WWW", flow![ - (type_text!("https://w"), suggests!("https://www.")), - ]); - } - - #[test] - fn test_tdsl_composable_https_www_reddit() { - // https://www.r->https://www.reddit.com/ - test_flow("TDSL Composable HTTPS WWW Reddit", flow![ - (type_text!("https://www.r"), suggests!("https://www.reddit.com")), - ]); - } - - #[test] - fn test_tdsl_full_composable_chain() { - // Complete chain showing composability: h -> https:// -> https://www.reddit.com - test_flow("TDSL Full Composable Chain Step 1", flow![ - (type_text!("h"), suggests!("https://")), - ]); - - test_flow("TDSL Full Composable Chain Step 2", flow![ - (type_text!("https://w"), suggests!("https://www.")), - ]); - - test_flow("TDSL Full Composable Chain Step 3", flow![ - (type_text!("https://www.r"), suggests!("https://www.reddit.com")), - ]); - } - - #[test] - fn test_tdsl_progressive_reddit_paths() { - // Test various reddit paths work as expected - test_flow("TDSL Progressive Reddit Paths", flow![ - (type_text!("reddit.com"), suggests!("reddit.com/")), - (KeyAction::Tab, ExpectedAction::MultipleAny), - ]); - } - - #[test] - fn test_tdsl_subreddit_sorting_options() { - // Test that subreddit sorting options work as described - test_flow("TDSL Subreddit Sorting", flow![ - (type_text!("reddit.com/r/rust/hot"), renders_subreddit!("rust", "hot")), - ]); - - test_flow("TDSL Subreddit Top", flow![ - (type_text!("reddit.com/r/rust/top"), renders_subreddit!("rust", "top")), - ]); - - test_flow("TDSL Subreddit New", flow![ - (type_text!("reddit.com/r/rust/new"), renders_subreddit!("rust", "new")), - ]); - } - - #[test] - fn test_tdsl_all_reddit_prefixes() { - // Test all the prefixes mentioned in parser.tdsl work - // "r" now suggests "r/", all others suggest "reddit.com" - let prefixes_to_reddit = vec!["re", "red", "redd", "reddi", "reddit", "reddit.", "reddit.c", "reddit.co"]; - - // Test "r" separately since it now suggests "r/" - test_flow("TDSL Prefix: r", flow![ - (type_text!("r"), suggests!("r/")), - ]); - - for prefix in prefixes_to_reddit { - test_flow(&format!("TDSL Prefix: {}", prefix), flow![ - (type_text!(prefix), suggests!("reddit.com")), - ]); - } - } - - #[test] - fn test_tdsl_protocol_combinations() { - // Test various protocol combinations from parser.tdsl - let test_cases = vec![ - ("http://r", "http://reddit.com"), - ("https://r", "https://reddit.com"), - ("www.r", "www.reddit.com"), - ("https://www.r", "https://www.reddit.com"), - ("http://www.r", "http://www.reddit.com"), - ]; - - for (input, expected) in test_cases { - test_flow(&format!("TDSL Protocol: {}", input), flow![ - (type_text!(input), suggests!(expected)), - ]); - } + fn rejects_empty() { + assert!(parse_reddit_url("").is_err()); + assert!(parse_reddit_url(" ").is_err()); } #[test] - fn test_tdsl_edge_case_completions() { - // Test edge cases mentioned in parser.tdsl - test_flow("TDSL Reddit.com completion", flow![ - (type_text!("reddit.com"), suggests!("reddit.com/")), - ]); - - // Test that typing full reddit.com suggests the slash - test_flow("TDSL Full domain completion", flow![ - (type_text!("reddit.com"), suggests!("reddit.com/")), - (KeyAction::Tab, ExpectedAction::MultipleAny), - ]); - } - - #[test] - fn test_real_user_trace_2025_08_07_fixed() { - // Based on actual WebSocket trace from 2025-08-07T01:35:46Z - // This test shows the CORRECT behavior after fixing the protocol preservation bug - test_flow("Real User Trace: Progressive Typing with Tab Completions (Fixed)", flow![ - // User started typing "h" - (type_text!("h"), suggests!("https://")), - - // User continued to "ht" - (type_text!("t"), suggests!("https://")), - - // User finished typing "https://" (trace shows full protocol) - (type_text!("tps://"), suggests!("https://")), - - // User started typing "r" after protocol - (type_text!("r"), suggests!("https://reddit.com")), - - // User continued typing "re" - (type_text!("e"), suggests!("https://reddit.com")), - - // User typed out or completed "https://reddit.com" - (type_text!("ddit.com"), suggests!("https://reddit.com/")), - - // User accepted completion to "https://reddit.com/" - // NOW suggestions should preserve the https:// protocol - (KeyAction::Tab, multiple![ - scrolling_suggestions!("https://reddit.com/r/", "https://reddit.com/u/"), // This is the key fix! - shows_guide!("Welcome to Sorter") - ]), - ]); - } - - #[test] - fn test_protocol_preservation_bug_fix() { - // This test specifically verifies the fix for the protocol preservation bug - test_flow("Protocol Preservation: HTTPS Reddit Homepage", flow![ - (type_text!("https://reddit.com/"), multiple![ - scrolling_suggestions!("https://reddit.com/r/", "https://reddit.com/u/"), // Should preserve https:// - shows_guide!("Welcome to Sorter") - ]), - ]); - - // Test that subreddit selection preserves protocol - test_flow("Protocol Preservation: HTTPS Subreddit Selection", flow![ - (type_text!("https://reddit.com/r/"), multiple![ - scrolling_suggestions!("https://reddit.com/r/programming", "https://reddit.com/r/askreddit", "https://reddit.com/r/aww", "https://reddit.com/r/rust", "https://reddit.com/r/webdev"), // Should preserve https:// - db_suggestions!("https://reddit.com/r/", "https://reddit.com/r/") - ]), - ]); - - // Test with different protocols - test_flow("Protocol Preservation: HTTP", flow![ - (type_text!("http://reddit.com/"), multiple![ - scrolling_suggestions!("http://reddit.com/r/", "http://reddit.com/u/"), // Should preserve http:// - shows_guide!("Welcome to Sorter") - ]), - ]); - - test_flow("Protocol Preservation: WWW", flow![ - (type_text!("www.reddit.com/"), multiple![ - scrolling_suggestions!("www.reddit.com/r/", "www.reddit.com/u/"), // Should preserve www. - shows_guide!("Welcome to Sorter") - ]), - ]); - - // Test that plain reddit.com still works - test_flow("Protocol Preservation: Plain Domain", flow![ - (type_text!("reddit.com/"), multiple![ - scrolling_suggestions!("reddit.com/r/", "reddit.com/u/"), // No protocol prefix - shows_guide!("Welcome to Sorter") - ]), - ]); - } - - #[test] - fn test_user_navigation_pattern() { - // Models how users actually navigate: type → tab → click suggestion → end up at destination - // This captures the "jump" from https://reddit.com/ to reddit.com/r/ seen in the trace - test_flow("User Navigation: Protocol to Domain", flow![ - // User types and gets to homepage - (type_text!("https://reddit.com/"), ExpectedAction::MultipleAny), - ]); - - // Then they navigate (perhaps clicking a suggestion) to subreddit selection - test_flow("User Navigation: Click to Subreddit", flow![ - // This is where they ended up - the suggestion in the guide probably said "reddit.com/r/" - (type_text!("reddit.com/r/"), ExpectedAction::MultipleAny), - ]); - } - - #[test] - fn test_progressive_typing_pattern() { - // Based on the trace pattern - users often type character by character - // This tests the exact sequence of suggestions they would see - test_flow("Progressive Typing Pattern", flow![ - (type_text!("h"), suggests!("https://")), - (type_text!("t"), suggests!("https://")), // Still suggests https:// - (type_text!("t"), suggests!("https://")), // ht -> htt, still suggests https:// - (type_text!("p"), suggests!("https://")), // http should still suggest https:// - (type_text!("s"), suggests!("https://")), // https should suggest https:// - (type_text!("://"), suggests!("https://")), // Even complete protocol still suggests itself - ]); - } - - #[test] - fn test_tab_completion_workflow() { - // Test what happens when user uses tab completions strategically - test_flow("Strategic Tab Completion Workflow", flow![ - // Start typing, get suggestion - (type_text!("h"), suggests!("https://")), - - // Accept suggestion with tab - this should move us to "https://" - (KeyAction::Tab, suggests!("https://")), // After tab, we're at "https://" which still suggests itself - - // Start typing reddit - (type_text!("r"), suggests!("https://reddit.com")), - - // Accept reddit suggestion - (KeyAction::Tab, suggests!("https://reddit.com/")), - - // Accept final suggestion to get to homepage - (KeyAction::Tab, ExpectedAction::MultipleAny), - ]); - } - - // Keep the original tests as well - #[test] - fn test_reddit_prefix_autocomplete() { - let graph = build_reddit_graph(); - - // "r" now suggests "r/", others suggest "reddit.com" - match graph.parse("r") { - ParserAction::ShowSuggestions(data) => { - assert_eq!(data.suggestion.as_ref().unwrap().completion, "r/"); - println!("✓ 'r' → r/"); - } - _ => panic!("Expected suggestion for 'r'"), - } - - // Test other prefixes that should suggest "reddit.com" - let prefixes = vec!["re", "red", "redd", "reddi", "reddit", "reddit.", "reddit.c", "reddit.co"]; - - for prefix in prefixes { - match graph.parse(prefix) { - ParserAction::ShowSuggestions(data) => { - assert_eq!(data.suggestion.as_ref().unwrap().completion, "reddit.com"); - println!("✓ '{}' → reddit.com", prefix); - } - _ => panic!("Expected suggestion for '{}'", prefix), - } - } - } - - #[test] - fn test_protocol_composition() { - let graph = build_reddit_graph(); - - // Test protocol + reddit compositions - let tests = vec![ - ("https://r", "https://reddit.com"), - ("https://re", "https://reddit.com"), - ("https://reddit", "https://reddit.com"), - ("https://www.r", "https://www.reddit.com"), - ("https://www.reddit", "https://www.reddit.com"), - ("http://r", "http://reddit.com"), - ("www.r", "www.reddit.com"), - ]; - - for (input, expected) in tests { - match graph.parse(input) { - ParserAction::ShowSuggestions(data) => { - assert_eq!(data.suggestion.as_ref().unwrap().completion, expected); - println!("✓ '{}' → {}", input, expected); - } - _ => panic!("Expected suggestion for '{}'", input), - } - } - } - - #[test] - fn test_alias_and_full_paths() { - let graph = build_reddit_graph(); - - // reddit.com/ should show guide, r/ should show subreddit selection - match graph.parse("reddit.com/") { - ParserAction::ShowMultiple { actions } => { - let has_guide = actions.iter() - .any(|a| matches!(a, ParserAction::ShowStaticGuide { .. })); - assert!(has_guide, "Path 'reddit.com/' should show guide"); - println!("✓ 'reddit.com/' shows Reddit root guide"); - } - _ => panic!("Expected Multiple action for 'reddit.com/'"), - } - - match graph.parse("r/") { - ParserAction::ShowMultiple { actions } => { - let has_scrolling_suggestions = actions.iter() - .any(|a| matches!(a, ParserAction::ShowScrollingSuggestions { .. })); - let has_db_suggestions = actions.iter() - .any(|a| matches!(a, ParserAction::SuggestSubredditsFromDb { .. })); - assert!(has_scrolling_suggestions && has_db_suggestions, - "Path 'r/' should show scrolling suggestions and DB suggestions"); - println!("✓ 'r/' shows subreddit selection"); - } - _ => panic!("Expected Multiple action for 'r/'"), - } - } - - #[test] - fn test_deep_navigation() { - let graph = build_reddit_graph(); - - // Test navigation to subreddit - now uses ResolveAndDisplaySubreddit - match graph.parse("reddit.com/r/rust") { - ParserAction::ResolveAndDisplaySubreddit { subreddit, prefix } => { - assert_eq!(subreddit, "rust"); - assert_eq!(prefix, "reddit.com/r/rust"); - println!("✓ reddit.com/r/rust recognized"); - } - _ => panic!("Expected ResolveAndDisplaySubreddit for subreddit"), - } - - // Test with alias (r/ goes directly to subreddit, no double r/) - match graph.parse("r/programming") { - ParserAction::ResolveAndDisplaySubreddit { subreddit, prefix } => { - assert_eq!(subreddit, "programming"); - assert_eq!(prefix, "r/programming"); - println!("✓ r/programming (alias) recognized"); - } - _ => panic!("Expected ResolveAndDisplaySubreddit for subreddit via alias"), - } + fn rejects_garbage() { + assert!(parse_reddit_url("hello world").is_err()); } } diff --git a/server/src/parser_action.rs b/server/src/parser_action.rs deleted file mode 100644 index 2b98556b562d548ce231a726f9e704f595e8f4a1..0000000000000000000000000000000000000000 --- a/server/src/parser_action.rs +++ /dev/null @@ -1,124 +0,0 @@ -//! Parser output actions — graph handlers return these; HTML render turns them into markup. - -#[derive(Debug, Clone, PartialEq)] -pub struct Suggestion { - pub text: String, - pub completion: String, - pub description: Option, - pub score: f64, -} - -#[derive(Debug, Clone, PartialEq)] -pub struct ScrollingSuggestion { - pub completion: String, -} - -#[derive(Debug, Clone, PartialEq)] -pub struct GuideOption { - pub key: String, - pub label: String, - pub description: String, - pub completion: String, -} - -#[derive(Debug, Clone, PartialEq)] -pub struct SuggestionsData { - pub query: String, - pub suggestion: Option, -} - -#[derive(Debug, Clone, PartialEq)] -pub struct ErrorData { - pub error_type: String, - pub message: String, -} - -#[derive(Debug, Clone, PartialEq)] -pub enum ParserAction { - ShowSuggestions(SuggestionsData), - ShowScrollingSuggestions { - query: String, - suggestions: Vec, - interval_ms: u64, - r#loop: bool, - }, - ShowStaticGuide { - query: String, - title: String, - subtitle: String, - options: Vec, - }, - ShowMultiple { - actions: Vec, - }, - ShowError(ErrorData), - SuggestSubredditsFromDb { - partial: String, - prefix: String, - }, - ResolveAndDisplaySubreddit { - subreddit: String, - prefix: String, - }, - RenderEntityView { - ns: String, - pk: String, - }, -} - -impl ParserAction { - pub fn suggest(query: String, suggestion: Option) -> Self { - Self::ShowSuggestions(SuggestionsData { query, suggestion }) - } - - pub fn error(error_type: String, message: String) -> Self { - Self::ShowError(ErrorData { - error_type, - message, - }) - } - - pub fn multiple(actions: Vec) -> Self { - Self::ShowMultiple { actions } - } - - pub fn scrolling_suggestions( - query: String, - suggestions: Vec, - interval_ms: u64, - r#loop: bool, - ) -> Self { - Self::ShowScrollingSuggestions { - query, - suggestions, - interval_ms, - r#loop, - } - } - - pub fn guide( - query: String, - title: String, - subtitle: String, - options: Vec, - ) -> Self { - Self::ShowStaticGuide { - query, - title, - subtitle, - options, - } - } - - /// Primary tab-completion string, if any. - pub fn primary_completion(&self) -> Option<&str> { - match self { - Self::ShowSuggestions(data) => data - .suggestion - .as_ref() - .map(|s| s.completion.as_str()), - Self::ShowMultiple { actions } => actions.iter().find_map(|a| a.primary_completion()), - _ => None, - } - } -} diff --git a/server/src/parser_render.rs b/server/src/parser_render.rs index 05e4f519cc3832392282b319272b61c104eaabf6..f2341afe21476b689a536137798d97277211a962 100644 --- a/server/src/parser_render.rs +++ b/server/src/parser_render.rs @@ -2,12 +2,9 @@ use maud::{html, Markup}; use crate::{ form_template::template_json_compact, - parser_action::{GuideOption, ParserAction, ScrollingSuggestion, Suggestion}, ui_action::UI_RPC_FIELD, }; -const SAMPLE_SUBREDDITS: &[&str] = &["programming", "askreddit", "rust", "webdev", "aww"]; - fn parse_query_rpc_template() -> String { template_json_compact(&serde_json::json!({ "action": "parse_query", @@ -16,193 +13,32 @@ fn parse_query_rpc_template() -> String { .expect("parse_query rpc template") } -fn completion_button(completion: &str, label: &str, primary: bool) -> Markup { - let class = if primary { - "parser-completion parser-suggestion-primary btn-link" - } else { - "parser-completion btn-link" - }; - html! { - button - type="button" - class=(class) - data-completion=(completion) - title="Use this path" { - (label) - } - } -} - -fn render_suggestion(s: &Suggestion) -> Markup { - let desc = s - .description - .as_deref() - .unwrap_or("Tab to complete"); - html! { - p class="parser-suggestion" { - (completion_button(&s.completion, &s.completion, true)) - span class="muted small" { " — " (desc) } - } - } -} - -fn render_scrolling(suggestions: &[ScrollingSuggestion]) -> Markup { - html! { - div class="parser-scrolling muted small" { - p { "Examples:" } - ul class="parser-scroll-list" { - @for s in suggestions { - li { (completion_button(&s.completion, &s.completion, false)) } - } - } - } - } -} - -fn render_guide(title: &str, subtitle: &str, options: &[GuideOption]) -> Markup { - html! { - div class="parser-guide" { - h3 { (title) } - p class="muted" { (subtitle) } - ul class="parser-guide-list" { - @for opt in options { - li { - strong { (opt.key) ": " } - (completion_button(&opt.completion, &opt.label, false)) - span class="muted small" { " — " (opt.description) } - } - } - } - } - } -} - -fn render_db_subs(partial: &str, prefix: &str) -> Markup { - let needle = partial.to_lowercase(); - let matches: Vec<_> = SAMPLE_SUBREDDITS - .iter() - .filter(|s| s.contains(&needle) || prefix.ends_with('/') && needle.is_empty()) - .take(6) - .collect(); - html! { - @if !matches.is_empty() { - div class="parser-db-subs muted small" { - p { "Subreddits:" } - ul { - @for sub in matches { - @let completion = format!("{prefix}{sub}"); - li { (completion_button(&completion, &format!("r/{sub}"), false)) } - } - } - } - } - } -} - -fn render_action(action: &ParserAction) -> Markup { - match action { - ParserAction::ShowSuggestions(data) => html! { - div class="parser-result parser-suggestions" { - @if let Some(s) = &data.suggestion { - (render_suggestion(s)) - } @else { - p class="muted" { "No completion" } - } - } - }, - ParserAction::ShowScrollingSuggestions { suggestions, .. } => { - render_scrolling(suggestions) - } - ParserAction::ShowStaticGuide { - title, - subtitle, - options, - .. - } => render_guide(title, subtitle, options), - ParserAction::ShowMultiple { actions } => html! { - div class="parser-multiple" { - @for a in actions { - (render_action(a)) - } - } - }, - ParserAction::ShowError(data) => html! { - p class="parser-error muted" { - strong { (data.error_type) ": " } - (data.message) - } - }, - ParserAction::SuggestSubredditsFromDb { partial, prefix } => { - render_db_subs(partial, prefix) - } - ParserAction::ResolveAndDisplaySubreddit { subreddit, prefix } => html! { - div class="parser-resolve" { - p { - "Subreddit " - strong { "r/" (subreddit) } - @if subreddit.len() <= 3 { - span class="muted small" { " (partial — tab or pick a match)" } - } - } - p { - a class="parser-rank-link" - href=(format!("/?sub={subreddit}")) { - "Rank r/" (subreddit) " →" - } - } - (render_db_subs(subreddit, prefix)) - } - }, - ParserAction::RenderEntityView { ns, pk } => html! { - div class="parser-entity" { - p { - "Would open " - code { (ns) "/" (pk) } - } - } - }, - } -} - -/// Parser output panel (inner content for `#parser-panel`). -pub fn parser_panel(query: &str, action: &ParserAction) -> Markup { +/// Navigate panel: paste a Reddit URL and click Go. +pub fn navigate_panel(query: &str, error: Option<&str>) -> Markup { html! { section id="parser-panel" class="demo-panel" { h2 { "Navigate" } p class="muted small" { - "Type a Reddit path — " - code { "r/rust" } - ", " - code { "reddit.com/r/programming/hot" } - ", etc. Tab completes; each keystroke posts " - code { "__rpc__" } - " to " - code { "/ui" } - "." + "Paste a Reddit URL or " + code { "r/subreddit" } + " path, then click Go to rank that subreddit." } form method="post" action="/ui" id="parser-form" { - input - type="text" + textarea name="query" id="parser-input" - value=(query) - placeholder="r/ or reddit.com/…" + rows="3" + placeholder="https://reddit.com/r/rust or r/rust" autocomplete="off" - spellcheck="false"; + spellcheck="false" { + (query) + } input type="hidden" name=(UI_RPC_FIELD) value=(parse_query_rpc_template()); + button type="submit" class="btn-primary" { "Go" } } - div id="parser-output" { - @if query.is_empty() { - p class="muted" { "Start typing…" } - } @else { - (render_action(action)) - } + @if let Some(msg) = error { + p class="parser-error muted" { (msg) } } } } } - -/// Wrap panel HTML for Idiomorph (morph `#parser-panel` only). -pub fn parser_panel_morph(query: &str, action: &ParserAction) -> Markup { - parser_panel(query, action) -} diff --git a/server/src/ui_action.rs b/server/src/ui_action.rs index 8c357b133599b798a9c06a77a826bcef05a3a407..0e030b3448b8e45acbe49d2de47ea26372445c54 100644 --- a/server/src/ui_action.rs +++ b/server/src/ui_action.rs @@ -22,7 +22,7 @@ pub enum HtmlUiAction { #[serde(default)] scope: String, }, - /// Parse address-bar query via Reddit transition graph; morph `#parser-panel`. + /// Parse pasted Reddit URL/path; redirect to subreddit ranking on success. ParseQuery { query: String, }, diff --git a/server/static/sorter_ui.js b/server/static/sorter_ui.js index 1e5c3c13d6f7d8f180a4126ba25b44e9801cdb5d..9c475d8e076ba527aaf7a25c177d0d651bd46ca9 100644 --- a/server/static/sorter_ui.js +++ b/server/static/sorter_ui.js @@ -19,39 +19,6 @@ }).then(evalJs); } - // Parser responses race: a slow response for an earlier keystroke can arrive - // after a newer one and clobber the panel. Tag each request with a monotonic - // sequence number and only apply a response if it is newer than the last one - // applied, so stale (superseded) responses are discarded. - var parserSeq = 0; - var parserApplied = 0; - - function postParserForm(form) { - var mySeq = ++parserSeq; - return fetch(form.action, { - method: 'POST', - body: new URLSearchParams(new FormData(form)), - headers: { 'Content-Type': 'application/x-www-form-urlencoded' }, - credentials: 'same-origin', - }).then(function (resp) { - return resp.text(); - }).then(function (js) { - if (mySeq <= parserApplied) return; - parserApplied = mySeq; - evalJs(js); - }); - } - - var parserTimer = null; - - function scheduleParserInput(input) { - if (parserTimer) clearTimeout(parserTimer); - parserTimer = setTimeout(function () { - var form = document.getElementById('parser-form'); - if (form) postParserForm(form); - }, 120); - } - function initSorterUi() { document.addEventListener('submit', async function (e) { var f = e.target; @@ -67,38 +34,6 @@ if (firstField) firstField.focus(); } }); - - document.addEventListener('input', function (e) { - if (e.target && e.target.id === 'parser-input') { - scheduleParserInput(e.target); - } - }); - - document.addEventListener('keydown', function (e) { - if (!e.target || e.target.id !== 'parser-input') return; - if (e.key !== 'Tab') return; - var completion = - e.target.dataset.completion || - (function () { - var btn = document.querySelector('#parser-output .parser-suggestion-primary'); - return btn && btn.getAttribute('data-completion'); - })(); - if (!completion) return; - e.preventDefault(); - e.target.value = completion; - scheduleParserInput(e.target); - }); - - document.addEventListener('click', function (e) { - var btn = e.target.closest('.parser-completion'); - if (!btn) return; - var input = document.getElementById('parser-input'); - if (!input) return; - var completion = btn.getAttribute('data-completion'); - if (!completion) return; - input.value = completion; - scheduleParserInput(input); - }); } if (document.readyState === 'loading') { diff --git a/server/static/theme_default.css b/server/static/theme_default.css index c5fc9e7abf865ed689f58ff0763d3656d73e35f5..6ad0ac712bbc840bedee60f613794de9385fbbd1 100644 --- a/server/static/theme_default.css +++ b/server/static/theme_default.css @@ -129,34 +129,12 @@ code { background: var(--bg); color: var(--fg); font-size: 1rem; + font-family: inherit; + resize: vertical; } -.btn-link { - background: none; - border: none; - padding: 0; - color: var(--accent); - cursor: pointer; - font: inherit; -} - -.btn-link:hover { - text-decoration: underline; -} - -.parser-rank-link { - display: inline-block; - margin: 0.5rem 0; - padding: 0.4rem 0.8rem; - background: var(--accent); - color: #0f1115; - border-radius: 4px; - font-weight: 600; - text-decoration: none; -} - -.parser-rank-link:hover { - filter: brightness(1.1); +#parser-form .btn-primary { + margin-top: 0.5rem; } .scope-name { diff --git a/server/tests/integration_ui.rs b/server/tests/integration_ui.rs index 5112cb79ffe6e53df3d2432646770fc316cf4787..afeeee24e32f2d7f1d852ca96ac799fac8bce665 100644 --- a/server/tests/integration_ui.rs +++ b/server/tests/integration_ui.rs @@ -71,11 +71,11 @@ async fn post_ui_record_vote_morphs_ranking_and_persists() { } #[tokio::test] -async fn post_ui_parse_query_morphs_parser_panel() { +async fn post_ui_parse_query_redirects_to_subreddit() { let (addr, _tmp) = start_test_server().await; let rpc = serde_json::json!({ "action": "parse_query", - "query": "r" + "query": "r/rust" }) .to_string(); let mut form = HashMap::new(); @@ -92,7 +92,6 @@ async fn post_ui_parse_query_morphs_parser_panel() { .await .unwrap(); - assert!(body.contains("Idiomorph.morph")); - assert!(body.contains("parser-panel")); - assert!(body.contains("r/")); + assert!(body.contains("window.location.href")); + assert!(body.contains("/?sub=rust")); } diff --git a/test/parser_race.clj b/test/parser_race.clj deleted file mode 100644 index 23783837f8192ff24a556b73bdd670dc63b606cb..0000000000000000000000000000000000000000 --- a/test/parser_race.clj +++ /dev/null @@ -1,185 +0,0 @@ -(ns test.parser-race - "Browser test (spel / Playwright) for the parser search-box response race. - - The search box debounces keystrokes, POSTs each to `/ui`, and eval()s the - returned JS, which morphs `#parser-panel` (input value + `#parser-output`). - If responses are applied in arrival order with no ordering guard, a slow - response for an *earlier* query can land after a newer one and clobber it. - - To force the race deterministically without touching the Rust server, the - browser talks to a small in-process reverse proxy that injects an asymmetric - per-query delay: the earlier query (`r/rust`) is delayed far longer than the - later query (`r/aww`). The later query therefore renders first, then the - stale earlier response arrives. Correct behavior: the panel reflects the - *latest* query the user typed (`r/aww`)." - (:require [babashka.process :as process] - [clojure.java.io :as io] - [clojure.string :as str] - [clojure.test :refer [deftest is testing]] - [com.blockether.spel.assertions :as assert] - [com.blockether.spel.core :as core] - [com.blockether.spel.locator :as locator] - [com.blockether.spel.page :as page]) - (:import [com.sun.net.httpserver HttpServer HttpHandler] - [java.io ByteArrayOutputStream] - [java.net InetSocketAddress URI URLDecoder] - [java.net.http HttpClient HttpClient$Version HttpRequest - HttpRequest$BodyPublishers HttpResponse$BodyHandlers] - [java.nio.charset StandardCharsets] - [java.util.concurrent Executors])) - -(def ^:private slow-query "r/rust") -(def ^:private fast-query "r/aww") -(def ^:private slow-delay-ms 800) - -(defn- repo-root [] - (.getCanonicalPath (io/file (System/getProperty "user.dir")))) - -(defn- pick-port [] - (with-open [s (java.net.ServerSocket. 0)] - (.getLocalPort s))) - -(defn- wait-health [base-url ms] - (let [deadline (+ (System/currentTimeMillis) ms) - url (str base-url "/healthz")] - (loop [] - (let [resp (try - (process/shell {:out :string :err :string} "curl" "-sf" url) - (catch Exception _ nil))] - (if (and resp (zero? (:exit resp)) (= "ok" (str/trim (:out resp "")))) - true - (if (< (System/currentTimeMillis) deadline) - (do (Thread/sleep 200) (recur)) - false)))))) - -(defn- start-server - "Builds the release binary and starts it on a random port. Returns a map with - :proc and :base." - [root] - (is (zero? (:exit (process/shell {:dir root} - "cargo" "build" "--release" "--package" "sorter2-server"))) - "release build succeeds") - (let [bin (str root "/target/release/sorter2-server")] - (is (.exists (io/file bin)) "binary exists") - (let [data-dir (.getAbsolutePath - (doto (io/file (System/getProperty "java.io.tmpdir") - (str "sorter2-race-" (System/currentTimeMillis))) - (.mkdirs))) - port (pick-port) - base (str "http://127.0.0.1:" port) - proc (process/process {:dir root - :env {"SORTER2_DATA_DIR" data-dir - "SORTER2_EVENT_LOG" (str data-dir "/events.jsonl") - "PORT" (str port)} - :out :string - :err :string} - bin)] - {:proc proc :base base}))) - -(defn- read-all-bytes ^bytes [in] - (let [bos (ByteArrayOutputStream.)] - (io/copy in bos) - (.toByteArray bos))) - -(defn- ui-query - "Extracts the decoded `query` form field from a urlencoded body, or nil." - [body-str] - (some-> (re-find #"(?:^|&)query=([^&]*)" body-str) - second - (URLDecoder/decode "UTF-8"))) - -(defn- start-proxy - "Reverse proxy to `upstream` that sleeps `(delay-fn method path body-str)` ms - before forwarding each request. Runs handlers on a thread pool so concurrent - requests are delayed independently (the race needs out-of-order arrival). - Returns a map with :server and :base." - [upstream delay-fn] - (let [client (-> (HttpClient/newBuilder) - (.version HttpClient$Version/HTTP_1_1) - (.build)) - server (HttpServer/create (InetSocketAddress. "127.0.0.1" 0) 0) - handler (reify HttpHandler - (handle [_ ex] - (try - (let [method (.getRequestMethod ex) - uri (.getRequestURI ex) - path (.getRawPath uri) - query (.getRawQuery uri) - req-body (read-all-bytes (.getRequestBody ex)) - body-str (String. req-body StandardCharsets/UTF_8) - delay-ms (delay-fn method path body-str)] - (when (pos? delay-ms) - (Thread/sleep (long delay-ms))) - (let [target (str upstream path (when query (str "?" query))) - builder (doto (HttpRequest/newBuilder) - (.uri (URI/create target))) - ct (.getFirst (.getRequestHeaders ex) "Content-Type") - _ (when ct (.header builder "Content-Type" ct)) - publisher (if (zero? (alength req-body)) - (HttpRequest$BodyPublishers/noBody) - (HttpRequest$BodyPublishers/ofByteArray req-body)) - _ (.method builder method publisher) - resp (.send client (.build builder) - (HttpResponse$BodyHandlers/ofByteArray)) - resp-body (.body resp) - resp-ct (-> (.headers resp) - (.firstValue "content-type") - (.orElse nil))] - (when resp-ct - (.set (.getResponseHeaders ex) "Content-Type" resp-ct)) - (.sendResponseHeaders ex (.statusCode resp) (alength resp-body)) - (with-open [os (.getResponseBody ex)] - (.write os resp-body)))) - (catch Throwable t - (let [msg (.getBytes (str "proxy error: " (.getMessage t)) - StandardCharsets/UTF_8)] - (try - (.sendResponseHeaders ex 500 (alength msg)) - (with-open [os (.getResponseBody ex)] - (.write os msg)) - (catch Throwable _ nil)))) - (finally - (.close ex)))))] - (.createContext server "/" handler) - (.setExecutor server (Executors/newCachedThreadPool)) - (.start server) - {:server server - :base (str "http://127.0.0.1:" (.getPort (.getAddress server)))})) - -(defn- parser-delay-fn [method path body-str] - (if (and (= method "POST") (= path "/ui") (= (ui-query body-str) slow-query)) - slow-delay-ms - 0)) - -(deftest latest-search-query-wins - (testing "a slow earlier parser response must not clobber a newer one" - (let [root (repo-root) - {:keys [proc base]} (start-server root)] - (try - (is (wait-health base 15000) "server responds to /healthz") - (let [{:keys [server] proxy-base :base} (start-proxy base parser-delay-fn)] - (try - (core/with-testing-page [pg] - (page/navigate pg proxy-base) - (let [input (page/locator pg "#parser-input") - output (page/locator pg "#parser-output")] - ;; Type the slow query first; wait past the 120ms debounce so its - ;; (slow) request is in flight, then type the fast query. - (locator/fill input slow-query) - (Thread/sleep 300) - (locator/fill input fast-query) - ;; Wait until the slow response has certainly arrived and been - ;; (mis)applied if the race exists. - (Thread/sleep 2000) - ;; The panel must reflect the latest query, not the stale one. - (is (nil? (assert/contains-text (assert/assert-that output) fast-query)) - "output shows the latest query (r/aww)") - (is (nil? (assert/contains-text - (assert/loc-not (assert/assert-that output)) slow-query)) - "output does NOT show the stale earlier query (r/rust)") - (is (nil? (assert/has-value (assert/assert-that input) fast-query)) - "input value is the latest query (r/aww)"))) - (finally - (.stop server 0)))) - (finally - (process/destroy proc)))))) Side B — contributor: tommy-mor Side B — commit message: [52f5c51c] Add Reddit OAuth linking and make UUID the only account identity. OAuth providers only attach to a session UUID (first link creates the principal); linked providers stay private on the account page. Co-authored-by: Cursor Side B — unified diff (full patch): diff --git a/AGENTS.md b/AGENTS.md index 6e0fd8ebb65d665c9c1438e3275971d62b98fd95..e9cc3173dbeb21ad0fc090ca7b407b027c7820a9 100644 --- a/AGENTS.md +++ b/AGENTS.md @@ -35,8 +35,11 @@ Environment variables (defaults in `server/src/state.rs`): - `SORTER2_DATA_DIR` — default `./data` (created on startup) - `SORTER2_EVENT_LOG` — default `{data_dir}/events.jsonl` - `SORTER2_BASE_URL` — public origin (also drives Secure cookies when `https://`) -- `GITHUB_CLIENT_ID` / `GITHUB_CLIENT_SECRET` — GitHub OAuth (optional; login disabled if unset) -- `SORTER2_ALLOW_MOCK_OAUTH=1` — allow `mock_user` on `/auth/github` (tests only) +- `GITHUB_CLIENT_ID` / `GITHUB_CLIENT_SECRET` — GitHub OAuth linking (optional) +- `REDDIT_CLIENT_ID` / `REDDIT_CLIENT_SECRET` (or `REDDIT_APP_*`) — Reddit API import + OAuth linking (optional) +- `SORTER2_ALLOW_MOCK_OAUTH=1` — allow `mock_user` on `/auth/github` and `/auth/reddit` (tests only) + +Identity: UUID is canonical. OAuth providers only *link* to a UUID (first link creates the principal). Linked providers are private to the account owner. Health check: `GET /healthz` → `ok`. diff --git a/server/src/auth/mod.rs b/server/src/auth/mod.rs index 5906f93b13853421e96a3c37bc9d8202a47842bf..c706ae8045a811e5941f5f6c72da88f42a403a82 100644 --- a/server/src/auth/mod.rs +++ b/server/src/auth/mod.rs @@ -1,4 +1,8 @@ -//! GitHub OAuth login, session cookies, and vote actor resolution. +//! OAuth linking, session cookies, and vote actor resolution. +//! +//! Canonical identity is a UUID. OAuth providers only *link* to that UUID +//! (first link creates the principal; later links attach while logged in). +//! Which providers are linked is private to the account owner. pub mod config; pub mod identity; @@ -22,7 +26,9 @@ use crate::{ form_template::template_json_compact, html::layout, state::AppState, - storage_schema::{oauth_link_owner, pseudonym_owner, Store, StoreFields}, + storage_schema::{ + linked_providers_for_uuid, oauth_link_owner, pseudonym_owner, Store, StoreFields, + }, ui_action::UI_RPC_FIELD, }; @@ -53,10 +59,12 @@ fn new_actor_uuid() -> String { pub struct LoginQuery { #[serde(default)] pub return_to: Option, + #[serde(default)] + pub error: Option, } #[derive(Debug, Deserialize)] -pub struct GitHubStartQuery { +pub struct OAuthStartQuery { #[serde(default)] pub return_to: Option, #[serde(default)] @@ -72,15 +80,22 @@ fn return_from_query_or_jar(jar: &CookieJar, query: Option<&str>) -> String { .unwrap_or_else(|| "/".to_string()) } -fn oauth_providers(base_url: &str, return_to: &str) -> Vec<(&'static str, String)> { +/// Available OAuth link targets: `(provider_key, label, start_href)`. +fn oauth_providers(base_url: &str, return_to: &str) -> Vec<(&'static str, &'static str, String)> { let mut out = Vec::new(); + let enc = urlencoding::encode(return_to); if oauth::GitHubConfig::from_env(base_url).is_some() { out.push(( - "GitHub", - format!( - "/auth/github?return_to={}", - urlencoding::encode(return_to) - ), + "github", + oauth::provider_label("github"), + format!("/auth/github?return_to={enc}"), + )); + } + if oauth::RedditConfig::from_env(base_url).is_some() { + out.push(( + "reddit", + oauth::provider_label("reddit"), + format!("/auth/reddit?return_to={enc}"), )); } out @@ -125,23 +140,41 @@ fn alias_claim_forms(return_to: &str, submit_label: &str) -> Result Markup { +fn login_error_message(code: Option<&str>) -> Option<&'static str> { + match code { + Some("oauth_taken") => { + Some("that OAuth account is already linked to a different sorter2 account") + } + Some("oauth_failed") => Some("OAuth failed — try again"), + _ => None, + } +} + +fn signed_out_body( + providers: &[(&str, &str, String)], + error: Option<&str>, +) -> Markup { html! { main class="panel login-page" { section class="login-section" { h1 { "sign in" } - p class="muted" { "link an account to vote under a lasting alias" } + p class="muted" { + "link an OAuth account to create your identity, then claim an alias to vote" + } + @if let Some(msg) = login_error_message(error) { + p class="alias-bad" data-testid="login-error" { (msg) } + } @if providers.is_empty() { p class="muted" { - "OAuth is not configured. Set GITHUB_CLIENT_ID and GITHUB_CLIENT_SECRET." + "OAuth is not configured. Set GitHub and/or Reddit client credentials." } } @else { ul class="oauth-provider-list" { - @for (name, href) in providers { + @for (key, label, href) in providers { li { a href=(href) class="btn-primary oauth-provider" - data-testid=(format!("oauth-{}", name.to_lowercase())) { - (format!("Continue with {name}")) + data-testid=(format!("oauth-{key}")) { + (format!("Link {label}")) } } } @@ -156,7 +189,10 @@ fn signed_out_body(providers: &[(&str, String)]) -> Markup { fn account_body( actor: &session::SessionActor, aliases: &[String], - providers: &[(&str, String)], + // Provider keys already linked to this UUID (private). + linked: &[String], + // Providers available to link: not yet attached. + unlinkable: &[(&str, &str, String)], claim_forms: Markup, ) -> Markup { let current = actor.pseudonym.trim(); @@ -212,16 +248,29 @@ fn account_body( (claim_forms) } - @if !providers.is_empty() { - section class="login-section" { - h2 { "linked sign-in" } - p class="muted small" { "sign in again with the same provider to return to this account" } + section class="login-section" { + h2 { "linked sign-in" } + p class="muted small" { + "private to you — linking more providers raises trust weight without publishing which accounts you use" + } + @if linked.is_empty() { + p class="muted" data-testid="linked-providers-empty" { "none yet" } + } @else { + ul class="linked-provider-list" data-testid="linked-providers" { + @for key in linked { + li data-testid=(format!("linked-{key}")) { + (oauth::provider_label(key)) + } + } + } + } + @if !unlinkable.is_empty() { ul class="oauth-provider-list" { - @for (name, href) in providers { + @for (key, label, href) in unlinkable { li { a href=(href) class="btn-secondary oauth-provider" - data-testid=(format!("oauth-relink-{}", name.to_lowercase())) { - (format!("Re-link {name}")) + data-testid=(format!("oauth-link-{key}")) { + (format!("Link {label}")) } } } @@ -243,12 +292,21 @@ fn account_body( fn login_body( session: Option<&session::SessionActor>, aliases: &[String], - providers: &[(&str, String)], + linked: &[String], + providers: &[(&str, &str, String)], claim_forms: Option, + error: Option<&str>, ) -> Markup { match (session, claim_forms) { - (Some(actor), Some(forms)) => account_body(actor, aliases, providers, forms), - _ => signed_out_body(providers), + (Some(actor), Some(forms)) => { + let unlinkable: Vec<_> = providers + .iter() + .filter(|(key, _, _)| !linked.iter().any(|p| p == key)) + .cloned() + .collect(); + account_body(actor, aliases, linked, &unlinkable, forms) + } + _ => signed_out_body(providers, error), } } @@ -268,6 +326,10 @@ pub async fn login_page( .as_ref() .map(|s| alias_list(db, &s.uuid)) .unwrap_or_default(); + let linked = session + .as_ref() + .map(|s| linked_providers_for_uuid(db, &s.uuid).unwrap_or_default()) + .unwrap_or_default(); let providers = oauth_providers(&base_url_from_env(state.cfg.port), &return_to); let claim_forms = if session.is_some() { @@ -282,7 +344,14 @@ pub async fn login_page( } else { "login · sorter2" }, - login_body(session.as_ref(), &aliases, &providers, claim_forms), + login_body( + session.as_ref(), + &aliases, + &linked, + &providers, + claim_forms, + query.error.as_deref(), + ), state.views.get_views("/login"), session .as_ref() @@ -302,7 +371,6 @@ pub async fn alias_page( let db = state.projection_store.db(); let session = session::load_valid_session(db, &session_id).ok_or(StatusCode::UNAUTHORIZED)?; if session::session_has_pseudonym(&session) { - // Already onboarded — manage aliases on the account page. return Ok(Redirect::to("/login").into_response()); } @@ -331,7 +399,7 @@ pub async fn alias_page( pub async fn github_start( State(state): State, jar: CookieJar, - Query(query): Query, + Query(query): Query, ) -> Result { let cfg = oauth::GitHubConfig::from_env(&base_url_from_env(state.cfg.port)) .ok_or(StatusCode::SERVICE_UNAVAILABLE)?; @@ -342,7 +410,28 @@ pub async fn github_start( } else { None }; - let url = oauth::authorize_url(&cfg, &state_token, mock_user); + let url = oauth::github_authorize_url(&cfg, &state_token, mock_user); + let jar = jar + .add(session::oauth_state_cookie_value(&state_token)) + .add(session::auth_return_cookie_value(&return_to)); + Ok((jar, Redirect::temporary(&url)).into_response()) +} + +pub async fn reddit_start( + State(state): State, + jar: CookieJar, + Query(query): Query, +) -> Result { + let cfg = oauth::RedditConfig::from_env(&base_url_from_env(state.cfg.port)) + .ok_or(StatusCode::SERVICE_UNAVAILABLE)?; + let return_to = return_from_query_or_jar(&jar, query.return_to.as_deref()); + let state_token = session::new_oauth_state(); + let mock_user = if config::mock_oauth_allowed() { + query.mock_user.as_deref() + } else { + None + }; + let url = oauth::reddit_authorize_url(&cfg, &state_token, mock_user); let jar = jar .add(session::oauth_state_cookie_value(&state_token)) .add(session::auth_return_cookie_value(&return_to)); @@ -355,6 +444,13 @@ pub struct OAuthCallbackQuery { pub state: String, } +/// Link `provider:provider_id` to a UUID. +/// +/// - Logged in + new provider → attach to session UUID +/// - Logged in + already ours → no-op +/// - Logged in + owned by someone else → conflict +/// - Logged out + known link → resume that UUID +/// - Logged out + unknown → create principal + first link async fn finish_oauth_login( state: &AppState, jar: CookieJar, @@ -364,11 +460,41 @@ async fn finish_oauth_login( let db = state.projection_store.db(); let return_to = return_from_query_or_jar(&jar, None); - let uuid = match oauth_link_owner(db, provider, &provider_id) - .map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)? - { - Some(existing) => existing, - None => { + let existing_owner = oauth_link_owner(db, provider, &provider_id) + .map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?; + + let session_uuid = session::session_id_from_jar(&jar) + .as_deref() + .and_then(|id| session::load_valid_session(db, id)) + .map(|s| s.uuid); + let linking_while_logged_in = session_uuid.is_some(); + + let uuid = match (session_uuid, existing_owner) { + (Some(session_uuid), Some(owner)) if owner == session_uuid => session_uuid, + (Some(_), Some(_)) => { + return Ok(( + jar.add(session::clear_oauth_state_cookie()), + "/login?error=oauth_taken".into(), + )); + } + (Some(session_uuid), None) => { + let ts = now_ms(); + state + .append_identity_events(vec![Event::OauthLinked { + uuid: session_uuid.clone(), + provider: provider.to_string(), + provider_id, + ts, + }]) + .await + .map_err(|e| { + tracing::warn!(err = %e, "oauth link append failed"); + StatusCode::INTERNAL_SERVER_ERROR + })?; + session_uuid + } + (None, Some(owner)) => owner, + (None, None) => { let uuid = new_actor_uuid(); let ts = now_ms(); state @@ -409,6 +535,9 @@ async fn finish_oauth_login( "/login/alias?return_to={}", urlencoding::encode(&return_to) ) + } else if linking_while_logged_in { + // Additional link while already in an account → stay on account page. + "/login".to_string() } else { return_to }; @@ -434,22 +563,57 @@ pub async fn github_callback( .build() .map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?; - let token = oauth::exchange_code(&client, &cfg, &query.code) + let token = oauth::github_exchange_code(&client, &cfg, &query.code) .await .map_err(|e| { tracing::warn!(err = %e, "github oauth token exchange failed"); StatusCode::BAD_GATEWAY })?; - let user = oauth::fetch_user(&client, &cfg.api_base, &token) + let user = oauth::github_fetch_user(&client, &cfg.api_base, &token) .await .map_err(|e| { tracing::warn!(err = %e, "github user fetch failed"); StatusCode::BAD_GATEWAY })?; - let provider = "github"; - let provider_id = oauth::provider_id(&user); - let (jar, dest) = finish_oauth_login(&state, jar, provider, provider_id).await?; + let (jar, dest) = + finish_oauth_login(&state, jar, "github", oauth::github_provider_id(&user)).await?; + Ok((jar, Redirect::to(&dest)).into_response()) +} + +pub async fn reddit_callback( + State(state): State, + jar: CookieJar, + Query(query): Query, +) -> Result { + let cfg = oauth::RedditConfig::from_env(&base_url_from_env(state.cfg.port)) + .ok_or(StatusCode::SERVICE_UNAVAILABLE)?; + + let expected_state = session::oauth_state_from_jar(&jar).ok_or(StatusCode::BAD_REQUEST)?; + if expected_state != query.state { + return Err(StatusCode::BAD_REQUEST); + } + + let client = Client::builder() + .timeout(std::time::Duration::from_secs(15)) + .build() + .map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?; + + let token = oauth::reddit_exchange_code(&client, &cfg, &query.code) + .await + .map_err(|e| { + tracing::warn!(err = %e, "reddit oauth token exchange failed"); + StatusCode::BAD_GATEWAY + })?; + let user = oauth::reddit_fetch_user(&client, &cfg, &token) + .await + .map_err(|e| { + tracing::warn!(err = %e, "reddit user fetch failed"); + StatusCode::BAD_GATEWAY + })?; + + let (jar, dest) = + finish_oauth_login(&state, jar, "reddit", oauth::reddit_provider_id(&user)).await?; Ok((jar, Redirect::to(&dest)).into_response()) } diff --git a/server/src/auth/oauth.rs b/server/src/auth/oauth.rs index b80078fee0c5acd905b9125d29454e46c4e0d066..369b1527d9032cf312a07819279e0f988ef4a36e 100644 --- a/server/src/auth/oauth.rs +++ b/server/src/auth/oauth.rs @@ -1,8 +1,15 @@ -//! GitHub OAuth (raw reqwest, same style as reddit.rs). +//! OAuth providers (GitHub + Reddit). Provider accounts only *link* to a UUID; +//! the UUID is the canonical identity. Which providers are linked is private. use reqwest::Client; use serde::Deserialize; +use crate::reddit::{ + default_user_agent, reddit_oauth_api_base, reddit_oauth_token_base, +}; + +// ── GitHub ────────────────────────────────────────────────────────────────── + #[derive(Debug, Clone)] pub struct GitHubConfig { pub client_id: String, @@ -40,7 +47,7 @@ impl GitHubConfig { } #[derive(Debug, Deserialize)] -struct TokenResponse { +struct GitHubTokenResponse { access_token: String, } @@ -50,7 +57,7 @@ pub struct GitHubUser { pub login: String, } -pub fn authorize_url(cfg: &GitHubConfig, state: &str, mock_user: Option<&str>) -> String { +pub fn github_authorize_url(cfg: &GitHubConfig, state: &str, mock_user: Option<&str>) -> String { let mut url = format!( "{}/login/oauth/authorize?client_id={}&redirect_uri={}&scope=read:user&state={}", cfg.oauth_base.trim_end_matches('/'), @@ -65,7 +72,7 @@ pub fn authorize_url(cfg: &GitHubConfig, state: &str, mock_user: Option<&str>) - url } -pub async fn exchange_code( +pub async fn github_exchange_code( client: &Client, cfg: &GitHubConfig, code: &str, @@ -90,14 +97,14 @@ pub async fn exchange_code( return Err(format!("github token HTTP {}", resp.status())); } - let body: TokenResponse = resp + let body: GitHubTokenResponse = resp .json() .await .map_err(|e| format!("github token parse failed: {e}"))?; Ok(body.access_token) } -pub async fn fetch_user( +pub async fn github_fetch_user( client: &Client, api_base: &str, access_token: &str, @@ -120,10 +127,149 @@ pub async fn fetch_user( .map_err(|e| format!("github user parse failed: {e}")) } -pub fn provider_id(user: &GitHubUser) -> String { +pub fn github_provider_id(user: &GitHubUser) -> String { user.id.to_string() } +// ── Reddit ────────────────────────────────────────────────────────────────── + +#[derive(Debug, Clone)] +pub struct RedditConfig { + pub client_id: String, + pub client_secret: String, + pub redirect_uri: String, + /// Host for `/api/v1/authorize` (www.reddit.com in production). + pub authorize_base: String, + /// Host for `POST /api/v1/access_token`. + pub token_base: String, + /// Host for bearer `GET /api/v1/me` (oauth.reddit.com). + pub api_base: String, + pub user_agent: String, +} + +/// Authorize page base; defaults to the same host as token POSTs. +pub fn reddit_authorize_base() -> String { + std::env::var("REDDIT_OAUTH_AUTHORIZE_BASE") + .or_else(|_| std::env::var("REDDIT_OAUTH_BASE")) + .unwrap_or_else(|_| "https://www.reddit.com".into()) +} + +impl RedditConfig { + pub fn from_env(base_url: &str) -> Option { + let client_id = std::env::var("REDDIT_CLIENT_ID") + .or_else(|_| std::env::var("REDDIT_APP_ID")) + .ok()?; + let client_secret = std::env::var("REDDIT_CLIENT_SECRET") + .or_else(|_| std::env::var("REDDIT_APP_SECRET")) + .ok()?; + if client_id.is_empty() || client_secret.is_empty() { + return None; + } + let base = base_url.trim_end_matches('/'); + Some(Self { + client_id, + client_secret, + redirect_uri: format!("{base}/auth/reddit/callback"), + authorize_base: reddit_authorize_base(), + token_base: reddit_oauth_token_base(), + api_base: reddit_oauth_api_base(), + user_agent: default_user_agent(), + }) + } +} + +#[derive(Debug, Deserialize)] +struct RedditTokenResponse { + access_token: String, +} + +#[derive(Debug, Deserialize)] +pub struct RedditUser { + /// Stable id (`t2_…`); never use `name` as identity. + pub id: String, + pub name: String, +} + +pub fn reddit_authorize_url(cfg: &RedditConfig, state: &str, mock_user: Option<&str>) -> String { + let mut url = format!( + "{}/api/v1/authorize?client_id={}&response_type=code&state={}&redirect_uri={}&duration=temporary&scope=identity", + cfg.authorize_base.trim_end_matches('/'), + urlencoding::encode(&cfg.client_id), + urlencoding::encode(state), + urlencoding::encode(&cfg.redirect_uri), + ); + if let Some(user) = mock_user { + url.push_str("&mock_user="); + url.push_str(&urlencoding::encode(user)); + } + url +} + +pub async fn reddit_exchange_code( + client: &Client, + cfg: &RedditConfig, + code: &str, +) -> Result { + let resp = client + .post(format!( + "{}/api/v1/access_token", + cfg.token_base.trim_end_matches('/') + )) + .header("User-Agent", &cfg.user_agent) + .basic_auth(&cfg.client_id, Some(&cfg.client_secret)) + .form(&[ + ("grant_type", "authorization_code"), + ("code", code), + ("redirect_uri", cfg.redirect_uri.as_str()), + ]) + .send() + .await + .map_err(|e| format!("reddit token request failed: {e}"))?; + + if !resp.status().is_success() { + let status = resp.status(); + let body = resp.text().await.unwrap_or_default(); + return Err(format!("reddit token HTTP {status}: {body}")); + } + + let body: RedditTokenResponse = resp + .json() + .await + .map_err(|e| format!("reddit token parse failed: {e}"))?; + Ok(body.access_token) +} + +pub async fn reddit_fetch_user( + client: &Client, + cfg: &RedditConfig, + access_token: &str, +) -> Result { + let resp = client + .get(format!( + "{}/api/v1/me", + cfg.api_base.trim_end_matches('/') + )) + .header("User-Agent", &cfg.user_agent) + .bearer_auth(access_token) + .send() + .await + .map_err(|e| format!("reddit user request failed: {e}"))?; + + if !resp.status().is_success() { + return Err(format!("reddit user HTTP {}", resp.status())); + } + + resp.json() + .await + .map_err(|e| format!("reddit user parse failed: {e}")) +} + +pub fn reddit_provider_id(user: &RedditUser) -> String { + user.id.clone() +} + +// ── Shared helpers ────────────────────────────────────────────────────────── + pub fn validate_pseudonym(raw: &str) -> Result { let trimmed = raw.trim(); if trimmed.is_empty() { @@ -144,3 +290,12 @@ pub fn validate_pseudonym(raw: &str) -> Result { pub fn sanitize_pseudonym(login: &str) -> String { validate_pseudonym(login).unwrap_or_else(|_| "user".to_string()) } + +/// Display name for a provider key (`github` → `GitHub`). Never show provider ids. +pub fn provider_label(provider: &str) -> &'static str { + match provider { + "github" => "GitHub", + "reddit" => "Reddit", + _ => "OAuth", + } +} diff --git a/server/src/lib.rs b/server/src/lib.rs index 84f2565b105fb302241b949af64bd5e49916eab2..0d948af1457504fdbd34d0b14261f65ac58da0ec 100644 --- a/server/src/lib.rs +++ b/server/src/lib.rs @@ -47,6 +47,8 @@ pub fn create_app(state: AppState) -> Router { .route("/login/alias", get(crate::auth::alias_page)) .route("/auth/github", get(crate::auth::github_start)) .route("/auth/github/callback", get(crate::auth::github_callback)) + .route("/auth/reddit", get(crate::auth::reddit_start)) + .route("/auth/reddit/callback", get(crate::auth::reddit_callback)) .route("/auth/logout", post(crate::auth::logout)) .route("/auth/switch", post(crate::auth::switch_pseudonym)) .route("/ui", post(crate::api::ui_html::post_ui_html)) diff --git a/server/src/projection_apply.rs b/server/src/projection_apply.rs index 4fb4b48ee0a53b7f673fae0c9731eed5acc33332..f50458c2ff3c447a3cfd28adb298e6883a2da4e9 100644 --- a/server/src/projection_apply.rs +++ b/server/src/projection_apply.rs @@ -4,6 +4,8 @@ //! child links, recent-vote appends) plus a cursor advance, all committed in //! one atomic `DisableWal` batch. +use std::collections::HashMap; + use crate::{ event_log::EventLogError, events::{Event, EventRecord}, @@ -44,6 +46,8 @@ pub fn apply_records( let db = projection_store.db(); let mut batch = db.batch(); let mut last_seq = 0u64; + // Weight reads must see earlier writes in this same batch. + let mut pending_weights: HashMap = HashMap::new(); for record in records { match &record.event { @@ -85,6 +89,7 @@ pub fn apply_records( ensure_path_writes(&mut batch, &parsed); } Event::PrincipalCreated { uuid, .. } => { + pending_weights.insert(uuid.clone(), BASE_TRUST_WEIGHT); batch.write( Store::root() .user_weights() @@ -112,17 +117,25 @@ pub fn apply_records( } } else { batch.write(Store::root().oauth_links().key(&link_key).set(uuid)); - let current = Store::root() - .user_weights() - .key(&uuid.clone()) - .get(db) - .map_err(|e| EventLogError::Apply(e.to_string()))? + let current = pending_weights + .get(uuid) + .copied() + .or_else(|| { + Store::root() + .user_weights() + .key(&uuid.clone()) + .get(db) + .ok() + .flatten() + }) .unwrap_or(BASE_TRUST_WEIGHT); + let next = trust_weight_after_link(current); + pending_weights.insert(uuid.clone(), next); batch.write( Store::root() .user_weights() .key(&uuid.clone()) - .set(&trust_weight_after_link(current)), + .set(&next), ); } } @@ -205,6 +218,15 @@ mod tests { ), record( 3, + Event::OauthLinked { + uuid: uuid.into(), + provider: "reddit".into(), + provider_id: "t2_abc".into(), + ts, + }, + ), + record( + 4, Event::PseudonymClaimed { uuid: uuid.into(), pseudonym: "octocat".into(), @@ -219,8 +241,12 @@ mod tests { oauth_link_owner(store.db(), "github", "42").unwrap(), Some(uuid.to_string()) ); + assert_eq!( + crate::storage_schema::linked_providers_for_uuid(store.db(), uuid).unwrap(), + vec!["github".to_string(), "reddit".to_string()] + ); assert_eq!(resolve_actor_uuid(store.db(), "octocat").unwrap(), uuid); - assert_eq!(user_trust_weight(store.db(), uuid).unwrap(), 1.5); + assert_eq!(user_trust_weight(store.db(), uuid).unwrap(), 2.0); let aliases = Store::root() .user_pseudonyms() .key(&uuid.to_string()) diff --git a/server/src/storage_schema.rs b/server/src/storage_schema.rs index b942810afd96d3bf01b7765718303a39be4ef8d2..dac6f8e6080e3f5683c24dbb8861f5a981d456c4 100644 --- a/server/src/storage_schema.rs +++ b/server/src/storage_schema.rs @@ -103,6 +103,25 @@ pub fn oauth_link_owner(db: &Db, provider: &str, provider_id: &str) -> durable:: .get(db) } +/// Provider names linked to a UUID (`github`, `reddit`, …). Private — for the +/// account owner's page only; never expose which providers are linked publicly. +pub fn linked_providers_for_uuid(db: &Db, uuid: &str) -> durable::Result> { + let mut providers = Vec::new(); + for (key, owner) in Store::root().oauth_links().iter(db)? { + if owner != uuid { + continue; + } + let Some((provider, _)) = key.split_once(':') else { + continue; + }; + if !providers.iter().any(|p| p == provider) { + providers.push(provider.to_string()); + } + } + providers.sort(); + Ok(providers) +} + pub const RECENT_VOTES_CAP: u64 = 200; fn id_key(id: &ItemId) -> String { diff --git a/test/support/harness.clj b/test/support/harness.clj index 4505ece5aa193e826ca61c52f1467d252080d66b..741024aab1d14269e225791189633287980ae2e4 100644 --- a/test/support/harness.clj +++ b/test/support/harness.clj @@ -48,12 +48,15 @@ "GITHUB_CLIENT_SECRET" "test-secret" "GITHUB_OAUTH_BASE" (str "http://127.0.0.1:" oauth-port) "GITHUB_API_BASE" (str "http://127.0.0.1:" oauth-port) + ;; Reddit import fixtures + Reddit OAuth on reddit-port. "REDDIT_API_BASE" (str "http://127.0.0.1:" reddit-port) + "REDDIT_CLIENT_ID" "test-reddit" + "REDDIT_CLIENT_SECRET" "test-reddit-secret" "REDDIT_OAUTH_BASE" (str "http://127.0.0.1:" reddit-port) - "REDDIT_CLIENT_ID" "" - "REDDIT_CLIENT_SECRET" "" - "REDDIT_APP_ID" "" - "REDDIT_APP_SECRET" ""})) + "REDDIT_OAUTH_AUTHORIZE_BASE" (str "http://127.0.0.1:" reddit-port) + "REDDIT_OAUTH_TOKEN_BASE" (str "http://127.0.0.1:" reddit-port) + "REDDIT_OAUTH_API_BASE" (str "http://127.0.0.1:" reddit-port) + "REDDIT_USER_AGENT" "web:sorter2-test:v0 (by /u/test)"})) (defn with-auth-servers "Start mock Reddit + mock OAuth + release sorter2-server. diff --git a/test/support/mock_oauth.clj b/test/support/mock_oauth.clj index 909d7a7be8b159ea54a122d69c46af3db3318118..5ba7e9be3cf6d2226648e9609a09ed45f306930f 100644 --- a/test/support/mock_oauth.clj +++ b/test/support/mock_oauth.clj @@ -1,5 +1,5 @@ (ns test.support.mock-oauth - "In-process HTTP stub for GitHub OAuth (authorize, token, /user)." + "In-process HTTP stub for GitHub + Reddit OAuth (authorize, token, user)." (:require [clojure.string :as str]) (:import [com.sun.net.httpserver HttpServer HttpHandler HttpExchange] [java.net InetSocketAddress URLDecoder])) @@ -12,11 +12,14 @@ (URLDecoder/decode (or v "") "UTF-8")))) (str/split query #"&")))) -(defn- parse-mock-user [raw] +(defn- parse-mock-user + "GitHub-style `id:login` (numeric id). Reddit-style `t2_xxx:name`." + [raw] (let [s (or raw "1002:newbie") [id login] (str/split s #":" 2)] - {:id (Long/parseLong id) - :login (or login "newbie")})) + {:id id + :login (or login "newbie") + :numeric? (re-matches #"\d+" id)})) (defn- send-json [^HttpExchange ex status body] (let [bytes (.getBytes body "UTF-8")] @@ -31,9 +34,10 @@ (.sendResponseHeaders ex 302 -1) (.close (.getResponseBody ex))) -(defn- read-form-code [^HttpExchange ex] +(defn- read-form [^HttpExchange ex] (let [body (slurp (.getInputStream ex))] - (query-param body "code"))) + {:code (query-param body "code") + :grant (query-param body "grant_type")})) (defn- bearer-token [^HttpExchange ex] (some-> (.getRequestHeaders ex) @@ -44,8 +48,18 @@ (when (str/starts-with? token "mock:") (parse-mock-user (subs token 5)))) +(defn- authorize-redirect [exchange query] + (let [redirect-uri (query-param query "redirect_uri") + state (query-param query "state") + mock-user (query-param query "mock_user") + user (parse-mock-user mock-user) + code (str "mock:" (:id user) ":" (:login user)) + loc (str redirect-uri "?code=" (java.net.URLEncoder/encode code "UTF-8") + "&state=" (java.net.URLEncoder/encode state "UTF-8"))] + (send-redirect exchange loc))) + (defn start-mock-oauth - "Start mock GitHub OAuth on `port`. Returns a zero-arg `stop` function." + "Start mock GitHub + Reddit OAuth on `port`. Returns a zero-arg `stop` function." [port] (let [server (HttpServer/create (InetSocketAddress. "127.0.0.1" port) 0) handler @@ -53,28 +67,45 @@ (handle [^HttpExchange exchange] (let [uri (.getRequestURI exchange) path (.getPath uri) - query (.getQuery uri)] + query (.getQuery uri) + method (.getRequestMethod exchange)] (cond + ;; GitHub authorize (str/ends-with? path "/login/oauth/authorize") - (let [redirect-uri (query-param query "redirect_uri") - state (query-param query "state") - mock-user (query-param query "mock_user") - user (parse-mock-user mock-user) - code (str "mock:" (:id user) ":" (:login user)) - loc (str redirect-uri "?code=" (java.net.URLEncoder/encode code "UTF-8") - "&state=" (java.net.URLEncoder/encode state "UTF-8"))] - (send-redirect exchange loc)) + (authorize-redirect exchange query) + + ;; Reddit authorize + (str/ends-with? path "/api/v1/authorize") + (authorize-redirect exchange query) - (str/ends-with? path "/login/oauth/access_token") - (let [code (or (read-form-code exchange) "mock:1002:newbie")] + ;; GitHub token + (and (= method "POST") (str/ends-with? path "/login/oauth/access_token")) + (let [code (or (:code (read-form exchange)) "mock:1002:newbie")] (send-json exchange 200 (str "{\"access_token\":\"" code "\",\"token_type\":\"bearer\"}"))) + ;; Reddit token (client_credentials for import + authorization_code for login) + (and (= method "POST") (str/ends-with? path "/api/v1/access_token")) + (let [form (read-form exchange) + grant (or (:grant form) "") + code (or (:code form) "mock:t2_test:redditor")] + (if (= grant "client_credentials") + (send-json exchange 200 "{\"access_token\":\"app-token\",\"token_type\":\"bearer\",\"expires_in\":3600}") + (send-json exchange 200 (str "{\"access_token\":\"" code "\",\"token_type\":\"bearer\",\"expires_in\":3600}")))) + + ;; GitHub user (= path "/user") (let [token (bearer-token exchange) - user (or (parse-token-user token) {:id 1002 :login "newbie"})] + user (or (parse-token-user token) {:id "1002" :login "newbie" :numeric? true})] (send-json exchange 200 (str "{\"id\":" (:id user) ",\"login\":\"" (:login user) "\"}"))) + ;; Reddit /api/v1/me + (str/ends-with? path "/api/v1/me") + (let [token (bearer-token exchange) + user (or (parse-token-user token) {:id "t2_test" :login "redditor"})] + (send-json exchange 200 + (str "{\"id\":\"" (:id user) "\",\"name\":\"" (:login user) "\"}"))) + :else (send-json exchange 404 "{\"error\":\"not found\"}")))))] (.createContext server "/" handler) diff --git a/test/support/mock_reddit.clj b/test/support/mock_reddit.clj index 5efa92db3e1f79b8f423a2f1adcbda959123c1ad..a630cf0938722193e9af88382d60e777ff371be4 100644 --- a/test/support/mock_reddit.clj +++ b/test/support/mock_reddit.clj @@ -1,14 +1,56 @@ (ns test.support.mock-reddit - "In-process HTTP stub for Reddit API fixtures (`test/fixtures/reddit/`)." + "In-process HTTP stub for Reddit API fixtures + OAuth login endpoints." (:require [clojure.java.io :as io] [clojure.string :as str]) (:import [com.sun.net.httpserver HttpServer HttpHandler HttpExchange] - [java.net InetSocketAddress])) + [java.net InetSocketAddress URLDecoder])) (defn fixtures-dir ([] (fixtures-dir (System/getProperty "user.dir"))) ([root] (str root "/test/fixtures/reddit"))) +(defn- query-param [query key] + (when query + (some (fn [pair] + (let [[k v] (str/split pair "=" 2)] + (when (= k key) + (URLDecoder/decode (or v "") "UTF-8")))) + (str/split query #"&")))) + +(defn- parse-mock-user [raw] + (let [s (or raw "t2_test:redditor") + [id login] (str/split s #":" 2)] + {:id id :login (or login "redditor")})) + +(defn- send-bytes [^HttpExchange ex status ^bytes body content-type] + (.set (.getResponseHeaders ex) "Content-Type" content-type) + (.sendResponseHeaders ex status (alength body)) + (doto (.getResponseBody ex) + (.write body) + (.close))) + +(defn- send-json [^HttpExchange ex status body] + (send-bytes ex status (.getBytes body "UTF-8") "application/json")) + +(defn- send-redirect [^HttpExchange ex location] + (.set (.getResponseHeaders ex) "Location" location) + (.sendResponseHeaders ex 302 -1) + (.close (.getResponseBody ex))) + +(defn- read-form [^HttpExchange ex] + (let [body (slurp (.getInputStream ex))] + {:code (query-param body "code") + :grant (query-param body "grant_type")})) + +(defn- bearer-token [^HttpExchange ex] + (some-> (.getRequestHeaders ex) + (.getFirst "Authorization") + (str/replace #"^[Bb]earer " ""))) + +(defn- parse-token-user [token] + (when (str/starts-with? token "mock:") + (parse-mock-user (subs token 5)))) + (defn start-mock-reddit "Start a mock Reddit API on `port`. Returns a zero-arg `stop` function." ([port] (start-mock-reddit port (fixtures-dir))) @@ -19,15 +61,38 @@ handler (proxy [HttpHandler] [] (handle [^HttpExchange exchange] - ;; `/r//about.json` → subreddit entity; `/r/.json` → listing. - (let [path (.getPath (.getRequestURI exchange)) - body (if (str/includes? path "/about") - about - listing)] - (.sendResponseHeaders exchange 200 (alength body)) - (let [out (.getResponseBody exchange)] - (.write out body) - (.close out)))))] + (let [uri (.getRequestURI exchange) + path (.getPath uri) + query (.getQuery uri) + method (.getRequestMethod exchange)] + (cond + (str/ends-with? path "/api/v1/authorize") + (let [redirect-uri (query-param query "redirect_uri") + state (query-param query "state") + user (parse-mock-user (query-param query "mock_user")) + code (str "mock:" (:id user) ":" (:login user)) + loc (str redirect-uri "?code=" (java.net.URLEncoder/encode code "UTF-8") + "&state=" (java.net.URLEncoder/encode state "UTF-8"))] + (send-redirect exchange loc)) + + (and (= method "POST") (str/ends-with? path "/api/v1/access_token")) + (let [form (read-form exchange) + grant (or (:grant form) "") + code (or (:code form) "mock:t2_test:redditor")] + (if (= grant "client_credentials") + (send-json exchange 200 "{\"access_token\":\"app-token\",\"token_type\":\"bearer\",\"expires_in\":3600}") + (send-json exchange 200 (str "{\"access_token\":\"" code "\",\"token_type\":\"bearer\",\"expires_in\":3600}")))) + + (str/ends-with? path "/api/v1/me") + (let [user (or (parse-token-user (bearer-token exchange)) + {:id "t2_test" :login "redditor"})] + (send-json exchange 200 + (str "{\"id\":\"" (:id user) "\",\"name\":\"" (:login user) "\"}"))) + + ;; `/r//about.json` → subreddit entity; `/r/.json` → listing. + :else + (let [body (if (str/includes? path "/about") about listing)] + (send-bytes exchange 200 body "application/json"))))))] (.createContext server "/" handler) (.setExecutor server nil) (.start server)