import {
  Env,
  Handler,
  InternalHandler,
  KeplrError,
  Message,
} from "@keplr-wallet/router";
import {
  DismissNewTokenFoundInMainMsg,
  GetTokenScansMsg,
  RevalidateTokenScansMsg,
} from "./messages";
import { TokenScanService } from "./service";

export const getHandler: (service: TokenScanService) => Handler = (
  service: TokenScanService
) => {
  return (env: Env, msg: Message<unknown>) => {
    switch (msg.constructor) {
      case GetTokenScansMsg:
        return handleGetTokenScansMsg(service)(env, msg as GetTokenScansMsg);
      case RevalidateTokenScansMsg:
        return handleRevalidateTokenScansMsg(service)(
          env,
          msg as RevalidateTokenScansMsg
        );
      case DismissNewTokenFoundInMainMsg:
        return handleDismissNewTokenFoundInMainMsg(service)(
          env,
          msg as DismissNewTokenFoundInMainMsg
        );
      default:
        throw new KeplrError("tx", 110, "Unknown msg type");
    }
  };
};

const handleGetTokenScansMsg: (
  service: TokenScanService
) => InternalHandler<GetTokenScansMsg> = (service) => {
  return (_, msg) => {
    const tokenScans = service.getTokenScans(msg.vaultId);

    return {
      vaultId: msg.vaultId,
      tokenScans: tokenScans,
      tokenScansWithoutDismissed: tokenScans.filter((scan) =>
        service.isMeaningfulTokenScanChangeBetweenDismissed(scan)
      ),
    };
  };
};

const handleRevalidateTokenScansMsg: (
  service: TokenScanService
) => InternalHandler<RevalidateTokenScansMsg> = (service) => {
  return async (_, msg) => {
    await service.scanAll(msg.vaultId);

    const tokenScans = service.getTokenScans(msg.vaultId);

    return {
      vaultId: msg.vaultId,
      tokenScans: tokenScans,
      tokenScansWithoutDismissed: tokenScans.filter((scan) =>
        service.isMeaningfulTokenScanChangeBetweenDismissed(scan)
      ),
    };
  };
};

const handleDismissNewTokenFoundInMainMsg: (
  service: TokenScanService
) => InternalHandler<DismissNewTokenFoundInMainMsg> = (service) => {
  return async (_, msg) => {
    service.dismissNewTokenFoundInHome(msg.vaultId);

    const tokenScans = service.getTokenScans(msg.vaultId);

    return {
      vaultId: msg.vaultId,
      tokenScans: tokenScans,
      tokenScansWithoutDismissed: tokenScans.filter((scan) =>
        service.isMeaningfulTokenScanChangeBetweenDismissed(scan)
      ),
    };
  };
};