geocode-csv 0.3.0-beta.2

// Async HTTP boilerplate based on
// https://github.com/daboross/futures-example-2019/

#![recursion_limit = "128"]

use common_failures::quick_main;
use failure::Error;
use futures::FutureExt;
use std::{path::PathBuf, result};
use structopt::StructOpt;

mod addresses;
mod async_util;
mod geocoder;
mod smartystreets;
mod structure;
mod unpack_vec;

use addresses::AddressColumnSpec;
use geocoder::{geocode_stdio, OnDuplicateColumns};
use smartystreets::MatchStrategy;
use structure::Structure;

type Result<T> = result::Result<T, Error>;

/// Our command-line arguments.
#[derive(Debug, StructOpt)]
#[structopt(about = "geocode CSV files passed on standard input")]
struct Opt {
    /// `strict` for valid postal addresses only, `range` for unknown addresses
    /// within a street's known range, and `invalid` to always generate some
    /// match.
    #[structopt(long = "match", default_value = "strict")]
    match_strategy: MatchStrategy,

    /// What should we if geocoding output columns have the same names as input
    /// columns? [error, replace, append]
    #[structopt(long = "duplicate-columns", default_value = "error")]
    on_duplicate_columns: OnDuplicateColumns,

    /// A JSON file describing what columns to geocode.
    #[structopt(long = "spec")]
    spec_path: PathBuf,

    /// What license to use. Leave blank for standard, `us-rooftop-geocoding-enterprise-cloud` for Rooftop.
    #[structopt(long = "license", default_value = "us-standard-cloud")]
    license: String,
}

// Generate a boilerplate `main` function.
quick_main!(run);

/// Our main entry point.
fn run() -> Result<()> {
    // Set up basic logging.
    env_logger::init();

    // Parse our command-line arguments.
    let opt = Opt::from_args();
    let spec = AddressColumnSpec::from_path(&opt.spec_path)?;
    let structure = Structure::complete()?;

    // Call our geocoder asynchronously.
    let geocode_fut = geocode_stdio(
        spec,
        opt.match_strategy,
        opt.license,
        opt.on_duplicate_columns,
        structure,
    );

    // Pass our future to our async runtime.
    let mut runtime =
        tokio::runtime::Runtime::new().expect("Unable to create a runtime");
    runtime.block_on(geocode_fut.boxed())?;
    Ok(())
}