logos-messaging-js/packages/enr/src/keypair/index.ts

import { unmarshalPrivateKey, unmarshalPublicKey } from "@libp2p/crypto/keys";
import { supportedKeys } from "@libp2p/crypto/keys";
import type { PeerId } from "@libp2p/interface-peer-id";
import { peerIdFromKeys } from "@libp2p/peer-id";

import { Secp256k1Keypair } from "./secp256k1";
import { IKeypair, KeypairType } from "./types";

export const ERR_TYPE_NOT_IMPLEMENTED = "Keypair type not implemented";
export * from "./types";
export * from "./secp256k1";

export function createKeypair(
  type: KeypairType,
  privateKey?: Uint8Array,
  publicKey?: Uint8Array
): IKeypair {
  switch (type) {
    case KeypairType.secp256k1:
      return new Secp256k1Keypair(privateKey, publicKey);
    default:
      throw new Error(ERR_TYPE_NOT_IMPLEMENTED);
  }
}

export async function createPeerIdFromKeypair(
  keypair: IKeypair
): Promise<PeerId> {
  switch (keypair.type) {
    case KeypairType.secp256k1: {
      const publicKey = new supportedKeys.secp256k1.Secp256k1PublicKey(
        keypair.publicKey
      );

      const privateKey = keypair.hasPrivateKey()
        ? new supportedKeys.secp256k1.Secp256k1PrivateKey(keypair.privateKey)
        : undefined;

      return peerIdFromKeys(publicKey.bytes, privateKey?.bytes);
    }
    default:
      throw new Error(ERR_TYPE_NOT_IMPLEMENTED);
  }
}

export async function createKeypairFromPeerId(
  peerId: PeerId
): Promise<IKeypair> {
  let keypairType;
  switch (peerId.type) {
    case "RSA":
      keypairType = KeypairType.rsa;
      break;
    case "Ed25519":
      keypairType = KeypairType.ed25519;
      break;
    case "secp256k1":
      keypairType = KeypairType.secp256k1;
      break;
    default:
      throw new Error("Unsupported peer id type");
  }

  const publicKey = peerId.publicKey
    ? unmarshalPublicKey(peerId.publicKey)
    : undefined;
  const privateKey = peerId.privateKey
    ? await unmarshalPrivateKey(peerId.privateKey)
    : undefined;

  return createKeypair(
    keypairType,
    privateKey?.marshal(),
    publicKey?.marshal()
  );
}