// This file is part of Substrate.

// Copyright (C) Parity Technologies (UK) Ltd.
// SPDX-License-Identifier: Apache-2.0

// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// 	http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

//! Operation on runtime storage using hashed keys.

use super::unhashed;
use alloc::vec::Vec;
use codec::{Decode, Encode};

/// Return the value of the item in storage under `key`, or `None` if there is no explicit entry.
pub fn get<T, HashFn, R>(hash: &HashFn, key: &[u8]) -> Option<T>
where
	T: Decode + Sized,
	HashFn: Fn(&[u8]) -> R,
	R: AsRef<[u8]>,
{
	unhashed::get(hash(key).as_ref())
}

/// Return the value of the item in storage under `key`, or the type's default if there is no
/// explicit entry.
pub fn get_or_default<T, HashFn, R>(hash: &HashFn, key: &[u8]) -> T
where
	T: Decode + Sized + Default,
	HashFn: Fn(&[u8]) -> R,
	R: AsRef<[u8]>,
{
	unhashed::get_or_default(hash(key).as_ref())
}

/// Return the value of the item in storage under `key`, or `default_value` if there is no
/// explicit entry.
pub fn get_or<T, HashFn, R>(hash: &HashFn, key: &[u8], default_value: T) -> T
where
	T: Decode + Sized,
	HashFn: Fn(&[u8]) -> R,
	R: AsRef<[u8]>,
{
	unhashed::get_or(hash(key).as_ref(), default_value)
}

/// Return the value of the item in storage under `key`, or `default_value()` if there is no
/// explicit entry.
pub fn get_or_else<T, F, HashFn, R>(hash: &HashFn, key: &[u8], default_value: F) -> T
where
	T: Decode + Sized,
	F: FnOnce() -> T,
	HashFn: Fn(&[u8]) -> R,
	R: AsRef<[u8]>,
{
	unhashed::get_or_else(hash(key).as_ref(), default_value)
}

/// Put `value` in storage under `key`.
pub fn put<T, HashFn, R>(hash: &HashFn, key: &[u8], value: &T)
where
	T: Encode,
	HashFn: Fn(&[u8]) -> R,
	R: AsRef<[u8]>,
{
	unhashed::put(hash(key).as_ref(), value)
}

/// Remove `key` from storage, returning its value if it had an explicit entry or `None` otherwise.
pub fn take<T, HashFn, R>(hash: &HashFn, key: &[u8]) -> Option<T>
where
	T: Decode + Sized,
	HashFn: Fn(&[u8]) -> R,
	R: AsRef<[u8]>,
{
	unhashed::take(hash(key).as_ref())
}

/// Remove `key` from storage, returning its value, or, if there was no explicit entry in storage,
/// the default for its type.
pub fn take_or_default<T, HashFn, R>(hash: &HashFn, key: &[u8]) -> T
where
	T: Decode + Sized + Default,
	HashFn: Fn(&[u8]) -> R,
	R: AsRef<[u8]>,
{
	unhashed::take_or_default(hash(key).as_ref())
}

/// Return the value of the item in storage under `key`, or `default_value` if there is no
/// explicit entry. Ensure there is no explicit entry on return.
pub fn take_or<T, HashFn, R>(hash: &HashFn, key: &[u8], default_value: T) -> T
where
	T: Decode + Sized,
	HashFn: Fn(&[u8]) -> R,
	R: AsRef<[u8]>,
{
	unhashed::take_or(hash(key).as_ref(), default_value)
}

/// Return the value of the item in storage under `key`, or `default_value()` if there is no
/// explicit entry. Ensure there is no explicit entry on return.
pub fn take_or_else<T, F, HashFn, R>(hash: &HashFn, key: &[u8], default_value: F) -> T
where
	T: Decode + Sized,
	F: FnOnce() -> T,
	HashFn: Fn(&[u8]) -> R,
	R: AsRef<[u8]>,
{
	unhashed::take_or_else(hash(key).as_ref(), default_value)
}

/// Check to see if `key` has an explicit entry in storage.
pub fn exists<HashFn, R>(hash: &HashFn, key: &[u8]) -> bool
where
	HashFn: Fn(&[u8]) -> R,
	R: AsRef<[u8]>,
{
	unhashed::exists(hash(key).as_ref())
}

/// Ensure `key` has no explicit entry in storage.
pub fn kill<HashFn, R>(hash: &HashFn, key: &[u8])
where
	HashFn: Fn(&[u8]) -> R,
	R: AsRef<[u8]>,
{
	unhashed::kill(hash(key).as_ref())
}

/// Get a Vec of bytes from storage.
pub fn get_raw<HashFn, R>(hash: &HashFn, key: &[u8]) -> Option<Vec<u8>>
where
	HashFn: Fn(&[u8]) -> R,
	R: AsRef<[u8]>,
{
	unhashed::get_raw(hash(key).as_ref())
}

/// Put a raw byte slice into storage.
pub fn put_raw<HashFn, R>(hash: &HashFn, key: &[u8], value: &[u8])
where
	HashFn: Fn(&[u8]) -> R,
	R: AsRef<[u8]>,
{
	unhashed::put_raw(hash(key).as_ref(), value)
}
