// DO NOT EDIT ! // This file was generated automatically from 'src/mako/cli/main.rs.mako' // DO NOT EDIT ! #![feature(plugin, exit_status)] #![plugin(docopt_macros)] #![allow(unused_variables, unused_imports, dead_code, unused_mut)] extern crate docopt; extern crate yup_oauth2 as oauth2; extern crate yup_hyper_mock as mock; extern crate rustc_serialize; extern crate serde; extern crate hyper; extern crate mime; extern crate google_admin2_email_migration as api; use std::env; use std::io::{self, Write}; docopt!(Options derive Debug, " Usage: admin2-email-migration [options] mail insert -r ... -u (simple|resumable) [-p ...] admin2-email-migration --help All documentation details can be found at http://byron.github.io/google-apis-rs/google_admin2_email_migration_cli/index.html Configuration: --scope Specify the authentication a method should be executed in. Each scope requires the user to grant this application permission to use it. If unset, it defaults to the shortest scope url for a particular method. --config-dir A directory into which we will store our persistent data. Defaults to a user-writable directory that we will create during the first invocation. [default: ~/.google-service-cli] --debug Output all server communication to standard error. `tx` and `rx` are placed into the same stream. --debug-auth Output all communication related to authentication to standard error. `tx` and `rx` are placed into the same stream. "); mod cmn; use cmn::{InvalidOptionsError, CLIError, JsonTokenStorage, arg_from_str, writer_from_opts, parse_kv_arg, input_file_from_opts, input_mime_from_opts, FieldCursor, FieldError}; use std::default::Default; use std::str::FromStr; use oauth2::{Authenticator, DefaultAuthenticatorDelegate}; use rustc_serialize::json; struct Engine { opt: Options, hub: api::Admin>, } impl Engine { fn _mail_insert(&self, dry_run: bool, err: &mut InvalidOptionsError) -> Option { let mut request = api::MailItem::default(); let mut field_cursor = FieldCursor::default(); for kvarg in self.opt.arg_kv.iter() { let last_errc = err.issues.len(); let (key, value) = parse_kv_arg(&*kvarg, err, false); let mut temp_cursor = field_cursor.clone(); if let Err(field_err) = temp_cursor.set(&*key) { err.issues.push(field_err); } if value.is_none() { field_cursor = temp_cursor.clone(); if err.issues.len() > last_errc { err.issues.remove(last_errc); } continue; } match &temp_cursor.to_string()[..] { "is-trash" => { request.is_trash = Some(arg_from_str(value.unwrap_or("false"), err, "is-trash", "boolean")); }, "kind" => { request.kind = Some(value.unwrap_or("").to_string()); }, "labels" => { if request.labels.is_none() { request.labels = Some(Default::default()); } request.labels.as_mut().unwrap().push(value.unwrap_or("").to_string()); }, "is-draft" => { request.is_draft = Some(arg_from_str(value.unwrap_or("false"), err, "is-draft", "boolean")); }, "is-inbox" => { request.is_inbox = Some(arg_from_str(value.unwrap_or("false"), err, "is-inbox", "boolean")); }, "is-sent" => { request.is_sent = Some(arg_from_str(value.unwrap_or("false"), err, "is-sent", "boolean")); }, "is-starred" => { request.is_starred = Some(arg_from_str(value.unwrap_or("false"), err, "is-starred", "boolean")); }, "is-unread" => { request.is_unread = Some(arg_from_str(value.unwrap_or("false"), err, "is-unread", "boolean")); }, "is-deleted" => { request.is_deleted = Some(arg_from_str(value.unwrap_or("false"), err, "is-deleted", "boolean")); }, _ => { err.issues.push(CLIError::Field(FieldError::Unknown(temp_cursor.to_string()))); } } } let mut call = self.hub.mail().insert(request, &self.opt.arg_user_key); for parg in self.opt.arg_v.iter() { let (key, value) = parse_kv_arg(&*parg, err, false); match key { "alt" |"fields" |"key" |"oauth-token" |"pretty-print" |"quota-user" |"user-ip" => { let map = [ ("oauth-token", "oauth_token"), ("pretty-print", "prettyPrint"), ("quota-user", "quotaUser"), ("user-ip", "userIp"), ]; call = call.param(map.iter().find(|t| t.0 == key).unwrap_or(&("", key)).1, value.unwrap_or("unset")) }, _ => err.issues.push(CLIError::UnknownParameter(key.to_string())), } } let protocol = if self.opt.cmd_simple { "simple" } else if self.opt.cmd_resumable { "resumable" } else { unreachable!() }; let mut input_file = input_file_from_opts(&self.opt.arg_file, err); let mime_type = input_mime_from_opts(&self.opt.arg_mime, err); if dry_run { None } else { assert!(err.issues.len() == 0); if self.opt.flag_scope.len() > 0 { call = call.add_scope(&self.opt.flag_scope); } match match protocol { "simple" => call.upload(input_file.unwrap(), mime_type.unwrap()), "resumable" => call.upload_resumable(input_file.unwrap(), mime_type.unwrap()), _ => unreachable!(), } { Err(api_err) => Some(api_err), Ok(mut response) => { None } } } } fn _doit(&self, dry_run: bool) -> (Option, Option) { let mut err = InvalidOptionsError::new(); let mut call_result: Option; let mut err_opt: Option = None; if self.opt.cmd_mail { if self.opt.cmd_insert { call_result = self._mail_insert(dry_run, &mut err); } else { unreachable!(); } } else { unreachable!(); } if dry_run { if err.issues.len() > 0 { err_opt = Some(err); } } (call_result, err_opt) } // Please note that this call will fail if any part of the opt can't be handled fn new(opt: Options) -> Result { let (config_dir, secret) = { let config_dir = match cmn::assure_config_dir_exists(&opt.flag_config_dir) { Err(e) => return Err(InvalidOptionsError::single(e, 3)), Ok(p) => p, }; match cmn::application_secret_from_directory(&config_dir, "admin2-email-migration-secret.json", "{\"installed\":{\"auth_uri\":\"https://accounts.google.com/o/oauth2/auth\",\"client_secret\":\"hCsslbCUyfehWMmbkG8vTYxG\",\"token_uri\":\"https://accounts.google.com/o/oauth2/token\",\"client_email\":\"\",\"redirect_uris\":[\"urn:ietf:wg:oauth:2.0:oob\",\"oob\"],\"client_x509_cert_url\":\"\",\"client_id\":\"620010449518-9ngf7o4dhs0dka470npqvor6dc5lqb9b.apps.googleusercontent.com\",\"auth_provider_x509_cert_url\":\"https://www.googleapis.com/oauth2/v1/certs\"}}") { Ok(secret) => (config_dir, secret), Err(e) => return Err(InvalidOptionsError::single(e, 4)) } }; let auth = Authenticator::new( &secret, DefaultAuthenticatorDelegate, if opt.flag_debug_auth { hyper::Client::with_connector(mock::TeeConnector { connector: hyper::net::HttpConnector(None) }) } else { hyper::Client::new() }, JsonTokenStorage { program_name: "admin2-email-migration", db_dir: config_dir.clone(), }, None); let client = if opt.flag_debug { hyper::Client::with_connector(mock::TeeConnector { connector: hyper::net::HttpConnector(None) }) } else { hyper::Client::new() }; let engine = Engine { opt: opt, hub: api::Admin::new(client, auth), }; match engine._doit(true) { (_, Some(err)) => Err(err), _ => Ok(engine), } } // Execute the call with all the bells and whistles, informing the caller only if there was an error. // The absense of one indicates success. fn doit(&self) -> Option { self._doit(false).0 } } fn main() { let opts: Options = Options::docopt().decode().unwrap_or_else(|e| e.exit()); let debug = opts.flag_debug; match Engine::new(opts) { Err(err) => { writeln!(io::stderr(), "{}", err).ok(); env::set_exit_status(err.exit_code); }, Ok(engine) => { if let Some(err) = engine.doit() { if debug { writeln!(io::stderr(), "{:?}", err).ok(); } else { writeln!(io::stderr(), "{}", err).ok(); } env::set_exit_status(1); } } } }