osm_pbf2json 0.2.2

use lib::output::Output;
use lib::{boundaries, filter, objects, streets};
use std::error::Error;
use std::fs::File;
use std::io;
use structopt::StructOpt;

mod lib;

#[derive(StructOpt)]
struct Cli {
    #[structopt(parse(from_os_str))]
    path: std::path::PathBuf,
    #[structopt(subcommand)]
    cmd: Command,
}

#[derive(StructOpt)]
enum Command {
    Objects {
        #[structopt(short, long)]
        tags: String,
    },
    Streets {
        #[structopt(short, long)]
        geojson: bool,
        #[structopt(short, long)]
        name: Option<String>,
        #[structopt(short, long)]
        boundary: Option<u8>,
    },
    Boundaries {
        #[structopt(short, long)]
        geojson: bool,
        #[structopt(short, long)]
        levels: Option<Vec<u8>>,
    },
}

fn main() -> Result<(), Box<dyn Error>> {
    let stdout = io::stdout();
    let mut handle = io::BufWriter::new(stdout);
    let args = Cli::from_args();

    let file = File::open(args.path)?;

    match args.cmd {
        Command::Objects { tags } => {
            let groups = filter::parse(&tags);
            let objects = objects(file, &groups)?;
            objects.write_json_lines(&mut handle)?;
        }
        Command::Streets {
            geojson,
            name,
            boundary,
        } => {
            let streets = streets(file, name.as_deref(), boundary)?;
            if geojson {
                streets.write_geojson(&mut handle)?;
            } else {
                streets.write_json_lines(&mut handle)?;
            }
        }
        Command::Boundaries { levels, geojson } => {
            let boundaries = boundaries(file, levels)?;
            if geojson {
                boundaries.write_geojson(&mut handle)?;
            } else {
                boundaries.write_json_lines(&mut handle)?;
            }
        }
    }
    Ok(())
}