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

use crate::{Manager, Proxy, OS};
pub use abstract_os::os_factory::{
    ExecuteMsgFns as OsFactoryExecFns, QueryMsgFns as OsFactoryQueryFns,
};
use abstract_os::{
    objects::gov_type::GovernanceDetails, os_factory::*, ABSTRACT_EVENT_NAME, MANAGER, PROXY,
};
use boot_core::{
    interface::{BootExecute, ContractInstance},
    prelude::boot_contract,
    state::StateInterface,
    BootEnvironment, Contract, IndexResponse, TxResponse,
};
use cosmwasm_std::Addr;

/// A helper struct that contains fields from [`abstract_os::manager::state::OsInfo`]
#[derive(Default)]
pub struct OsDetails {
    name: String,
    description: Option<String>,
    link: Option<String>,
}

#[boot_contract(InstantiateMsg, ExecuteMsg, QueryMsg, MigrateMsg)]
pub struct OSFactory<Chain>;

impl<Chain: BootEnvironment> OSFactory<Chain> {
    pub fn new(name: &str, chain: Chain) -> Self {
        let mut contract = Contract::new(name, chain);
        contract = contract.with_wasm_path("os_factory");
        Self(contract)
    }

    pub fn create_new_os(
        &self,
        os_details: OsDetails,
        governance_details: GovernanceDetails,
    ) -> Result<OS<Chain>, crate::AbstractBootError> {
        let OsDetails {
            name,
            link,
            description,
        } = os_details;

        let result = self.execute(
            &ExecuteMsg::CreateOs {
                governance: governance_details,
                name,
                link,
                description,
            },
            None,
        )?;

        let manager_address = &result.event_attr_value(ABSTRACT_EVENT_NAME, "manager_address")?;
        self.get_chain()
            .state()
            .set_address(MANAGER, &Addr::unchecked(manager_address));
        let proxy_address = &result.event_attr_value(ABSTRACT_EVENT_NAME, "proxy_address")?;
        self.get_chain()
            .state()
            .set_address(PROXY, &Addr::unchecked(proxy_address));
        Ok(OS {
            manager: Manager::new(MANAGER, self.get_chain().clone()),
            proxy: Proxy::new(PROXY, self.get_chain().clone()),
        })
    }

    pub fn create_default_os(
        &self,
        governance_details: GovernanceDetails,
    ) -> Result<OS<Chain>, crate::AbstractBootError> {
        self.create_new_os(
            OsDetails {
                name: "Default Abstract OS".into(),
                ..Default::default()
            },
            governance_details,
        )
    }

    pub fn set_subscription_contract(
        &self,
        addr: String,
    ) -> Result<TxResponse<Chain>, crate::AbstractBootError> {
        self.execute(
            &ExecuteMsg::UpdateConfig {
                admin: None,
                ans_host_contract: None,
                version_control_contract: None,
                module_factory_address: None,
                subscription_address: Some(addr),
            },
            None,
        )
        .map_err(Into::into)
    }
}