pezkuwi-ui/packages/vue-identicon/src/Identicon.ts

// Copyright 2017-2023 @polkadot/vue-identicon authors & contributors
// SPDX-License-Identifier: Apache-2.0

import type { VNode } from 'vue';
import type { Prefix } from '@polkadot/util-crypto/address/types';

import { defineComponent, h } from 'vue';

import { isHex, isU8a, u8aToHex } from '@polkadot/util';
import { decodeAddress, encodeAddress } from '@polkadot/util-crypto';

import { Beachball, Empty, Jdenticon, Polkadot } from './icons/index.js';
import { adaptVNodeAttrs } from './util.js';

interface Account {
  address: string;
  publicKey: string;
}

interface Data {
  address: string;
  iconSize: number;
  isAlternativeIcon: boolean;
  publicKey: string;
  type: 'beachball' | 'empty' | 'jdenticon' | 'polkadot' | 'substrate';
}

const DEFAULT_SIZE = 64;

export function encodeAccount (value: string | Uint8Array, prefix?: Prefix): Account {
  try {
    const address = isU8a(value) || isHex(value)
      ? encodeAddress(value as string, prefix)
      : value;
    const publicKey = u8aToHex(decodeAddress(address, false, prefix));

    return { address, publicKey };
  } catch {
    return { address: '', publicKey: '0x' };
  }
}

/**
 * @name Identicon
 * @description The main Identicon component, taking a number of properties
 * @example
 * ```html
 * <Identicon :size="128" :theme="polkadot" :value="..." />
 * ```
 */
export const Identicon = defineComponent({
  components: {
    Beachball,
    Empty,
    Jdenticon,
    Polkadot
  },
  created: function (): void {
    this.createData();
  },
  data: function (): Data {
    return {
      address: '',
      iconSize: DEFAULT_SIZE,
      isAlternativeIcon: false,
      publicKey: '0x',
      type: 'empty'
    };
  },
  methods: {
    createData: function (): void {
      this.iconSize = this.size as number || DEFAULT_SIZE;
      this.type = this.theme as 'empty';
      this.isAlternativeIcon = this.isAlternative as boolean || false;
      this.recodeAddress();
    },
    recodeAddress: function (): void {
      const { address, publicKey } = encodeAccount(this.value as string);

      this.address = address;
      this.publicKey = publicKey;
    }
  },
  props: ['prefix', 'isAlternative', 'size', 'theme', 'value'],
  render (): VNode {
    const { address, iconSize, isAlternativeIcon, publicKey, type } = this.$data;

    if (type === 'empty') {
      return h(Empty, {
        ...adaptVNodeAttrs({
          key: address,
          size: iconSize
        })
      }, []);
    } else if (type === 'jdenticon') {
      return h(Jdenticon, {
        ...adaptVNodeAttrs(
          {
            key: address,
            publicKey,
            size: iconSize
          }
        )
      }, []);
    } else if (type === 'substrate') {
      throw new Error('substrate type is not supported');
    }

    const cmp = type.charAt(0).toUpperCase() + type.slice(1);

    if (['Beachball', 'Polkadot'].includes(cmp)) {
      const component = cmp === 'Beachball'
        ? Beachball
        : Polkadot;

      return h(component, {
        ...adaptVNodeAttrs({
          address,
          isAlternative: isAlternativeIcon,
          key: address,
          size: iconSize
        })
      }, []);
    } else {
      return h(cmp, {}, []);
    }
  },
  watch: {
    value: function (): void {
      this.recodeAddress();
    }
  }
});