import { fromBinary } from "@bufbuild/protobuf"

import { ErrorSchema, Frame, Type } from "../proto/rpc"
import { HeadlessClientError, HeadlessClientErrorCode } from "./client"

export enum ServerErrorCode {
  OK = 0,
  Canceled = 1,
  Unknown = 2,
  InvalidArgument = 3,
  DeadlineExceeded = 4,
  NotFound = 5,
  AlreadyExists = 6,
  PermissionDenied = 7,
  ResourceExhausted = 8,
  FailedPrecondition = 9,
  Aborted = 10,
  OutOfRange = 11,
  Unimplemented = 12,
  Internal = 13,
  Unavailable = 14,
  DataLoss = 15,
  Unauthenticated = 16,
  NotSupported = 17,

  // Custom Error
  MPCInitializeProtocolFailed = 20,
  MPCHandshakeProtocolFailed = 21,
  MPCBadSignature = 22,
  MPCSignatureVerifyFailed = 23,
  MPCServerStoreKeyFailed = 24,
  MPCBadResult = 25,
  MPCSendTxRequestFailed = 26,
  MPCBadKey = 27,
  MPCSignMessageFailed = 28,
  MPCSendTxFailed = 29,
  MPCAddressAlreadyExisted = 30,
  MPCChallengeFailed = 31,
  MPCGetNonceFailed = 32,

  DialSocketFailed = 100,
  WriteDataFailed = 101,
  ReadDataFailed = 102,
  BadRPCData = 103,
  BadSignMessageData = 104,
  BadTxData = 105,
  InitHTTPFailed = 106,
  DoHTTPFailed = 107,
  BadHTTPData = 108,
  DialRPCNodeFailed = 109,
  HitRateLimitUUID = 110,
  InvalidNonce = 111,
  BadPayerSignature = 112,
  GetPayerSignatureFailed = 113,
  DecodePayerSignatureFailed = 114,
  InitSiweMessageFailed = 115,
  ParseSiweResourceFailed = 116,
  BadExpiredTime = 117,
  GetPayerAccessTokenFailed = 118,
  BadRPCURL = 119,
  TxNotSponsored = 120,
  TxSponsorNoAppLocalSponsorProgram = 121,
  TxSponsorExceedcap = 122,

  // Policy rules
  PolicyFailed = 1001,
  UserIDNotAllowed = 1002,
  PolicyMatchDenyList = 1003,
  DailyNativeTokenTransferredReachLimit = 1005,
  TxToAddressNotAllowed = 1006,
  TxValueReachedLimit = 1007,
}

type ServerErrorOpts = {
  code: number
  message: string
}

export type ServerErrorType = ServerError & {
  name: "ServerError"
}

export class ServerError extends Error {
  override name = "MpcServerError"
  code: number
  shortMessage: string

  constructor({ code, message }: ServerErrorOpts) {
    const fullMessage = ["", `code: ${code}`, `message: ${message}`].join("\n• ")

    super(fullMessage)

    this.code = code
    this.shortMessage = message
  }
}

export const decodeServerError = (frame: Frame) => {
  if (frame.type === Type.ERROR) {
    const error = fromBinary(ErrorSchema, frame.data)

    return new ServerError({ code: Number(error.code), message: error.message })
  }

  if (frame.type === Type.UNSPECIFIED) {
    return new ServerError({
      code: ServerErrorCode.Unknown,
      message: "unspecified server response",
    })
  }

  return new HeadlessClientError({
    cause: undefined,
    code: HeadlessClientErrorCode.MissingMessageError,
    message:
      "The client has not processed the message from socket. This is most likely the SDK bug, please upgrade to the latest version.",
  })
}