Force shutdown on second ^C

This commit is contained in:
Paul Lietar 2017-02-21 23:25:04 +00:00
parent 2708aa4fef
commit 2853e2230b

View file

@ -1,18 +1,19 @@
#[macro_use] extern crate log; #[macro_use] extern crate log;
extern crate getopts;
extern crate librespot;
extern crate ctrlc; extern crate ctrlc;
extern crate env_logger; extern crate env_logger;
extern crate futures; extern crate futures;
extern crate getopts;
extern crate librespot;
extern crate tokio_core; extern crate tokio_core;
use env_logger::LogBuilder; use env_logger::LogBuilder;
use std::io::{stderr, Write};
use std::process::exit;
use std::env;
use std::path::PathBuf;
use std::str::FromStr;
use futures::Future; use futures::Future;
use std::cell::{RefCell, Cell};
use std::env;
use std::io::{stderr, Write};
use std::path::PathBuf;
use std::process::exit;
use std::str::FromStr;
use tokio_core::reactor::Core; use tokio_core::reactor::Core;
use librespot::spirc::Spirc; use librespot::spirc::Spirc;
@ -173,10 +174,18 @@ fn main() {
}); });
let (spirc, task) = Spirc::new(session.clone(), player, mixer); let (spirc, task) = Spirc::new(session.clone(), player, mixer);
let spirc = ::std::cell::RefCell::new(spirc); let spirc = RefCell::new(spirc);
let shutting_down = Cell::new(false);
ctrlc::set_handler(move || { ctrlc::set_handler(move || {
if shutting_down.get() {
warn!("Forced shutdown");
exit(1);
} else {
info!("Shutting down");
spirc.borrow_mut().shutdown(); spirc.borrow_mut().shutdown();
shutting_down.set(true);
}
}); });
task.map_err(|()| panic!("spirc error")) task.map_err(|()| panic!("spirc error"))