chore(assets): scripted screenshot pipeline
This commit is contained in:
@@ -1,3 +1,6 @@
|
||||
[alias]
|
||||
xtask = "run -p xtask --"
|
||||
|
||||
[target.x86_64-unknown-linux-musl]
|
||||
linker = "x86_64-linux-gnu-gcc"
|
||||
rustflags = ["-C", "target-feature=+crt-static", "-C", "link-arg=-lgcc"]
|
||||
|
||||
1
.gitignore
vendored
1
.gitignore
vendored
@@ -3,6 +3,7 @@
|
||||
# will have compiled files and executables
|
||||
debug/
|
||||
target/
|
||||
images/generated/
|
||||
dev/
|
||||
.agents/
|
||||
.env
|
||||
|
||||
@@ -160,6 +160,12 @@ For more detailed information, please refer to the following documents:
|
||||
- **Experimental providers staging area**: [crates/providers/experimental/README.md](crates/providers/experimental/README.md) records the placeholder crates (OpenAI, Anthropic, Gemini) and their current status.
|
||||
- **[docs/platform-support.md](docs/platform-support.md)**: Current OS support matrix and cross-check instructions.
|
||||
|
||||
## Developer Tasks
|
||||
|
||||
- `cargo xtask screenshots` regenerates deterministic ANSI dumps (and, when
|
||||
`chafa` is available, PNG renders) for the documentation gallery. Use
|
||||
`--no-png` to skip the PNG step or `--output <dir>` to redirect the output.
|
||||
|
||||
## Configuration
|
||||
|
||||
OWLEN stores its configuration in the standard platform-specific config directory:
|
||||
|
||||
@@ -176,3 +176,16 @@ When modifying TUI UX or keybindings:
|
||||
|
||||
Keeping these steps in sync ensures Owlen’s keyboard-first UX remains
|
||||
predictable, accessible, and discoverable.
|
||||
|
||||
## 8. Screenshot Pipeline
|
||||
|
||||
Use the scripted pipeline to regenerate gallery assets and documentation
|
||||
illustrations:
|
||||
|
||||
- `cargo xtask screenshots` emits ANSI dumps under `images/generated` and, by
|
||||
default, converts them to PNG with `chafa`. Pass `--no-png` to skip PNG
|
||||
rendering or `--chafa /path/to/chafa` to override the binary. Each scene is
|
||||
deterministic—no network calls—and mirrors the regression snapshots.
|
||||
- The pipeline reuses the same stub provider harness as the snapshot tests, so
|
||||
new scenes should be added in tandem with `chat_snapshots.rs` to keep visual
|
||||
regression coverage and documentation imagery aligned.
|
||||
|
||||
@@ -7,3 +7,12 @@ publish = false
|
||||
[dependencies]
|
||||
anyhow = { workspace = true }
|
||||
clap = { workspace = true, features = ["derive"] }
|
||||
crossterm = { workspace = true }
|
||||
async-trait = { workspace = true }
|
||||
futures-util = { workspace = true }
|
||||
owlen-core = { path = "../crates/owlen-core" }
|
||||
owlen-tui = { path = "../crates/owlen-tui" }
|
||||
ratatui = { workspace = true }
|
||||
serde_json = { workspace = true }
|
||||
tempfile = { workspace = true }
|
||||
tokio = { workspace = true }
|
||||
|
||||
@@ -4,6 +4,8 @@ use std::process::Command;
|
||||
use anyhow::{Context, Result, bail};
|
||||
use clap::{Parser, Subcommand};
|
||||
|
||||
mod screenshots;
|
||||
|
||||
#[derive(Parser)]
|
||||
#[command(author, version, about = "Owlen developer tasks", long_about = None)]
|
||||
struct Xtask {
|
||||
@@ -36,6 +38,23 @@ enum Task {
|
||||
#[arg(long, value_name = "PATH", help = "Override the repo map output path")]
|
||||
output: Option<PathBuf>,
|
||||
},
|
||||
/// Generate deterministic TUI screenshots and optional PNG renders.
|
||||
Screenshots {
|
||||
#[arg(
|
||||
long,
|
||||
value_name = "PATH",
|
||||
help = "Output directory (defaults to images/generated)"
|
||||
)]
|
||||
output: Option<PathBuf>,
|
||||
#[arg(long, help = "Skip PNG conversion and only emit ANSI dumps")]
|
||||
no_png: bool,
|
||||
#[arg(
|
||||
long,
|
||||
value_name = "PATH",
|
||||
help = "Path to chafa binary (default: chafa in PATH)"
|
||||
)]
|
||||
chafa: Option<PathBuf>,
|
||||
},
|
||||
}
|
||||
|
||||
fn main() -> Result<()> {
|
||||
@@ -49,6 +68,11 @@ fn main() -> Result<()> {
|
||||
Task::DevRun { args } => dev_run(args),
|
||||
Task::ReleaseCheck => release_check(),
|
||||
Task::GenRepoMap { output } => gen_repo_map(output),
|
||||
Task::Screenshots {
|
||||
output,
|
||||
no_png,
|
||||
chafa,
|
||||
} => screenshots::run(output, chafa, no_png),
|
||||
}
|
||||
}
|
||||
|
||||
@@ -154,7 +178,7 @@ fn run_cargo(args: Vec<String>) -> Result<()> {
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn workspace_root() -> PathBuf {
|
||||
pub(crate) fn workspace_root() -> PathBuf {
|
||||
Path::new(env!("CARGO_MANIFEST_DIR"))
|
||||
.parent()
|
||||
.expect("xtask has a parent directory")
|
||||
|
||||
444
xtask/src/screenshots.rs
Normal file
444
xtask/src/screenshots.rs
Normal file
@@ -0,0 +1,444 @@
|
||||
use std::env;
|
||||
use std::fs;
|
||||
use std::path::{Path, PathBuf};
|
||||
use std::process::Command;
|
||||
|
||||
use anyhow::{Context, Result, bail};
|
||||
use async_trait::async_trait;
|
||||
use crossterm::event::{KeyCode, KeyEvent, KeyModifiers};
|
||||
use futures_util::FutureExt;
|
||||
use futures_util::future::BoxFuture;
|
||||
use owlen_core::{
|
||||
Config, McpMode, Mode, Provider,
|
||||
session::SessionController,
|
||||
storage::StorageManager,
|
||||
types::{Message, ModelInfo, ToolCall},
|
||||
ui::{NoOpUiController, UiController},
|
||||
};
|
||||
use owlen_tui::ChatApp;
|
||||
use owlen_tui::events::Event;
|
||||
use owlen_tui::ui::render_chat;
|
||||
use ratatui::{Terminal, backend::TestBackend};
|
||||
use serde_json::json;
|
||||
use tokio::runtime::Runtime;
|
||||
use tokio::sync::mpsc;
|
||||
|
||||
use crate::workspace_root;
|
||||
|
||||
#[derive(Clone, Copy)]
|
||||
struct SceneSpec {
|
||||
name: &'static str,
|
||||
width: u16,
|
||||
height: u16,
|
||||
builder: fn() -> BoxFuture<'static, Result<ChatApp>>,
|
||||
}
|
||||
|
||||
pub(crate) fn run(output: Option<PathBuf>, chafa: Option<PathBuf>, no_png: bool) -> Result<()> {
|
||||
let output_dir = output.unwrap_or_else(|| workspace_root().join("images/generated"));
|
||||
fs::create_dir_all(&output_dir)
|
||||
.with_context(|| format!("failed to create output directory {}", output_dir.display()))?;
|
||||
|
||||
let config_home = workspace_root().join("dev/xtask-config");
|
||||
fs::create_dir_all(&config_home).with_context(|| {
|
||||
format!(
|
||||
"failed to create config home directory {}",
|
||||
config_home.display()
|
||||
)
|
||||
})?;
|
||||
unsafe {
|
||||
// Safe: we control the value and keep it within the workspace for deterministic snapshots.
|
||||
env::set_var("XDG_CONFIG_HOME", &config_home);
|
||||
env::set_var("XDG_DATA_HOME", &config_home);
|
||||
}
|
||||
|
||||
let runtime = Runtime::new().context("failed to create tokio runtime")?;
|
||||
|
||||
for scene in scenes() {
|
||||
let mut app = runtime.block_on((scene.builder)())?;
|
||||
let ansi = render_snapshot(&mut app, scene.width, scene.height)?;
|
||||
let ans_path = output_dir.join(format!("{}.ans", scene.name));
|
||||
fs::write(&ans_path, ansi.as_bytes())
|
||||
.with_context(|| format!("failed to write ANSI dump {}", ans_path.display()))?;
|
||||
|
||||
if !no_png
|
||||
&& let Err(err) = convert_to_png(
|
||||
&ans_path,
|
||||
scene.width,
|
||||
scene.height,
|
||||
&output_dir,
|
||||
scene.name,
|
||||
chafa.as_deref(),
|
||||
)
|
||||
{
|
||||
eprintln!("warning: {}", err);
|
||||
}
|
||||
}
|
||||
|
||||
println!("Screenshots written to {}", output_dir.display());
|
||||
if no_png {
|
||||
println!("PNG conversion skipped (use --no-png=false or omit flag to enable)");
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn convert_to_png(
|
||||
ans_path: &Path,
|
||||
width: u16,
|
||||
height: u16,
|
||||
output_dir: &Path,
|
||||
name: &str,
|
||||
chafa_path: Option<&Path>,
|
||||
) -> Result<()> {
|
||||
let png_path = output_dir.join(format!("{}.png", name));
|
||||
let mut command = if let Some(path) = chafa_path {
|
||||
Command::new(path)
|
||||
} else {
|
||||
Command::new("chafa")
|
||||
};
|
||||
let status = command
|
||||
.arg("--size")
|
||||
.arg(format!("{}x{}", width, height))
|
||||
.arg("--save")
|
||||
.arg(&png_path)
|
||||
.arg(ans_path)
|
||||
.status()
|
||||
.with_context(|| "failed to spawn chafa".to_string())?;
|
||||
if !status.success() {
|
||||
bail!(
|
||||
"chafa exited with status {} when rendering {}",
|
||||
status.code().unwrap_or_default(),
|
||||
name
|
||||
);
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn scenes() -> &'static [SceneSpec] {
|
||||
&[
|
||||
SceneSpec {
|
||||
name: "chat-idle-dark",
|
||||
width: 120,
|
||||
height: 32,
|
||||
builder: || async move { Ok(build_chat_app(|_| {}, |_| {}).await) }.boxed(),
|
||||
},
|
||||
SceneSpec {
|
||||
name: "chat-tool-call",
|
||||
width: 120,
|
||||
height: 32,
|
||||
builder: || {
|
||||
async move {
|
||||
let app = build_chat_app(
|
||||
|_| {},
|
||||
|session| {
|
||||
let conversation = session.conversation_mut();
|
||||
conversation.push_user_message("What happened in the Rust ecosystem today?");
|
||||
let stream_id = conversation.start_streaming_response();
|
||||
conversation
|
||||
.set_stream_placeholder(stream_id, "Consulting the knowledge base…")
|
||||
.expect("placeholder");
|
||||
let tool_call = ToolCall {
|
||||
id: "call-search-1".into(),
|
||||
name: "web_search".into(),
|
||||
arguments: json!({ "query": "Rust language news" }),
|
||||
};
|
||||
conversation
|
||||
.set_tool_calls_on_message(stream_id, vec![tool_call.clone()])
|
||||
.expect("tool call metadata");
|
||||
conversation
|
||||
.append_stream_chunk(stream_id, "Found multiple articles…", false)
|
||||
.expect("stream chunk");
|
||||
conversation.push_message(Message::tool(
|
||||
tool_call.id.clone(),
|
||||
"Rust 1.85 released with generics cleanups and faster async compilation.".into(),
|
||||
));
|
||||
conversation.push_message(Message::assistant(
|
||||
"Summarising the latest Rust release and the async runtime updates.".into(),
|
||||
));
|
||||
},
|
||||
)
|
||||
.await;
|
||||
Ok(app)
|
||||
}
|
||||
.boxed()
|
||||
},
|
||||
},
|
||||
SceneSpec {
|
||||
name: "command-palette",
|
||||
width: 110,
|
||||
height: 28,
|
||||
builder: || {
|
||||
async move {
|
||||
let mut app = build_chat_app(|_| {}, |_| {}).await;
|
||||
send_key(&mut app, KeyCode::Char(':'), KeyModifiers::NONE).await;
|
||||
type_text(&mut app, "focus").await;
|
||||
send_key(&mut app, KeyCode::Down, KeyModifiers::NONE).await;
|
||||
Ok(app)
|
||||
}
|
||||
.boxed()
|
||||
},
|
||||
},
|
||||
SceneSpec {
|
||||
name: "guidance-onboarding",
|
||||
width: 100,
|
||||
height: 28,
|
||||
builder: || {
|
||||
async move {
|
||||
let app = build_chat_app(
|
||||
|cfg| {
|
||||
cfg.ui.show_onboarding = true;
|
||||
cfg.ui.guidance.coach_marks_complete = false;
|
||||
},
|
||||
|_| {},
|
||||
)
|
||||
.await;
|
||||
Ok(app)
|
||||
}
|
||||
.boxed()
|
||||
},
|
||||
},
|
||||
SceneSpec {
|
||||
name: "guidance-cheatsheet",
|
||||
width: 120,
|
||||
height: 30,
|
||||
builder: || {
|
||||
async move {
|
||||
let mut app =
|
||||
build_chat_app(|cfg| cfg.ui.guidance.coach_marks_complete = true, |_| {})
|
||||
.await;
|
||||
send_key(&mut app, KeyCode::Char('?'), KeyModifiers::NONE).await;
|
||||
Ok(app)
|
||||
}
|
||||
.boxed()
|
||||
},
|
||||
},
|
||||
SceneSpec {
|
||||
name: "editing-mode",
|
||||
width: 120,
|
||||
height: 32,
|
||||
builder: || {
|
||||
async move {
|
||||
let mut app = build_chat_app(|_| {}, |_| {}).await;
|
||||
send_key(&mut app, KeyCode::Char('i'), KeyModifiers::NONE).await;
|
||||
type_text(&mut app, "Editing mode demonstration").await;
|
||||
Ok(app)
|
||||
}
|
||||
.boxed()
|
||||
},
|
||||
},
|
||||
SceneSpec {
|
||||
name: "visual-mode",
|
||||
width: 120,
|
||||
height: 32,
|
||||
builder: || {
|
||||
async move {
|
||||
let mut app = build_chat_app(
|
||||
|_| {},
|
||||
|session| {
|
||||
let conversation = session.conversation_mut();
|
||||
conversation.push_user_message(
|
||||
"Render visual selection across multiple lines.",
|
||||
);
|
||||
conversation.push_message(Message::assistant(
|
||||
"Assistant reply for visual mode highlighting.".into(),
|
||||
));
|
||||
},
|
||||
)
|
||||
.await;
|
||||
send_key(&mut app, KeyCode::Char('v'), KeyModifiers::NONE).await;
|
||||
send_key(&mut app, KeyCode::Char('j'), KeyModifiers::NONE).await;
|
||||
send_key(&mut app, KeyCode::Char('j'), KeyModifiers::NONE).await;
|
||||
Ok(app)
|
||||
}
|
||||
.boxed()
|
||||
},
|
||||
},
|
||||
SceneSpec {
|
||||
name: "accessibility-high-contrast",
|
||||
width: 120,
|
||||
height: 32,
|
||||
builder: || {
|
||||
async move {
|
||||
let app = build_chat_app(
|
||||
|cfg| {
|
||||
cfg.ui.accessibility.high_contrast = true;
|
||||
cfg.ui.guidance.coach_marks_complete = true;
|
||||
},
|
||||
|_| {},
|
||||
)
|
||||
.await;
|
||||
Ok(app)
|
||||
}
|
||||
.boxed()
|
||||
},
|
||||
},
|
||||
SceneSpec {
|
||||
name: "accessibility-reduced-chrome",
|
||||
width: 120,
|
||||
height: 32,
|
||||
builder: || {
|
||||
async move {
|
||||
let app = build_chat_app(
|
||||
|cfg| {
|
||||
cfg.ui.accessibility.reduced_chrome = true;
|
||||
cfg.ui.guidance.coach_marks_complete = true;
|
||||
},
|
||||
|_| {},
|
||||
)
|
||||
.await;
|
||||
Ok(app)
|
||||
}
|
||||
.boxed()
|
||||
},
|
||||
},
|
||||
SceneSpec {
|
||||
name: "emacs-profile",
|
||||
width: 120,
|
||||
height: 32,
|
||||
builder: || {
|
||||
async move {
|
||||
let mut app = build_chat_app(
|
||||
|cfg| {
|
||||
cfg.ui.keymap_profile = Some("emacs".into());
|
||||
cfg.ui.guidance.coach_marks_complete = true;
|
||||
},
|
||||
|_| {},
|
||||
)
|
||||
.await;
|
||||
send_key(&mut app, KeyCode::Char(':'), KeyModifiers::NONE).await;
|
||||
type_text(&mut app, "help").await;
|
||||
Ok(app)
|
||||
}
|
||||
.boxed()
|
||||
},
|
||||
},
|
||||
]
|
||||
}
|
||||
|
||||
fn render_snapshot(app: &mut ChatApp, width: u16, height: u16) -> Result<String> {
|
||||
let backend = TestBackend::new(width, height);
|
||||
let mut terminal = Terminal::new(backend).context("failed to create test terminal")?;
|
||||
terminal
|
||||
.draw(|frame| render_chat(frame, app))
|
||||
.context("failed to render chat frame")?;
|
||||
let buffer = terminal.backend().buffer();
|
||||
Ok(buffer_to_string(buffer))
|
||||
}
|
||||
|
||||
fn buffer_to_string(buffer: &ratatui::buffer::Buffer) -> String {
|
||||
let mut output = String::new();
|
||||
for y in 0..buffer.area.height {
|
||||
for x in 0..buffer.area.width {
|
||||
output.push_str(buffer[(x, y)].symbol());
|
||||
}
|
||||
output.push('\n');
|
||||
}
|
||||
output
|
||||
}
|
||||
|
||||
async fn send_key(app: &mut ChatApp, code: KeyCode, modifiers: KeyModifiers) {
|
||||
app.handle_event(Event::Key(KeyEvent::new(code, modifiers)))
|
||||
.await
|
||||
.expect("send key event");
|
||||
}
|
||||
|
||||
async fn type_text(app: &mut ChatApp, text: &str) {
|
||||
for ch in text.chars() {
|
||||
send_key(app, KeyCode::Char(ch), KeyModifiers::NONE).await;
|
||||
}
|
||||
}
|
||||
|
||||
async fn build_chat_app<C, F>(configure_config: C, configure_session: F) -> ChatApp
|
||||
where
|
||||
C: FnOnce(&mut Config) + Send + 'static,
|
||||
F: FnOnce(&mut SessionController) + Send + 'static,
|
||||
{
|
||||
let temp_dir = tempfile::tempdir().expect("temp dir");
|
||||
let storage =
|
||||
StorageManager::with_database_path(temp_dir.path().join("owlen-tui-screenshots.db"))
|
||||
.await
|
||||
.expect("storage");
|
||||
let storage = std::sync::Arc::new(storage);
|
||||
|
||||
let mut config = Config::default();
|
||||
configure_config(&mut config);
|
||||
config.general.default_model = Some("stub-model".into());
|
||||
config.general.enable_streaming = true;
|
||||
config.privacy.encrypt_local_data = false;
|
||||
config.privacy.require_consent_per_session = false;
|
||||
config.ui.show_timestamps = false;
|
||||
config.mcp.mode = McpMode::LocalOnly;
|
||||
config.mcp.allow_fallback = false;
|
||||
config.mcp.warn_on_legacy = false;
|
||||
|
||||
let provider: std::sync::Arc<dyn Provider> = std::sync::Arc::new(StubProvider);
|
||||
let ui: std::sync::Arc<dyn UiController> = std::sync::Arc::new(NoOpUiController);
|
||||
let (event_tx, controller_event_rx) = mpsc::unbounded_channel();
|
||||
|
||||
let mut session = SessionController::new(provider, config, storage, ui, true, Some(event_tx))
|
||||
.await
|
||||
.expect("session controller");
|
||||
|
||||
session
|
||||
.set_operating_mode(Mode::Chat)
|
||||
.await
|
||||
.expect("chat mode");
|
||||
|
||||
configure_session(&mut session);
|
||||
|
||||
let (app, mut session_rx) = ChatApp::new(session, controller_event_rx)
|
||||
.await
|
||||
.expect("chat app");
|
||||
session_rx.close();
|
||||
|
||||
app
|
||||
}
|
||||
|
||||
#[derive(Default)]
|
||||
struct StubProvider;
|
||||
|
||||
#[async_trait]
|
||||
impl Provider for StubProvider {
|
||||
fn name(&self) -> &str {
|
||||
"stub-provider"
|
||||
}
|
||||
|
||||
async fn list_models(&self) -> owlen_core::Result<Vec<owlen_core::types::ModelInfo>> {
|
||||
Ok(vec![ModelInfo {
|
||||
id: "stub-model".into(),
|
||||
name: "Stub Model".into(),
|
||||
description: Some("Stub model for screenshot generation".into()),
|
||||
provider: self.name().into(),
|
||||
context_window: Some(8_192),
|
||||
capabilities: vec!["chat".into(), "tool-use".into()],
|
||||
supports_tools: true,
|
||||
}])
|
||||
}
|
||||
|
||||
async fn send_prompt(
|
||||
&self,
|
||||
_request: owlen_core::types::ChatRequest,
|
||||
) -> owlen_core::Result<owlen_core::types::ChatResponse> {
|
||||
Ok(owlen_core::types::ChatResponse {
|
||||
message: Message::assistant("stub completion".into()),
|
||||
usage: None,
|
||||
is_streaming: false,
|
||||
is_final: true,
|
||||
})
|
||||
}
|
||||
|
||||
async fn stream_prompt(
|
||||
&self,
|
||||
_request: owlen_core::types::ChatRequest,
|
||||
) -> owlen_core::Result<owlen_core::ChatStream> {
|
||||
Ok(Box::pin(futures_util::stream::empty()))
|
||||
}
|
||||
|
||||
async fn health_check(&self) -> owlen_core::Result<()> {
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn as_any(&self) -> &(dyn std::any::Any + Send + Sync) {
|
||||
self
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user