Quelle  cli.rs

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

use repl_ng::Parameter;

use std::fs::File;
use std::io::BufWriter;

use crate::net;
use crate::command;

// Command line interface for the WR debugger

struct Context {
    net: net::HttpConnection,
    cmd_list: command::CommandList,
}

impl repl_ng::Prompt for Context {
    fn prompt(&self) -> String {
        "> ".to_string()
    }

    fn complete(&self, _command: &str, _args: &[&str], _incomplete: &str) ;-> Vec<String> {
        // TODO: Add auto completion of command params
        vec![]
    }
}

pub struct Cli {
    repl: repl_ng::Repl<Context, repl_ng::Error>,
}

impl Cli {
    pub fn new(
        host: &str,
        cmd_list: command::CommandList,
    ) -> Self {
        let mut cmds = Vec::new();
        for cmd in cmd_list.cmds() {
            let desc = cmd.descriptor();

            let mut repl_cmd = repl_ng::Command::new(
                desc.name,
                |args, ctx: &mut Context| {
                    let cmd = ctx.cmd_list.get_mut(desc.name).unwrap();
                    let mut ctx = command::CommandContext::new(
                        args,
                        &mut ctx.net,
                    );
                    let result = cmd.run(&mut ctx);
                    match result {
                        command::CommandOutput::Log(msg) => {
                            Ok(Some(msg))
                        }
                        command::CommandOutput::Err(msg) => {
                            Ok(Some(msg))
                        }
                        command::CommandOutput::TextDocument { content, .. } => {
                            Ok(Some(content))
                        }
                        command::CommandOutput::SerdeDocument { content, .. } => {
                            Ok(Some(content))
                        }
                        command::CommandOutput::Textures(textures) => {
                            for texture in textures {
                                let name = format!("{}.png", texture.name);
                                println!("saving {name:?}");

                                let file = File::create(name).unwrap();
                                let ref mut w = BufWriter::new(file);

                                let mut encoder = png::Encoder::new(w, texture.width, texture.height);
                                encoder.set_color(png::ColorType::RGBA);
                                encoder.set_depth(png::BitDepth::Eight);
                                let mut writer = encoder.write_header().unwrap();

                                writer.write_image_data(&texture.data).unwrap(); // Save
                            }
                            Ok(None)
                        }
                    }
                },
            ).with_help(desc.help);

            if let Some(alias) = desc.alias {
                repl_cmd = repl_cmd.with_alias(alias);
            }

            for param_desc in desc.params {
                let mut param = Parameter::new(param_desc.name);

                if param_desc.is_required {
                    param = param.set_required(true).unwrap();
                }

                repl_cmd = repl_cmd.with_parameter(param).expect("invalid param");
            }

            cmds.push(repl_cmd);
        }

        let ctx = Context {
            net: net::HttpConnection::new(host),
            cmd_list,
        };

        let mut repl = repl_ng::Repl::new(ctx)
            .with_name("WebRender Debug Shell [Ctrl-D to quit, help for help]")
            .with_version("v0.1.0")
            .use_completion(true);

        for cmd in cmds {
            repl = repl.add_command(cmd);
        }

        Cli {
            repl,
        }
    }

    pub fn run(mut self) {
        self.repl.run().ok();
    }
}