import type { AccountEntity, AssetEntity, AssetTransactionEntity, NftEntity, SessionEntity, TokenEntity, TransactionEntity, AddressEntity, VaultEntity } from "../../types/db.types";
import type { AssetType } from "../../types/wallet.types";

export interface DBStore {
  clearData(): Promise<void>;
  initSchema(): Promise<boolean>;
  
  upsertTransaction(txEntry: TransactionEntity): Promise<void>;
  upsertAssetTransaction(assetTx: AssetTransactionEntity): Promise<void>;
  getTransactions(account: number, tokenId?: string): Promise<TransactionEntity[]>;
  getPageTransactions(account: number, pageNum: number, pageSize: number, tokenId?: string): Promise<TransactionEntity[]>;
  countTransactions(account: number, tokenId?: string): Promise<number>;
  clearTransactions(account: number): Promise<void>;
  
  countTokens(account: number): Promise<number>;
  upsertToken(token: TokenEntity): Promise<void>;
  getTokens(account: number, pageNum: number, pageSize: number): Promise<TokenEntity[]>;
  getToken(id: string): Promise<TokenEntity | undefined>;
  deleteToken(id: string): Promise<void>;
  
  countNfts(account: number): Promise<number>;
  upsertNft(nft: NftEntity): Promise<void>;
  getNfts(account: number, pageNum: number, pageSize: number): Promise<NftEntity[]>;
  getNft(id: string): Promise<NftEntity | undefined>;
  deleteNft(id: string): Promise<void>;

  upsertAsset(asset: AssetEntity): Promise<void>;
  getAssets(account: number, type: AssetType): Promise<AssetEntity[]>;
  deleteAsset(account: number, id: string): Promise<void>;
  countAssetsById(id: string): Promise<number>;

  upsertAccount(account: AccountEntity): Promise<void>;
  getAccounts(): Promise<AccountEntity[]>;
  countAccounts(): Promise<number>;
  updateAccountName(account: number, name: string): Promise<void>;

  upsertSession(session: SessionEntity): Promise<void>;
  getSessionsByAccount(accountId: number): Promise<SessionEntity[]>;
  deleteSession(sessionId: string): Promise<void>;

  upsertAddress(address: AddressEntity): Promise<void>;
  getReceiveAddresses(): Promise<AddressEntity[]>;
  getChangeAddresses(): Promise<AddressEntity[]>;
  countAddresses(): Promise<number>;

  getVaults(): Promise<VaultEntity[]>;
  upsertVault(vault: VaultEntity): Promise<void>;
}