type tweaks

Author: TateB

src/async.ts

@@ -1,24 +1,31 @@
 import { coinTypeMap } from "./consts/coinTypeMap.js";
 import type {
+  Coin,
   CoinName,
   CoinType,
   CoinTypeInvertedReference,
   Formats,
 } from "./types.js";
 
-export const getCoderByCoinNameAsync = async <TCoinName extends CoinName>(
+export const getCoderByCoinNameAsync = async <
+  TCoinName extends CoinName | string = string
+>(
   name: TCoinName
-): Promise<Formats[TCoinName]> => {
+): Promise<TCoinName extends CoinName ? Formats[TCoinName] : Coin> => {
   const mod = await import(`./coin/${name}`);
   if (!mod) {
     throw new Error(`Unsupported coin: ${name}`);
   }
   return mod[name];
 };
 
-export const getCoderByCoinTypeAsync = async <TCoinType extends CoinType>(
+export const getCoderByCoinTypeAsync = async <
+  TCoinType extends CoinType | number = number
+>(
   coinType: TCoinType
-): Promise<CoinTypeInvertedReference[TCoinType]> => {
+): Promise<
+  TCoinType extends CoinType ? CoinTypeInvertedReference[TCoinType] : Coin
+> => {
   const name = coinTypeMap[String(coinType) as keyof typeof coinTypeMap];
   if (!name) {
     throw new Error(`Unsupported coin type: ${coinType}`);

src/index.ts

@@ -1,32 +1,51 @@
 import * as formats from "./coins.js";
 import { coinTypeMap } from "./consts/coinTypeMap.js";
 import type {
+  Coin,
   CoinName,
   CoinType,
   CoinTypeInvertedReference,
+  DecoderFunction,
+  EncoderFunction,
   Formats,
 } from "./types.js";
 
-export const getCoderByCoinName = <TCoinName extends CoinName>(
+export type {
+  Coin,
+  CoinName,
+  CoinType,
+  CoinTypeInvertedReference,
+  DecoderFunction,
+  EncoderFunction,
+  Formats,
+};
+
+export const getCoderByCoinName = <
+  TCoinName extends CoinName | string = string
+>(
   name: TCoinName
-): Formats[TCoinName] => {
-  const format = formats[name];
+): TCoinName extends CoinName ? Formats[TCoinName] : Coin => {
+  const format = formats[name as keyof typeof formats];
   if (!format) {
     throw new Error(`Unsupported coin: ${name}`);
   }
-  return format;
+  return format as TCoinName extends CoinName ? Formats[TCoinName] : Coin;
 };
 
-export const getCoderByCoinType = <TCoinType extends CoinType>(
+export const getCoderByCoinType = <
+  TCoinType extends CoinType | number = number
+>(
   coinType: TCoinType
-): CoinTypeInvertedReference[TCoinType] => {
+): TCoinType extends CoinType ? CoinTypeInvertedReference[TCoinType] : Coin => {
   const name = coinTypeMap[String(coinType) as keyof typeof coinTypeMap];
   if (!name) {
     throw new Error(`Unsupported coin type: ${coinType}`);
   }
-  const format = formats[name] as CoinTypeInvertedReference[TCoinType];
+  const format = formats[name];
   if (!format) {
     throw new Error(`Unsupported coin type: ${coinType}`);
   }
-  return format;
+  return format as TCoinType extends CoinType
+    ? CoinTypeInvertedReference[TCoinType]
+    : Coin;
 };