@difizen/mana-app/src/file-tree/file-tree-label-provider.ts

import type { URI } from '@difizen/mana-common';
import { singleton, inject } from '@difizen/mana-syringe';

import type { DidChangeLabelEvent } from '../label';
import { LabelProvider, LabelProviderContribution, URIIconReference } from '../label';
import { TreeLabelProvider } from '../tree/tree-label-provider';

import { FileStatNode } from './file-tree';
import type { FileStat } from './files';

@singleton({ contrib: LabelProviderContribution })
export class FileTreeLabelProvider implements LabelProviderContribution {
  protected readonly labelProvider: LabelProvider;
  protected readonly treeLabelProvider: TreeLabelProvider;

  constructor(
    @inject(LabelProvider)
    labelProvider: LabelProvider,
    @inject(TreeLabelProvider)
    treeLabelProvider: TreeLabelProvider,
  ) {
    this.labelProvider = labelProvider;
    this.treeLabelProvider = treeLabelProvider;
  }
  protected asURIIconReference(element: FileStat): URI | URIIconReference {
    return URIIconReference.create(
      element.isDirectory ? 'folder' : 'file',
      element.resource,
    );
  }
  canHandle(element: object): number {
    return FileStatNode.is(element) ? this.treeLabelProvider.canHandle(element) + 1 : 0;
  }

  getIcon(node: FileStatNode): string {
    return this.labelProvider.getIcon(this.asURIIconReference(node.fileStat));
  }

  getName(node: FileStatNode): string {
    return this.labelProvider.getName(this.asURIIconReference(node.fileStat));
  }

  getDescription(node: FileStatNode): string {
    return this.labelProvider.getLongName(this.asURIIconReference(node.fileStat));
  }

  affects(node: FileStatNode, event: DidChangeLabelEvent): boolean {
    return event.affects(node.fileStat);
  }
}