// Copyright 2018 Yuya Nishihara <yuya@tcha.org>

//

// This software may be used and distributed according to the terms of the

// GNU General Public License version 2 or any later version.

use async_trait::async_trait;

use std::io;

use std::os::unix::io::AsRawFd;

use std::os::unix::process::ExitStatusExt;

use std::process::Stdio;

use tokio;

use tokio::process::{Child, ChildStdin, Command};

use crate::message::CommandSpec;

use crate::procutil;

/// Callback to process shell command requests received from server.

#[async_trait]

pub trait SystemHandler {

    type PagerStdin: AsRawFd;

    /// Handles pager command request.

    ///

    /// Returns the pipe to be attached to the server if the pager is spawned.

    /// Handles system command request.

    ///

    /// Returns command exit code (positive) or signal number (negative).

    async fn run_system(&mut self, spec: &CommandSpec) -> io::Result<i32>;

}

/// Default cHg implementation to process requests received from server.

pub struct ChgUiHandler {

    pager: Option<Child>,

}

impl ChgUiHandler {

    pub fn new() -> ChgUiHandler {

        ChgUiHandler { pager: None }

    }

    /// Waits until the pager process exits.

    pub async fn wait_pager(&mut self) -> io::Result<()> {

        if let Some(p) = self.pager.take() {

            p.await?;

        }

        Ok(())

    }

}

#[async_trait]

impl SystemHandler for ChgUiHandler {

    type PagerStdin = ChildStdin;

        let pin = pager.stdin.take().unwrap();

        procutil::set_blocking_fd(pin.as_raw_fd())?;

        // TODO: if pager exits, notify the server with SIGPIPE immediately.

        // otherwise the server won't get SIGPIPE if it does not write

        // anything. (issue5278)

        // kill(peerpid, SIGPIPE);

        self.pager = Some(pager);

        Ok(pin)

    }

    async fn run_system(&mut self, spec: &CommandSpec) -> io::Result<i32> {

        let status = new_shell_command(&spec).spawn()?.await?;

        let code = status

            .code()

            .or_else(|| status.signal().map(|n| -n))

            .expect("either exit code or signal should be set");

        Ok(code)

    }

}

fn new_shell_command(spec: &CommandSpec) -> Command {

    let mut builder = Command::new("/bin/sh");

    builder

        .arg("-c")

        .arg(&spec.command)

        .current_dir(&spec.current_dir)

        .env_clear()

        .envs(spec.envs.iter().cloned());

    builder

}