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

// This file is part of a fork of Substrate which has had various changes.

// Copyright (C) Parity Technologies (UK) Ltd.
// Copyright (C) 2022-2023 Luke Parker
// 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.

//! Implementation of trait `StorageInfoTrait` on module structure.

use super::DeclStorageDefExt;
use proc_macro2::TokenStream;
use quote::quote;

pub fn impl_storage_info(def: &DeclStorageDefExt) -> TokenStream {
	let scrate = &def.hidden_crate;

	let mut res_append_storage = TokenStream::new();

	for line in def.storage_lines.iter() {
		let storage_struct = &line.storage_struct;

		let (trait_, method) = if def.generate_storage_info {
			(quote!(#scrate::traits::StorageInfoTrait), quote!(storage_info))
		} else {
			(quote!(#scrate::traits::PartialStorageInfoTrait), quote!(partial_storage_info))
		};

		res_append_storage.extend(quote!(
			let mut storage_info = <
				#storage_struct as #trait_
			>::#method();
			res.append(&mut storage_info);
		));
	}

	let module_struct = &def.module_struct;
	let module_impl = &def.module_impl;
	let where_clause = &def.where_clause;

	quote!(
		impl #module_impl #scrate::traits::StorageInfoTrait for #module_struct #where_clause {
			fn storage_info() -> #scrate::sp_std::vec::Vec<#scrate::traits::StorageInfo> {
				let mut res = #scrate::sp_std::vec![];
				#res_append_storage
				res
			}
		}
	)
}