ownership.ts

import ora from "ora"
import chalk from "chalk"
import { SigningCosmWasmClient } from "@cosmjs/cosmwasm-stargate"
import { GasPrice } from "@cosmjs/stargate";
import { getChainConfig } from "./lighthouse"
import inquirer from "inquirer"
import { Command } from "commander";

export const command_ownership = (root: Command) => {
    root
        .command("update-lighthouse-admin")
        .description("Change the admin of a collection in Lighthouse Core. Will give the new admin full control over the collections lighthouse configuration.")
        .argument("<collection>")
        .argument("<new_admin>")
        .option("-g --gas-price <gas_price>", "Gas price to use for transaction  (default: 0.1)")
        .action(async (collection, admin, options) => {

            let { wallet, lighthouse, rpc_wasm } = await getChainConfig(true)

            const [firstAccount] = await wallet.getAccounts()

            const client = await SigningCosmWasmClient.connectWithSigner(rpc_wasm, wallet, {
                gasPrice: GasPrice.fromString(options.gasPrice ? options.gasPrice + "usei" : "0.1usei")
            })

            //confirm
            const answers = await inquirer.prompt([{
                type: 'confirm',
                name: 'confirm',
                message: 'Are you sure you want to update the lighthouse admin of collection ' + collection + ' to ' + admin + '?',
                default: false
            }])

            if (!answers.confirm) {
                console.log("Exiting")
                return
            }

            let spinner = ora("Updating admin of collection").start()
            const Msg = {
                update_admin: { collection, admin }
            }

            const txReceipt = await client.execute(firstAccount.address, lighthouse, Msg, "auto", "",)

            spinner.succeed("Admin of collection " + collection + " updated to " + admin)
            console.log("Transaction hash: " + chalk.green(txReceipt.transactionHash))
        })

        root
        .command("transfer-cw-ownable-ownership")
        .description("Transfer all CW-Ownable rights of a collection to a new address. Transfer proposal will be sent to the new owner. (CW721 and CW404 collections only) Lighthouse Core should be the owner of the collection to execute this command.")
        .argument("<collection>")
        .argument("<new_owner>")
        .option("-g --gas-price <gas_price>", "Gas price to use for transaction  (default: 0.1)")
        .action(async (collection, admin, options) => {

            let { wallet, lighthouse, rpc_wasm } = await getChainConfig(true)

            const [firstAccount] = await wallet.getAccounts()

            const client = await SigningCosmWasmClient.connectWithSigner(rpc_wasm, wallet, {
                gasPrice: GasPrice.fromString(options.gasPrice ? options.gasPrice + "usei" : "0.1usei")
            })

            const collectionData = await client.queryContractSmart(lighthouse, { get_collection: { collection } })

            //confirm
            const answers = await inquirer.prompt([{
                type: 'confirm',
                name: 'confirm',
                message: 'Are you sure you want to transfer the owner rights of collection ' + collection + ' to ' + admin + '?',
                default: false
            }])

            if (!answers.confirm) {
                console.log("Exiting")
                return
            }

            let spinner = ora("Transferring ownership of collection").start()
            const Msg = {
                update_nft_contract_owner: { collection, new_admin: admin }
            }

            const txReceipt = await client.execute(firstAccount.address, lighthouse, Msg, "auto", "",)

            if (collectionData.chain === "v2") {
                spinner.succeed("Updated Minter role of collection " + collection + " to " + admin)
            } else {
                spinner.succeed("Ownership proposal sent to " + admin + " for collection " + collection)
            }

            console.log("Transaction hash: " + chalk.green(txReceipt.transactionHash))
        })

        root
        .command("accept-cw-ownable-ownership")
        .description("Accept the CW-Ownable ownership proposal for a collection")
        .argument("<collection>")
        .option("-g --gas-price <gas_price>", "Gas price to use for transaction  (default: 0.1)")
        .action(async (collection, options) => {

            let { wallet, lighthouse, rpc_wasm } = await getChainConfig(true)

            const [firstAccount] = await wallet.getAccounts()

            const client = await SigningCosmWasmClient.connectWithSigner(rpc_wasm, wallet, {
                gasPrice: GasPrice.fromString(options.gasPrice ? options.gasPrice + "usei" : "0.1usei")
            })

            const collectionData = await client.queryContractSmart(lighthouse, { get_collection: { collection } })

            if (collectionData.chain === "v2") {
                console.log("Accept ownership is not available for EVM collections")
                return
            }

            //confirm
            const answers = await inquirer.prompt([{
                type: 'confirm',
                name: 'confirm',
                message: 'Are you sure you want to accept the ownership transfer request for collection ' + collection + '?',
                default: false
            }])

            if (!answers.confirm) {
                console.log("Exiting")
                return
            }

            let spinner = ora("Accepting ownership transfer request").start()
            const Msg = collectionData.collection_type === "404" ? {
                accept_ownership: {}
            } : {
                update_ownership: { accept_ownership: {} }
            }

            const txReceipt = await client.execute(firstAccount.address, collection, Msg, "auto", "",)

            spinner.succeed("Ownership of collection " + collection + " transferred to you")
            console.log("Transaction hash: " + chalk.green(txReceipt.transactionHash))
        })

    root
        .command("update-nft-contract-admin")
        .description("Transfer contract admin rights of a collection to a new address. New owner can migrate and alter the collection code. For ERC721 collections, Minter role will be transferred to the new owner. Lighthouse Core should be the owner of the collection to execute this command.")
        .argument("<collection>")
        .argument("<new_owner>")
        .option("-g --gas-price <gas_price>", "Gas price to use for transaction  (default: 0.1)")
        .action(async (collection, admin, options) => {

            let { wallet, lighthouse, rpc_wasm } = await getChainConfig(true)

            const [firstAccount] = await wallet.getAccounts()

            const client = await SigningCosmWasmClient.connectWithSigner(rpc_wasm, wallet, {
                gasPrice: GasPrice.fromString(options.gasPrice ? options.gasPrice + "usei" : "0.1usei")
            })

            const collectionData = await client.queryContractSmart(lighthouse, { get_collection: { collection } })

            //confirm
            const answers = await inquirer.prompt([{
                type: 'confirm',
                name: 'confirm',
                message: 'Are you sure you want to transfer the admin rights of collection ' + collection + ' to ' + admin + '?',
                default: false
            }])

            if (!answers.confirm) {
                console.log("Exiting")
                return
            }

            let spinner = ora("Transferring admin rights of collection").start()
            const Msg = {
                update_nft_contract_admin: { collection, new_admin: admin }
            }

            const txReceipt = await client.execute(firstAccount.address, lighthouse, Msg, "auto", "",)

            if (collectionData.chain === "v2") {
                spinner.succeed("Updated Minter role of collection " + collection + " to " + admin)
            } else {
                spinner.succeed("Admin of collection " + collection + " transferred to " + admin)
            }

            console.log("Transaction hash: " + chalk.green(txReceipt.transactionHash))
        })
    }