// Copyright 2015-2020 Parity Technologies (UK) Ltd.
// This file is part of OpenEthereum.

// OpenEthereum is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// OpenEthereum is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with OpenEthereum.  If not, see <http://www.gnu.org/licenses/>.

use std::sync::Arc;

use accounts::AccountProvider;
use crypto::publickey::{self, Address, Error, Public};
use ethkey::Password;

/// An implementation of EngineSigner using internal account management.
pub struct EngineSigner {
    accounts: Arc<AccountProvider>,
    address: Address,
    password: Password,
}

impl EngineSigner {
    /// Creates new `EngineSigner` given account manager and account details.
    pub fn new(accounts: Arc<AccountProvider>, address: Address, password: Password) -> Self {
        EngineSigner {
            accounts,
            address,
            password,
        }
    }
}

impl ethcore::engines::EngineSigner for EngineSigner {
    fn sign(&self, message: publickey::Message) -> Result<publickey::Signature, publickey::Error> {
        match self
            .accounts
            .sign(self.address, Some(self.password.clone()), message)
        {
            Ok(ok) => Ok(ok),
            Err(e) => Err(publickey::Error::Custom(e.to_string())),
        }
    }

    fn decrypt(&self, auth_data: &[u8], cipher: &[u8]) -> Result<Vec<u8>, Error> {
        self.accounts
            .decrypt(self.address, None, auth_data, cipher)
            .map_err(|e| {
                warn!("Unable to decrypt message: {:?}", e);
                Error::InvalidMessage
            })
    }

    fn address(&self) -> Address {
        self.address
    }

    fn public(&self) -> Option<Public> {
        self.accounts
            .account_public(self.address, &self.password)
            .ok()
    }
}
