use core::abi::{ByteInput, Decode}

use core::Copy

use core::num::IntDowncast

use ingot::abi::Sol

use ingot::abi::sol::decode_input

use ingot::evm::ops::mload

/// A view over an in-memory byte buffer.

pub struct MemoryBytes {

pub base: u256,

pub len: u256,

}

impl Copy for MemoryBytes {}

impl MemoryBytes {

pub fn new(base: u256, len: u256) -> Self {

MemoryBytes { base, len }

}

/// Decode Solidity ABI data from this memory view.

pub fn decode<T>(self) -> T

where T: Decode<Sol>

{

decode_input(self)

}

}

impl ByteInput for MemoryBytes {

fn len(self) -> u256 {

self.len

}

#[arithmetic(unchecked)]

fn word_at(self, _ byte_offset: u256) -> u256 {

mload(self.base + byte_offset)

}

#[arithmetic(unchecked)]

fn byte_at(self, _ byte_offset: u256) -> u8 {

let word_offset = byte_offset - (byte_offset % 32)

let word = mload(self.base + word_offset)

let idx = byte_offset - word_offset

let shift = (31 - idx) * 8

(word >> shift).downcast_truncate()

}

}