oglens 0.1.1

Tools for viewing OGP
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

use std::path::PathBuf;

use anyhow::{Context, Result};
use clap::{CommandFactory, Parser};
use oglens::{builder::Prefix, input, ogp::Ogp};
use tokio::io::BufReader;

#[derive(Parser, Debug)]
#[clap(author, version, about, long_about = None)]
pub struct App {
    /// HTML file path
    #[clap(value_parser)]
    path: Option<PathBuf>,

    /// Displayed in JSON format
    #[clap(short, long)]
    json: bool,

    /// Prefix settings other than OG. Specify key separated by colons. ex: --prefix=twitter:name
    #[clap(short, long, value_parser = Prefix::value_parser)]
    prefix: Option<Vec<Prefix>>,
}

impl App {
    pub async fn run(self) -> Result<()> {
        if self.path.is_none() && !input::is_pipe() {
            Self::command()
                .print_long_help()
                .with_context(|| "failed")?;
            return Ok(());
        }

        let raw = match self.path.clone() {
            Some(p) => {
                let file = tokio::fs::File::open(p).await?;
                let mut reader = BufReader::new(file);
                input::read(&mut reader).await
            }
            None => input::read_stdin().await,
        }?;

        let ogp = Ogp::from(&raw, self.prefix.unwrap_or_default())?;

        let output = if self.json {
            ogp.render()?
        } else {
            format!("{}", ogp)
        };
        input::write_stdout(output.as_bytes()).await?;

        Ok(())
    }
}