Quelle  send_ping.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 https://mozilla.org/MPL/2.0/. */

//! An entry point for sending a crash ping.

use crate::std::{env, io::stdin};
use crate::{glean, logging, net::ping};

pub fn main() {
    logging::init();

    let mut args = env::args_os().skip(2);
    let data_path = args.next().expect("no data path provided");
    let reason = args.next().expect("no crash reason provided");

    let extra: serde_json::Value =
        serde_json::from_reader(stdin()).expect("failed to read extra data from stdin");

    let _glean_handle = glean::InitOptions {
        data_dir: data_path.into(),
        locale: None,
        // Assume that this is only invoked to send a ping when upload is enabled.
        upload_enabled: true,
    }
    .init()
    .expect("failed to acquire Glean store");

    ping::CrashPing {
        extra: &extra,
        reason: reason.to_str(),
    }
    .send();

    // Increase our chances of sending the ping immediately by explicitly shutting down Glean.
    ::glean::shutdown();
}

/// Just initialize Glean to allow any unsubmitted pings to be sent.
pub fn cleanup_main() {
    logging::init();

    let mut args = env::args_os().skip(2);
    let data_path = args.next().expect("no data path provided");
    let upload_enabled: bool = args
        .next()
        .expect("upload enabled missing")
        .to_str()
        .expect("non-unicode upload enabled value")
        .parse()
        .expect("invalid upload enabled value");

    let _glean_handle = glean::InitOptions {
        data_dir: data_path.into(),
        locale: None,
        upload_enabled,
    }
    .init()
    .expect("failed to acquire Glean store");

    // Sleep for a short period for Glean to do its thing in the background (and so that
    // `glean::shutdown()` won't log a warning about waiting for init to complete).
    std::thread::sleep(std::time::Duration::from_secs(2));

    // Glean shutdown will block (for a period) on at least one ping to be sent.
    ::glean::shutdown();
}