nightshade-editor 0.36.0

//! Standalone editor: hosts the same web bundle the browser runs, served from
//! a local port into a native webview window. The build script stages the
//! bundle: a local `trunk` build when one exists, otherwise the published
//! bundle downloaded from the matching GitHub release. Debug builds read the
//! staged copy from disk so a fresh `trunk build` shows up on the next run;
//! release builds embed it into the executable.

#[cfg(feature = "agent")]
mod agent;

use rust_embed::RustEmbed;
use winit::application::ApplicationHandler;
use winit::event::WindowEvent;
use winit::event_loop::{ActiveEventLoop, ControlFlow, EventLoop};
use winit::window::{Window, WindowId};
use wry::{WebView, WebViewBuilder};

#[derive(RustEmbed)]
#[folder = "$OUT_DIR/dist"]
struct Dist;

fn content_type(path: &str) -> &'static str {
    let extension = path.rsplit('.').next().unwrap_or_default();
    match extension {
        "html" => "text/html; charset=utf-8",
        "js" => "application/javascript",
        "wasm" => "application/wasm",
        "css" => "text/css",
        "png" => "image/png",
        "svg" => "image/svg+xml",
        "json" => "application/json",
        _ => "application/octet-stream",
    }
}

/// Serves the bundle on a localhost port from a background thread and returns
/// the port. Localhost is a secure context, so WebGPU and module workers
/// behave exactly as they do in a browser tab. A fixed port is preferred so
/// the page origin is stable and settings persisted in local storage (like
/// the MCP toggle) survive relaunches; a second instance falls back to an
/// ephemeral port.
fn serve_dist() -> u16 {
    const PREFERRED_PORT: u16 = 8780;
    let server = tiny_http::Server::http((std::net::Ipv4Addr::LOCALHOST, PREFERRED_PORT))
        .or_else(|_| tiny_http::Server::http((std::net::Ipv4Addr::LOCALHOST, 0)))
        .expect("failed to bind localhost");
    let port = server
        .server_addr()
        .to_ip()
        .expect("expected an ip address")
        .port();
    std::thread::spawn(move || {
        for request in server.incoming_requests() {
            let path = request.url().split('?').next().unwrap_or("/");
            let path = path.trim_start_matches('/');
            let path = if path.is_empty() { "index.html" } else { path };
            match Dist::get(path) {
                Some(file) => {
                    let header = tiny_http::Header::from_bytes(
                        &b"Content-Type"[..],
                        content_type(path).as_bytes(),
                    )
                    .expect("static header is valid");
                    let response =
                        tiny_http::Response::from_data(file.data.into_owned()).with_header(header);
                    let _ = request.respond(response);
                }
                None => {
                    let _ = request.respond(tiny_http::Response::empty(404));
                }
            }
        }
    });
    port
}

struct App {
    port: u16,
    window: Option<Window>,
    webview: Option<WebView>,
}

impl ApplicationHandler for App {
    fn resumed(&mut self, event_loop: &ActiveEventLoop) {
        if self.window.is_some() {
            return;
        }
        let attributes = Window::default_attributes()
            .with_title("Nightshade Editor")
            .with_maximized(true);
        let window = event_loop
            .create_window(attributes)
            .expect("failed to create window");

        let builder = WebViewBuilder::new()
            .with_url(format!("http://127.0.0.1:{}/", self.port))
            .with_navigation_handler(|url| {
                url.starts_with("http://127.0.0.1") || url.starts_with("https://127.0.0.1")
            });
        #[cfg(feature = "agent")]
        let builder = builder.with_ipc_handler(|request| {
            if request.body() == "enable-mcp" {
                agent::start();
            }
        });
        #[cfg(target_os = "windows")]
        let builder = {
            use wry::WebViewBuilderExtWindows;
            builder.with_additional_browser_args("--enable-features=WebGPU")
        };
        let webview = builder.build(&window).expect("failed to create webview");

        self.window = Some(window);
        self.webview = Some(webview);
    }

    fn window_event(
        &mut self,
        event_loop: &ActiveEventLoop,
        _window_id: WindowId,
        event: WindowEvent,
    ) {
        if let WindowEvent::CloseRequested = event {
            event_loop.exit();
        }
    }
}

fn main() {
    if Dist::get("index.html").is_none() {
        eprintln!("the web bundle is missing from this build");
        eprintln!("from the repo, build it with `just dist` and rebuild");
        eprintln!("from cargo install, reinstall with network access so the build");
        eprintln!("can download the published bundle from the GitHub release");
        std::process::exit(1);
    }
    #[cfg(feature = "agent")]
    if std::env::args().any(|argument| argument == "--mcp") {
        agent::start();
    }
    let port = serve_dist();
    let event_loop = EventLoop::new().expect("failed to create event loop");
    event_loop.set_control_flow(ControlFlow::Wait);
    let mut app = App {
        port,
        window: None,
        webview: None,
    };
    event_loop.run_app(&mut app).expect("event loop failed");
}