1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92

// Copyright (C) 2022 Parity Technologies (UK) Ltd. (admin@parity.io)
// This file is a part of the scale-value crate.
//
// 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.

use super::ScaleTypeDef as TypeDef;
use scale_info::{form::PortableForm, PortableRegistry, TypeDefBitSequence, TypeDefPrimitive};

/// An error that can occur when we try to encode or decode to a SCALE bit sequence type.
#[derive(Debug, Clone, thiserror::Error, PartialEq)]
pub enum BitSequenceError {
	/// The registry did not contain the bit order type listed.
	#[error("Bit order type {0} not found in registry")]
	BitOrderTypeNotFound(u32),
	/// The registry did not contain the bit store type listed.
	#[error("Bit store type {0} not found in registry")]
	BitStoreTypeNotFound(u32),
	/// The bit order type did not have a valid identifier/name.
	#[error("Bit order cannot be identified")]
	NoBitOrderIdent,
	/// The bit store type that we found was not what we expected (a primitive u8/u16/u32/u64).
	#[error("Bit store type {0} is not supported")]
	StoreTypeNotSupported(String),
	/// The bit order type name that we found was not what we expected ("Lsb0" or "Msb0").
	#[error("Bit order type {0} is not supported")]
	OrderTypeNotSupported(String),
}

/// Obtain details about a bit sequence.
pub fn get_bitsequence_details(
	ty: &TypeDefBitSequence<PortableForm>,
	types: &PortableRegistry,
) -> Result<(BitStoreTy, BitOrderTy), BitSequenceError> {
	let bit_store_ty = ty.bit_store_type().id();
	let bit_order_ty = ty.bit_order_type().id();

	// What is the backing store type expected?
	let bit_store_def = types
		.resolve(bit_store_ty)
		.ok_or(BitSequenceError::BitStoreTypeNotFound(bit_store_ty))?
		.type_def();

	// What is the bit order type expected?
	let bit_order_def = types
		.resolve(bit_order_ty)
		.ok_or(BitSequenceError::BitOrderTypeNotFound(bit_order_ty))?
		.path()
		.ident()
		.ok_or(BitSequenceError::NoBitOrderIdent)?;

	let bit_store_out = match bit_store_def {
		TypeDef::Primitive(TypeDefPrimitive::U8) => Some(BitStoreTy::U8),
		TypeDef::Primitive(TypeDefPrimitive::U16) => Some(BitStoreTy::U16),
		TypeDef::Primitive(TypeDefPrimitive::U32) => Some(BitStoreTy::U32),
		TypeDef::Primitive(TypeDefPrimitive::U64) => Some(BitStoreTy::U64),
		_ => None,
	}
	.ok_or_else(|| BitSequenceError::StoreTypeNotSupported(format!("{bit_store_def:?}")))?;

	let bit_order_out = match &*bit_order_def {
		"Lsb0" => Some(BitOrderTy::Lsb0),
		"Msb0" => Some(BitOrderTy::Msb0),
		_ => None,
	}
	.ok_or(BitSequenceError::OrderTypeNotSupported(bit_order_def))?;

	Ok((bit_store_out, bit_order_out))
}

#[derive(Copy, Clone, PartialEq)]
pub enum BitOrderTy {
	Lsb0,
	Msb0,
}

#[derive(Copy, Clone, PartialEq)]
pub enum BitStoreTy {
	U8,
	U16,
	U32,
	U64,
}