leekTreeItem.ts

import { join } from 'path';
import { ExtensionContext, TreeItem, TreeItemCollapsibleState } from 'vscode';
import globalState from '../globalState';
import { DEFAULT_LABEL_FORMAT } from './constant';
import { FundInfo, IconType, TreeItemType } from './typed';
import { formatLabelString, formatTreeText } from './utils';

export class LeekTreeItem extends TreeItem {
  info: FundInfo;
  type: string | undefined;
  isCategory: boolean;
  contextValue: string | undefined;
  _itemType?: TreeItemType;

  constructor(info: FundInfo, context: ExtensionContext | undefined, isCategory = false) {
    super('', TreeItemCollapsibleState.None);
    this.info = info;
    this.isCategory = isCategory;
    const {
      showLabel,
      isStock,
      name,
      code,
      type,
      symbol,
      percent,
      price,
      open,
      yestclose,
      high,
      low,
      updown,
      volume,
      amount = 0,
      earnings,
      priceDate,
      time,
      isStop,
      t2,
      contextValue,
      _itemType,
    } = info;

    if (_itemType) {
      this._itemType = _itemType;
    } else {
      this._itemType = isStock ? TreeItemType.STOCK : TreeItemType.FUND;
    }

    this.type = type;
    this.contextValue = contextValue;
    let _percent: number | string = Math.abs(percent);
    if (isNaN(_percent)) {
      _percent = '--';
    } else {
      _percent = _percent.toFixed(2);
    }
    let icon = 'up';
    const grow = percent.indexOf('-') === 0 ? false : true;
    const val = Math.abs(percent);
    if (grow) {
      if (IconType.ARROW === globalState.iconType) {
        icon = val >= 2 ? 'up' : 'up1';
      } else if (IconType.FOOD1 === globalState.iconType) {
        icon = 'meat2';
      } else if (IconType.FOOD2 === globalState.iconType) {
        icon = 'kabob';
      } else if (IconType.FOOD3 === globalState.iconType) {
        icon = 'wine';
      } else if (IconType.ICON_FOOD === globalState.iconType) {
        icon = '🍗';
      }
      _percent = '+' + _percent;
    } else {
      if (IconType.ARROW === globalState.iconType) {
        icon = val >= 2 ? 'down' : 'down1';
      } else if (IconType.FOOD1 === globalState.iconType) {
        icon = 'noodles';
      } else if (IconType.FOOD2 === globalState.iconType) {
        icon = 'bakeleek';
      } else if (IconType.FOOD3 === globalState.iconType) {
        icon = 'noodles';
      } else if (IconType.ICON_FOOD === globalState.iconType) {
        icon = '🍜';
      }
      _percent = '-' + _percent;
    }
    if (isStop) {
      icon = 'stop';
    }
    let iconPath: string | undefined = '';
    if (showLabel) {
      iconPath =
        globalState.iconType !== IconType.ICON_FOOD
          ? context?.asAbsolutePath(join('resources', `${icon}.svg`))
          : icon;
    }
    const isIconPath = iconPath?.lastIndexOf('.svg') !== -1;
    if (isIconPath && type !== 'nodata') {
      this.iconPath = iconPath;
    }
    let text = '';

    if (showLabel) {
      /* `showLabel: true` */
      if (this._itemType === TreeItemType.STOCK) {
        const risePercent = isStop ? '停牌' : `${_percent}%`;
        if (type === 'nodata') {
          text = info.name;
        } else {
          /* text = `${!isIconPath ? iconPath : ''}${risePercent}${formatTreeText(
            price,
            15
          )}「${name}」`; */
          text = formatLabelString(
            globalState.labelFormat?.['sidebarStockLabelFormat'] ??
              DEFAULT_LABEL_FORMAT.sidebarStockLabelFormat,
            {
              ...info,
              icon: !isIconPath ? iconPath : '',
              percent: risePercent,
            }
          );
        }
      } else if (this._itemType === TreeItemType.FUND) {
        /* text =
          `${!isIconPath ? iconPath : ''}${formatTreeText(`${_percent}%`)}「${name}」${
            t2 || !(globalState.showEarnings && amount > 0)
              ? ''
              : `(${grow ? '盈' : '亏'}：${grow ? '+' : ''}${earnings})`
          }` + `${t2 ? `(${time})` : ''}`; */
        text = formatLabelString(
          globalState.labelFormat?.['sidebarFundLabelFormat'] ??
            DEFAULT_LABEL_FORMAT.sidebarFundLabelFormat,
          {
            ...info,
            icon: !isIconPath ? iconPath : '',
            percent: `${_percent}%`,
            earnings:
              t2 || !(globalState.showEarnings && amount > 0)
                ? ''
                : `(${grow ? '盈' : '亏'}：${grow ? '+' : ''}${earnings})`,
            time: t2 ? `(${time})` : '',
          }
        );
        // ${earningPercent !== 0 ? '，率：' + earningPercent + '%' : ''}
      } else if (this._itemType === TreeItemType.BINANCE) {
        text = formatLabelString(
          globalState.labelFormat?.['sidebarBinanceLabelFormat'] ??
            DEFAULT_LABEL_FORMAT.sidebarBinanceLabelFormat,
          {
            ...info,
            icon: !isIconPath ? iconPath : '',
            percent: `${_percent}%`,
          }
        );
      }
    } else {
      /* `showLabel: false` */
      text =
        this._itemType === TreeItemType.STOCK
          ? `${formatTreeText(`${_percent}%`, 11)}${formatTreeText(price, 15)} 「${code}」`
          : `${formatTreeText(`${_percent}%`)}「${code}」`;
    }

    if (globalState.enableHighlights === 0) {
      this.label = text;
    } else if (globalState.enableHighlights >= 1) {
      this.label = grow ? { label: text, highlights: [[0, text.length]] } : text;
    } else if (globalState.enableHighlights <= -1) {
      this.label = grow ? text : { label: text, highlights: [[0, text.length]] };
    }

    this.id = info.id || code;

    if (this._itemType === TreeItemType.STOCK || this._itemType === TreeItemType.FUND) {
      this.command = {
        title: name, // 标题
        command:
          this._itemType === TreeItemType.STOCK
            ? 'leek-fund.stockItemClick'
            : 'leek-fund.fundItemClick', // 命令 ID
        arguments: [
          this._itemType === TreeItemType.STOCK ? '0' + symbol : code, // 基金/股票编码
          name, // 基金/股票名称
          text,
          `${type}${symbol}`,
        ],
      };
      if (type === 'nodata') {
        this.command.command = '';
      }
    }

    if (this._itemType === TreeItemType.STOCK) {
      const labelText = !showLabel ? name : '';

      const isFuture = type === 'cnf_';
      // type字段：期货没有前缀，去掉自定义的 `cnf_`。股票的 type 是交易所 (sz,sh)
      const typeText = isFuture ? symbol?.replace('cnf_', '') : type;
      const symbolText = isFuture ? '' : symbol;

      if (type === 'nodata') {
        this.tooltip = '接口不支持，右键删除关注';
      } else {
        this.tooltip = `【今日行情】${labelText}${typeText}${symbolText}\n 涨跌：${updown}   百分比：${_percent}%\n 最高：${high}   最低：${low}\n 今开：${open}   昨收：${yestclose}\n 成交量：${volume}   成交额：${amount}`;
      }
    } else if (this._itemType === TreeItemType.BINANCE) {
      this.tooltip = `【今日行情】${name}\n 涨跌：${updown}   百分比：${_percent}%\n 最高：${high}   最低：${low}\n 今开：${open}   昨收：${yestclose}\n 成交量：${volume}   成交额：${amount}`;
    } else {
      this.tooltip = `「${name}」(${code})`;
    }
  }
}