// Copyright 2021 TiKV Project Authors. Licensed under Apache-2.0.

use std::error;

use kvproto::raft_cmdpb::RaftCmdResponse;

use crate::Error;

pub fn bind_term(resp: &mut RaftCmdResponse, term: u64) {
    if term == 0 {
        return;
    }

    resp.mut_header().set_current_term(term);
}

pub fn bind_error(resp: &mut RaftCmdResponse, err: Error) {
    resp.mut_header().set_error(err.into());
}

pub fn new_error(err: Error) -> RaftCmdResponse {
    let mut resp = RaftCmdResponse::default();
    bind_error(&mut resp, err);
    resp
}

pub fn new_with_key_error(
    mut err: Error,
) -> (RaftCmdResponse, Option<Vec<kvproto::kvrpcpb::KeyError>>) {
    let key_errs = match &mut err {
        Error::KeyErrors(key_errs) => Some(std::mem::take(key_errs)),
        _ => None,
    };
    (new_error(err), key_errs)
}

pub fn err_resp(e: Error, term: u64) -> RaftCmdResponse {
    let mut resp = new_error(e);
    bind_term(&mut resp, term);
    resp
}

pub fn message_error<E>(err: E) -> RaftCmdResponse
where
    E: Into<Box<dyn error::Error + Send + Sync>>,
{
    new_error(Error::Other(err.into()))
}