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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114

// This file is part of Astar.

// Copyright (C) Stake Technologies Pte.Ltd.
// SPDX-License-Identifier: GPL-3.0-or-later

// Astar 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.

// Astar 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 Astar. If not, see <http://www.gnu.org/licenses/>.

//! Astar XCM CLI handlers.

use crate::cli::*;

use clap::Parser;
use cumulus_primitives_core::ParaId;
use polkadot_parachain::primitives::Sibling;
use polkadot_primitives::AccountId;
use sp_core::hexdisplay::HexDisplay;
use sp_runtime::traits::AccountIdConversion;
use xcm::latest::prelude::*;
use xcm_builder::{
    DescribeAllTerminal, DescribeFamily, HashedDescription, ParentIsPreset,
    SiblingParachainConvertsVia,
};
use xcm_executor::traits::ConvertLocation;

/// CLI error type.
pub type Error = String;

/// Parse command line arguments into service configuration.
pub fn run() -> Result<(), Error> {
    let cli = Cli::parse();

    match &cli.subcommand {
        Some(Subcommand::RelayChainAccount) => {
            let relay_account =
                ParentIsPreset::<AccountId>::convert_location(&Location::parent()).unwrap();
            println!("{}", relay_account);
        }
        Some(Subcommand::SovereignAccount(cmd)) => {
            let parachain_account = if cmd.sibling {
                let location = Location {
                    parents: 1,
                    interior: Parachain(cmd.parachain_id).into(),
                };
                SiblingParachainConvertsVia::<Sibling, AccountId>::convert_location(&location)
                    .unwrap()
            } else {
                let para_id = ParaId::from(cmd.parachain_id);
                AccountIdConversion::<AccountId>::into_account_truncating(&para_id)
            };
            println!("{}", parachain_account);
        }
        Some(Subcommand::AssetId(cmd)) => {
            const ASSET_PRECOMPILE_ADDRESS_PREFIX: &[u8] = &[255u8; 4];
            let mut data = [0u8; 20];
            data[0..4].copy_from_slice(ASSET_PRECOMPILE_ADDRESS_PREFIX);
            data[4..20].copy_from_slice(&cmd.asset_id.to_be_bytes());
            println!("pallet_assets: {}", cmd.asset_id);
            println!("EVM XC20: 0x{}", HexDisplay::from(&data));
        }
        Some(Subcommand::RemoteAccount(cmd)) => {
            let mut sender_multilocation = Location::parent();

            if let Some(parachain_id) = cmd.parachain_id {
                sender_multilocation
                    .append_with(Parachain(parachain_id))
                    .expect("infallible, short sequence");
            }

            match cmd.account_key {
                AccountWrapper::SS58(id) => {
                    sender_multilocation
                        .append_with(AccountId32 {
                            id,
                            // network is not relevant for account derivation
                            network: None,
                        })
                        .expect("infallible, short sequence");
                }
                AccountWrapper::H160(key) => {
                    sender_multilocation
                        .append_with(AccountKey20 {
                            key,
                            // network is not relevant for account derivation
                            network: None,
                        })
                        .expect("infallible, short sequence");
                }
            }

            let derived_acc =
                HashedDescription::<AccountId, DescribeFamily<DescribeAllTerminal>>::convert_location(
                    &sender_multilocation,
                );
            if let Some(derived_acc) = derived_acc {
                println!("{}", derived_acc);
            } else {
                println!("Failed to derive account Id.");
            }
        }
        None => {}
    }
    Ok(())
}