import { createPublicClient, createWalletClient, http, parseAbi } from 'viem';
import { kyaChain } from './chains';

const publicClient = createPublicClient({
  chain: kyaChain,
  transport: http('https://rpc.kyachain.xyz'),
});

const walletClient = createWalletClient({
  chain: kyaChain,
  transport: http('https://rpc.kyachain.xyz'),
});

const ORG_REGISTRY = '0x19977A3093FbA73f8DAd3d60ef9A53468A6DaFA9';

// Create a new organization
const { request: createReq } = await publicClient.simulateContract({
  address: ORG_REGISTRY,
  abi: parseAbi([
    'function createOrganization(string name, string metadataURI) external returns (uint256)',
  ]),
  functionName: 'createOrganization',
  args: ['Nethara Labs', 'ipfs://QmOrgMetadata...'],
  account: '0xAdminAddress',
});

const createHash = await walletClient.writeContract(createReq);
const receipt = await publicClient.waitForTransactionReceipt({ hash: createHash });

// Invite an agent to join
const { request: inviteReq } = await publicClient.simulateContract({
  address: ORG_REGISTRY,
  abi: parseAbi(['function inviteAgent(uint256 orgId, uint256 agentId) external']),
  functionName: 'inviteAgent',
  args: [1n, 42n],
  account: '0xAdminAddress',
});

const inviteHash = await walletClient.writeContract(inviteReq);

// Agent owner accepts the invite
const { request: acceptReq } = await publicClient.simulateContract({
  address: ORG_REGISTRY,
  abi: parseAbi(['function acceptInvite(uint256 orgId, uint256 agentId) external']),
  functionName: 'acceptInvite',
  args: [1n, 42n],
  account: '0xAgentOwnerAddress',
});

const acceptHash = await walletClient.writeContract(acceptReq);

// Check membership
const isMember = await publicClient.readContract({
  address: ORG_REGISTRY,
  abi: parseAbi(['function isMember(uint256 orgId, uint256 agentId) external view returns (bool)']),
  functionName: 'isMember',
  args: [1n, 42n],
});

console.log('Is member:', isMember); // true

// Get all organizations for an agent
const orgs = await publicClient.readContract({
  address: ORG_REGISTRY,
  abi: parseAbi(['function getAllAgentOrgs(uint256 agentId) external view returns (uint256[])']),
  functionName: 'getAllAgentOrgs',
  args: [42n],
});

console.log('Agent organizations:', orgs);

// Get all members of an organization
const members = await publicClient.readContract({
  address: ORG_REGISTRY,
  abi: parseAbi(['function getAllOrgMembers(uint256 orgId) external view returns (uint256[])']),
  functionName: 'getAllOrgMembers',
  args: [1n],
});

console.log('Organization members:', members);

// Check maximum organizations for agent's tier
const [name, metadataURI, admin, createdAt] = await publicClient.readContract({
  address: ORG_REGISTRY,
  abi: parseAbi([
    'function getOrganization(uint256 orgId) external view returns (string, string, address, uint256)',
  ]),
  functionName: 'getOrganization',
  args: [1n],
});

console.log('Organization:', { name, metadataURI, admin, createdAt });

[
  "function createOrganization(string calldata name, string calldata metadataURI) external returns (uint256 orgId)",
  "function updateMetadata(uint256 orgId, string calldata newURI) external",
  "function transferAdmin(uint256 orgId, address newAdmin) external",
  "function inviteAgent(uint256 orgId, uint256 agentId) external",
  "function cancelInvite(uint256 orgId, uint256 agentId) external",
  "function acceptInvite(uint256 orgId, uint256 agentId) external",
  "function declineInvite(uint256 orgId, uint256 agentId) external",
  "function removeAgent(uint256 orgId, uint256 agentId) external",
  "function leaveOrganization(uint256 orgId, uint256 agentId) external",
  "function setTrustVault(address _trustVault) external",
  "function getMaxOrgsForTier(uint8 tier) external pure returns (uint256)",
  "function getOrganization(uint256 orgId) external view returns (string memory name, string memory metadataURI, address admin, uint256 createdAt)",
  "function isMember(uint256 orgId, uint256 agentId) external view returns (bool)",
  "function hasPendingInvite(uint256 orgId, uint256 agentId) external view returns (bool)",
  "function getOrgMemberCount(uint256 orgId) external view returns (uint256)",
  "function getAllOrgMembers(uint256 orgId) external view returns (uint256[] memory)",
  "function getAllAgentOrgs(uint256 agentId) external view returns (uint256[] memory)",
  "function getAllPendingInvites(uint256 agentId) external view returns (uint256[] memory)",
  "function registerValidator(address validator) external",
  "function removeValidator(address validator) external",
  "function validateOrg(uint256 orgId, bool passed, string calldata evidence) external",
  "function getOrgValidationCount(uint256 orgId) external view returns (uint256)",
  "function getOrgValidation(uint256 orgId, uint256 index) external view returns (address validator, bool passed, string memory evidence, uint256 timestamp)",
  "event OrganizationCreated(uint256 indexed orgId, string name, address indexed admin)",
  "event MetadataUpdated(uint256 indexed orgId, string newURI)",
  "event AdminTransferred(uint256 indexed orgId, address indexed oldAdmin, address indexed newAdmin)",
  "event AgentInvited(uint256 indexed orgId, uint256 indexed agentId)",
  "event InviteCanceled(uint256 indexed orgId, uint256 indexed agentId)",
  "event AgentJoined(uint256 indexed orgId, uint256 indexed agentId)",
  "event InviteDeclined(uint256 indexed orgId, uint256 indexed agentId)",
  "event AgentRemoved(uint256 indexed orgId, uint256 indexed agentId)",
  "event AgentLeft(uint256 indexed orgId, uint256 indexed agentId)",
  "event ValidatorRegistered(address indexed validator)",
  "event ValidatorRemoved(address indexed validator)",
  "event OrgValidated(uint256 indexed orgId, address indexed validator, bool passed, string evidence)",
  "event TrustVaultUpdated(address indexed oldVault, address indexed newVault)",
  "error MembershipLimitExceeded(uint256 agentId, uint8 tier, uint256 currentCount, uint256 maxAllowed)",
  "error UnverifiedAgent(uint256 agentId)"
]