Implement REPL functionality
This commit is contained in:
parent
7d15f27465
commit
bddb79d0ca
13 changed files with 295 additions and 18 deletions
|
@ -95,9 +95,10 @@ where
|
|||
None => None,
|
||||
}
|
||||
} else {
|
||||
if self.tokenizer.end_of_file() {
|
||||
/*if self.tokenizer.end_of_file() {
|
||||
return None;
|
||||
}
|
||||
}*/
|
||||
//println!("try get next statement");
|
||||
// build new statement
|
||||
let token_result = self
|
||||
.tokenizer
|
||||
|
|
|
@ -29,13 +29,13 @@ impl MpsLanguageDictionary {
|
|||
Some(x) => Ok(x),
|
||||
None => Err(SyntaxError {
|
||||
line: 0,
|
||||
token: MpsToken::Name("???".into()),
|
||||
token: MpsToken::Name("{something}".into()),
|
||||
got: None,
|
||||
}),
|
||||
}?;
|
||||
Err(SyntaxError {
|
||||
line: 0,
|
||||
token: MpsToken::Name("???".into()),
|
||||
token: MpsToken::Name("{any of many}".into()),
|
||||
got: Some(result),
|
||||
})
|
||||
}
|
||||
|
|
|
@ -2,7 +2,7 @@ use std::collections::VecDeque;
|
|||
//use std::fmt::{Debug, Display, Error, Formatter};
|
||||
use std::marker::PhantomData;
|
||||
|
||||
use crate::lang::utility::{assert_token, assert_token_raw, assert_token_raw_back};
|
||||
use crate::lang::utility::{assert_token, assert_token_raw, assert_token_raw_back, assert_empty};
|
||||
use crate::lang::MpsLanguageDictionary;
|
||||
use crate::lang::SyntaxError;
|
||||
use crate::lang::{BoxedMpsOpFactory, MpsOp};
|
||||
|
@ -67,8 +67,8 @@ impl<Op: MpsOp + 'static, F: MpsFunctionFactory<Op> + 'static> BoxedMpsOpFactory
|
|||
)?;
|
||||
assert_token_raw(MpsToken::OpenBracket, tokens)?;
|
||||
assert_token_raw_back(MpsToken::CloseBracket, tokens)?;
|
||||
Ok(Box::new(
|
||||
self.op_factory.build_function_params(name, tokens, dict)?,
|
||||
))
|
||||
let func = self.op_factory.build_function_params(name, tokens, dict)?;
|
||||
assert_empty(tokens)?;
|
||||
Ok(Box::new(func))
|
||||
}
|
||||
}
|
||||
|
|
|
@ -169,6 +169,17 @@ pub fn assert_type(tokens: &mut VecDeque<MpsToken>) -> Result<MpsTypePrimitive,
|
|||
}
|
||||
}
|
||||
|
||||
pub fn assert_empty(tokens: &mut VecDeque<MpsToken>) -> Result<(), SyntaxError> {
|
||||
match tokens.pop_front() {
|
||||
None => Ok(()),
|
||||
Some(t) => Err(SyntaxError {
|
||||
line: 0,
|
||||
token: MpsToken::Name("{nothing}".into()),
|
||||
got: Some(t),
|
||||
}),
|
||||
}
|
||||
}
|
||||
|
||||
pub fn music_folder() -> PathBuf {
|
||||
dirs::home_dir()
|
||||
.unwrap_or_else(|| PathBuf::from("./"))
|
||||
|
|
|
@ -49,6 +49,7 @@ where
|
|||
{
|
||||
byte_buf[0] = 0; // clear to null char (nothing read is assumed to mean end of file)
|
||||
}
|
||||
//println!("tokenizer read char: {}", byte_buf[0]);
|
||||
self.do_tracking(byte_buf[0]);
|
||||
self.fsm = self.fsm.next_state(byte_buf[0]);
|
||||
let mut bigger_buf: Vec<u8> = Vec::new();
|
||||
|
@ -176,6 +177,7 @@ where
|
|||
fn next_statement(&mut self, buf: &mut VecDeque<MpsToken>) -> Result<(), ParseError> {
|
||||
// read until buffer gets some tokens, in case multiple end of line tokens are at start of stream
|
||||
let original_size = buf.len();
|
||||
self.read_line(buf)?; // always try once, even if at end of file
|
||||
while original_size == buf.len() && !self.end_of_file() {
|
||||
self.read_line(buf)?;
|
||||
}
|
||||
|
|
|
@ -17,13 +17,29 @@ pub struct MpsController {
|
|||
|
||||
impl MpsController {
|
||||
pub fn create<F: FnOnce() -> MpsPlayer<T> + Send + 'static, T: MpsTokenReader>(
|
||||
player_gen: F
|
||||
player_gen: F,
|
||||
) -> Self {
|
||||
let (control_tx, control_rx) = channel();
|
||||
let (event_tx, event_rx) = channel();
|
||||
let mut sys_ctrl = SystemControlWrapper::new(control_tx.clone());
|
||||
sys_ctrl.init();
|
||||
let handle = MpsPlayerServer::spawn(player_gen, control_tx.clone(), control_rx, event_tx);
|
||||
let handle = MpsPlayerServer::spawn(player_gen, control_tx.clone(), control_rx, event_tx, false);
|
||||
Self {
|
||||
control: control_tx,
|
||||
event: event_rx,
|
||||
handle: handle,
|
||||
sys_ctrl: sys_ctrl,
|
||||
}
|
||||
}
|
||||
|
||||
pub fn create_repl<F: FnOnce() -> MpsPlayer<T> + Send + 'static, T: MpsTokenReader>(
|
||||
player_gen: F,
|
||||
) -> Self {
|
||||
let (control_tx, control_rx) = channel();
|
||||
let (event_tx, event_rx) = channel();
|
||||
let mut sys_ctrl = SystemControlWrapper::new(control_tx.clone());
|
||||
sys_ctrl.init();
|
||||
let handle = MpsPlayerServer::spawn(player_gen, control_tx.clone(), control_rx, event_tx, true);
|
||||
Self {
|
||||
control: control_tx,
|
||||
event: event_rx,
|
||||
|
@ -58,7 +74,8 @@ impl MpsController {
|
|||
match event {
|
||||
PlayerAction::Acknowledge(_) => Ok(()),
|
||||
PlayerAction::Exception(e) => Err(e),
|
||||
PlayerAction::End => Ok(())
|
||||
PlayerAction::End => Ok(()),
|
||||
PlayerAction::Empty => Ok(()),
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -112,4 +129,55 @@ impl MpsController {
|
|||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub fn wait_for_empty(&self) -> Result<(), PlaybackError> {
|
||||
for msg in self.event.try_iter() {
|
||||
Self::handle_event(msg)?;
|
||||
}
|
||||
self.control.send(ControlAction::CheckEmpty{ack: true}).map_err(PlaybackError::from_err)?;
|
||||
loop {
|
||||
let msg = self.event.recv().map_err(PlaybackError::from_err)?;
|
||||
if let PlayerAction::Empty = msg {
|
||||
break;
|
||||
} else {
|
||||
Self::handle_event(msg)?;
|
||||
}
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Check for any errors in the event queue.
|
||||
/// This is non-blocking, so it only handles events sent since the last time events were handled.
|
||||
pub fn check(&self) -> Vec<PlaybackError> {
|
||||
let mut result = Vec::new();
|
||||
for msg in self.event.try_iter() {
|
||||
if let Err(e) = Self::handle_event(msg) {
|
||||
result.push(e);
|
||||
}
|
||||
}
|
||||
result
|
||||
}
|
||||
|
||||
/// Like check(), but it also waits for an acknowledgement to ensure it gets the latest events.
|
||||
pub fn check_ack(&self) -> Vec<PlaybackError> {
|
||||
let mut result = Vec::new();
|
||||
let to_send = ControlAction::NoOp{ack: true};
|
||||
if let Err(e) = self.control.send(to_send.clone()).map_err(PlaybackError::from_err) {
|
||||
result.push(e);
|
||||
}
|
||||
for msg in self.event.iter() {
|
||||
if let PlayerAction::Acknowledge(action) = msg {
|
||||
if action == to_send {
|
||||
break;
|
||||
} else {
|
||||
result.push(PlaybackError {
|
||||
msg: "Incorrect acknowledgement received for MpsController control action".into()
|
||||
});
|
||||
}
|
||||
} else if let Err(e) = Self::handle_event(msg) {
|
||||
result.push(e);
|
||||
}
|
||||
}
|
||||
result
|
||||
}
|
||||
}
|
||||
|
|
|
@ -11,6 +11,10 @@ impl PlaybackError {
|
|||
msg: format!("{}", err),
|
||||
}
|
||||
}
|
||||
|
||||
pub fn message(&self) -> &'_ str {
|
||||
&self.msg
|
||||
}
|
||||
}
|
||||
|
||||
impl Display for PlaybackError {
|
||||
|
|
|
@ -83,6 +83,7 @@ impl<T: MpsTokenReader> MpsPlayer<T> {
|
|||
items_left -= 1;
|
||||
if items_left == 0 { break; }
|
||||
}
|
||||
//println!("Enqueued {} items", count - items_left);
|
||||
Ok(())
|
||||
}
|
||||
|
||||
|
|
|
@ -10,14 +10,16 @@ pub struct MpsPlayerServer<T: MpsTokenReader> {
|
|||
player: MpsPlayer<T>,
|
||||
control: Receiver<ControlAction>,
|
||||
event: Sender<PlayerAction>,
|
||||
keep_alive: bool,
|
||||
}
|
||||
|
||||
impl<T: MpsTokenReader> MpsPlayerServer<T> {
|
||||
pub fn new(player: MpsPlayer<T>, ctrl: Receiver<ControlAction>, event: Sender<PlayerAction>) -> Self {
|
||||
pub fn new(player: MpsPlayer<T>, ctrl: Receiver<ControlAction>, event: Sender<PlayerAction>, keep_alive: bool) -> Self {
|
||||
Self {
|
||||
player: player,
|
||||
control: ctrl,
|
||||
event: event,
|
||||
keep_alive: keep_alive,
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -27,11 +29,14 @@ impl<T: MpsTokenReader> MpsPlayerServer<T> {
|
|||
if let Err(e) = self.player.enqueue(1) {
|
||||
self.event.send(PlayerAction::Exception(e)).unwrap();
|
||||
}
|
||||
let mut is_empty = self.player.queue_len() == 0;
|
||||
loop {
|
||||
let command = self.control.recv().unwrap();
|
||||
|
||||
let mut is_exiting = false;
|
||||
|
||||
let mut check_empty = false;
|
||||
|
||||
// process command
|
||||
match command {
|
||||
ControlAction::Next{..} => {
|
||||
|
@ -66,6 +71,9 @@ impl<T: MpsTokenReader> MpsPlayerServer<T> {
|
|||
ControlAction::NoOp{..} => {}, // empty by design
|
||||
ControlAction::SetVolume{volume,..} => {
|
||||
self.player.set_volume((volume as f32) / (u32::MAX as f32));
|
||||
},
|
||||
ControlAction::CheckEmpty{..} => {
|
||||
check_empty = true;
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -75,7 +83,7 @@ impl<T: MpsTokenReader> MpsPlayerServer<T> {
|
|||
self.event.send(PlayerAction::Exception(e)).unwrap();
|
||||
}
|
||||
if self.player.queue_len() == 0 { // no more music to add
|
||||
is_exiting = true;
|
||||
is_exiting = !self.keep_alive || is_exiting;
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -83,8 +91,21 @@ impl<T: MpsTokenReader> MpsPlayerServer<T> {
|
|||
self.event.send(PlayerAction::Acknowledge(command)).unwrap();
|
||||
}
|
||||
|
||||
// always check for empty state change
|
||||
if self.player.queue_len() == 0 && !is_empty { // just became empty
|
||||
is_empty = true;
|
||||
self.event.send(PlayerAction::Empty).unwrap();
|
||||
} else if self.player.queue_len() != 0 && is_empty { // just got filled
|
||||
is_empty = false;
|
||||
}
|
||||
|
||||
if is_empty && check_empty {
|
||||
self.event.send(PlayerAction::Empty).unwrap();
|
||||
}
|
||||
|
||||
if is_exiting { break; }
|
||||
}
|
||||
println!("Exiting playback server");
|
||||
self.event.send(PlayerAction::End).unwrap();
|
||||
}
|
||||
|
||||
|
@ -92,12 +113,13 @@ impl<T: MpsTokenReader> MpsPlayerServer<T> {
|
|||
factory: F,
|
||||
ctrl_tx: Sender<ControlAction>,
|
||||
ctrl_rx: Receiver<ControlAction>,
|
||||
event: Sender<PlayerAction>
|
||||
event: Sender<PlayerAction>,
|
||||
keep_alive: bool
|
||||
) -> JoinHandle<()> {
|
||||
thread::spawn(move || Self::unblocking_timer_loop(ctrl_tx, 50));
|
||||
thread::spawn(move || {
|
||||
let player = factory();
|
||||
let mut server_obj = Self::new(player, ctrl_rx, event);
|
||||
let mut server_obj = Self::new(player, ctrl_rx, event, keep_alive);
|
||||
server_obj.run_loop();
|
||||
})
|
||||
}
|
||||
|
@ -126,7 +148,8 @@ pub enum ControlAction {
|
|||
Exit{ack: bool},
|
||||
Enqueue {amount: usize, ack: bool},
|
||||
NoOp{ack: bool},
|
||||
SetVolume{ack: bool, volume: u32}
|
||||
SetVolume{ack: bool, volume: u32},
|
||||
CheckEmpty{ack: bool},
|
||||
}
|
||||
|
||||
impl ControlAction {
|
||||
|
@ -142,6 +165,7 @@ impl ControlAction {
|
|||
Self::Enqueue{ack,..} => ack,
|
||||
Self::NoOp{ack,..} => ack,
|
||||
Self::SetVolume{ack,..} => ack,
|
||||
Self::CheckEmpty{ack} => ack,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
@ -152,6 +176,7 @@ pub enum PlayerAction {
|
|||
Acknowledge(ControlAction),
|
||||
Exception(PlaybackError),
|
||||
End,
|
||||
Empty,
|
||||
}
|
||||
|
||||
impl PlayerAction {
|
||||
|
|
62
src/channel_io.rs
Normal file
62
src/channel_io.rs
Normal file
|
@ -0,0 +1,62 @@
|
|||
use std::sync::mpsc::{channel, Sender, Receiver};
|
||||
use std::io::{Read, Write, self};
|
||||
|
||||
pub struct ChannelWriter {
|
||||
tx: Sender<u8>,
|
||||
}
|
||||
|
||||
impl Write for ChannelWriter {
|
||||
fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
|
||||
let mut count = 0;
|
||||
for &b in buf {
|
||||
self.tx.send(b).map_err(|e| {
|
||||
eprintln!("Send error: {}", e);
|
||||
io::Error::new(io::ErrorKind::ConnectionRefused, e)
|
||||
})?;
|
||||
count += 1;
|
||||
}
|
||||
Ok(count)
|
||||
}
|
||||
|
||||
fn flush(&mut self) -> io::Result<()> {
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
pub struct ChannelReader {
|
||||
rx: Receiver<u8>,
|
||||
blocking: bool,
|
||||
}
|
||||
|
||||
impl Read for ChannelReader {
|
||||
fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
|
||||
let mut count = 0;
|
||||
if self.blocking {
|
||||
for b in self.rx.iter() {
|
||||
buf[count] = b;
|
||||
count += 1;
|
||||
if count >= buf.len() {break;}
|
||||
}
|
||||
} else {
|
||||
for b in self.rx.try_iter() {
|
||||
buf[count] = b;
|
||||
count += 1;
|
||||
if count >= buf.len() {break;}
|
||||
}
|
||||
}
|
||||
Ok(count)
|
||||
}
|
||||
}
|
||||
|
||||
pub fn channel_io() -> (ChannelWriter, ChannelReader) {
|
||||
let (sender, receiver) = channel();
|
||||
(
|
||||
ChannelWriter {
|
||||
tx: sender,
|
||||
},
|
||||
ChannelReader {
|
||||
rx: receiver,
|
||||
blocking: false,
|
||||
}
|
||||
)
|
||||
}
|
|
@ -9,6 +9,9 @@ pub struct CliArgs {
|
|||
/// Generate m3u8 playlist
|
||||
#[clap(short, long)]
|
||||
pub playlist: Option<String>,
|
||||
/// In REPL mode, wait for all music in the queue to complete
|
||||
#[clap(short, long)]
|
||||
pub wait: bool,
|
||||
}
|
||||
|
||||
pub fn parse() -> CliArgs {
|
||||
|
|
|
@ -4,7 +4,9 @@
|
|||
//! Future home of a MPS REPL for playing music ergonomically through a CLI.
|
||||
//!
|
||||
|
||||
mod channel_io;
|
||||
mod cli;
|
||||
mod repl;
|
||||
|
||||
use std::io;
|
||||
use std::path::PathBuf;
|
||||
|
@ -61,8 +63,9 @@ fn main() {
|
|||
}
|
||||
} else {
|
||||
// start REPL
|
||||
// TODO
|
||||
eprintln!("Abort: Cannot start REPL because it is not implemented yet :(");
|
||||
println!("Welcome to MPS interactive mode!");
|
||||
println!("End a statement with ; to execute it.");
|
||||
repl::repl(args)
|
||||
}
|
||||
}
|
||||
|
||||
|
|
97
src/repl.rs
Normal file
97
src/repl.rs
Normal file
|
@ -0,0 +1,97 @@
|
|||
//! Read, Execute, Print Loop functionality
|
||||
|
||||
use std::io::{self, Write, Read};
|
||||
|
||||
use mps_interpreter::MpsRunner;
|
||||
use mps_player::{MpsPlayer, MpsController};
|
||||
|
||||
use super::cli::CliArgs;
|
||||
use super::channel_io::channel_io;
|
||||
|
||||
pub fn repl(args: CliArgs) {
|
||||
let (mut writer, reader) = channel_io();
|
||||
let player_builder = move || {
|
||||
let runner = MpsRunner::with_stream(reader);
|
||||
let player = MpsPlayer::new(runner).unwrap();
|
||||
player
|
||||
};
|
||||
let stdin_t = io::stdin();
|
||||
let mut stdin = stdin_t.lock();
|
||||
let mut buf: Vec<u8> = Vec::new();
|
||||
let mut read_buf = [0];
|
||||
let mut current_line = 0;
|
||||
// TODO: enable raw mode (char by char) reading of stdin
|
||||
// TODO: generalize loop for better code reuse between playlist and playback mode
|
||||
if let Some(playlist_file) = &args.playlist {
|
||||
println!("Playlist mode (output: `{}`)", playlist_file);
|
||||
let mut player = player_builder();
|
||||
let mut playlist_writer = io::BufWriter::new(
|
||||
std::fs::File::create(playlist_file)
|
||||
.expect(&format!("Abort: Cannot create writeable file `{}`", playlist_file))
|
||||
);
|
||||
prompt(&mut current_line);
|
||||
loop {
|
||||
read_buf[0] = 0;
|
||||
while read_buf[0] == 0 {
|
||||
stdin.read_exact(&mut read_buf).expect("Failed to read stdin");
|
||||
}
|
||||
match read_buf[0] as char {
|
||||
';' => {
|
||||
buf.push(read_buf[0]);
|
||||
writer.write(buf.as_slice()).expect("Failed to write to MPS interpreter");
|
||||
match player.save_m3u8(&mut playlist_writer) {
|
||||
Ok(_) => {},
|
||||
Err(e) => eprintln!("{}", e.message()),
|
||||
}
|
||||
buf.clear();
|
||||
playlist_writer.flush().expect("Failed to flush playlist to file");
|
||||
prompt(&mut current_line);
|
||||
},
|
||||
/*
|
||||
'\x27' => break, // ESC key
|
||||
'\x03' => break, // Ctrl+C
|
||||
*/
|
||||
_ => buf.push(read_buf[0]),
|
||||
}
|
||||
}
|
||||
} else {
|
||||
println!("Playback mode (output: audio device)");
|
||||
let ctrl = MpsController::create_repl(player_builder);
|
||||
prompt(&mut current_line);
|
||||
loop {
|
||||
read_buf[0] = 0;
|
||||
while read_buf[0] == 0 {
|
||||
stdin.read_exact(&mut read_buf).expect("Failed to read stdin");
|
||||
}
|
||||
match read_buf[0] as char {
|
||||
';' => {
|
||||
buf.push(read_buf[0]);
|
||||
writer.write(buf.as_slice()).expect("Failed to write to MPS interpreter");
|
||||
//ctrl.play().expect("Failed to start playback");
|
||||
if args.wait {
|
||||
match ctrl.wait_for_empty() {
|
||||
Ok(_) => {},
|
||||
Err(e) => eprintln!("{}", e.message()),
|
||||
}
|
||||
} else {
|
||||
for e in ctrl.check_ack() {
|
||||
eprintln!("{}", e.message());
|
||||
}
|
||||
}
|
||||
buf.clear();
|
||||
prompt(&mut current_line);
|
||||
},
|
||||
'\x27' => break, // ESC key
|
||||
'\x03' => break, // Ctrl+C
|
||||
_ => buf.push(read_buf[0]),
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[inline(always)]
|
||||
fn prompt(line: &mut usize) {
|
||||
print!("{} |-> ", line);
|
||||
*line += 1;
|
||||
std::io::stdout().flush().expect("Failed to flush stdout");
|
||||
}
|
Loading…
Reference in a new issue