dougbot/bridge/lib/mineflayer/lib/bedrockPlugins/bossbar.mts

import type { BedrockBot } from '../../index.js';
import { createRequire } from 'module';
const require = createRequire(import.meta.url);

export default function inject(bot: BedrockBot) {
  const BossBar = require('../bossbar')(bot.registry);
  const bars: Record<string, any> = {};

  bot.on('entityGone', (entity) => {
    if (!entity) return;
    if (!(entity.unique_id in bars)) return;
    bot.emit('bossBarDeleted', bars[entity.unique_id]);
    delete bars[entity.unique_id];
  });

  bot._client.on('boss_event', (packet) => {
    if (packet.type === 'show_bar') {
      let flags = 0;

      if (packet.screen_darkening === 1) {
        flags |= 0x1;
      }

      bars[packet.boss_entity_id] = new BossBar(packet.boss_entity_id, packet.title, getBossHealth(packet.boss_entity_id), 4, packet.color, flags);

      bot._client.write('boss_event', {
        boss_entity_id: packet.boss_entity_id,
        type: 'register_player',
        player_id: bot.entity.unique_id,
      });

      bot.emit('bossBarCreated', bars[packet.boss_entity_id]);
    } else if (packet.type === 'set_bar_progress') {
      if (!(packet.boss_entity_id in bars)) {
        return;
      }

      bars[packet.boss_entity_id].health = getBossHealth(packet.boss_entity_id);

      bot.emit('bossBarUpdated', bars[packet.boss_entity_id]);
    }
  });

  function getBossHealth(boss_id: string | number) {
    let boss_entity = (bot as any).fetchEntity(boss_id);

    let boss_health = 0;

    if ('minecraft:health' in boss_entity.attributes) {
      boss_health = boss_entity.attributes['minecraft:health'].value;
    }

    return boss_health;
  }

  Object.defineProperty(bot, 'bossBars', {
    get() {
      return Object.values(bars);
    },
  });
}