import { Resource, ResourceLink } from '../model/note';
import { URI } from '../model/uri';
import { FoamWorkspace } from '../model/workspace';
import { IDisposable } from '../common/lifecycle';
import { ResourceProvider } from '../model/provider';

export const imageExtensions = [
  '.png',
  '.jpg',
  '.jpeg',
  '.gif',
  '.svg',
  '.webp',
];

export const defaultAttachmentExtensions = [
  '.pdf',
  '.mp3',
  '.webm',
  '.wav',
  '.m4a',
  '.mp4',
  '.avi',
  '.mov',
  '.rtf',
  '.txt',
  '.doc',
  '.docx',
  '.pages',
  '.xls',
  '.xlsx',
  '.numbers',
  '.ppt',
  '.pptm',
  '.pptx',
];

const asResource = (uri: URI): Resource => {
  const type = imageExtensions.includes(uri.getExtension())
    ? 'image'
    : 'attachment';
  return {
    uri: uri,
    title: uri.getBasename(),
    type: type,
    aliases: [],
    properties: { type: type },
    sections: [],
    blocks: [],
    links: [],
    tags: [],
    footnotes: [],
  };
};

export class AttachmentResourceProvider implements ResourceProvider {
  private disposables: IDisposable[] = [];
  public readonly attachmentExtensions: string[];

  constructor(attachmentExtensions: string[] = []) {
    this.attachmentExtensions = [...imageExtensions, ...attachmentExtensions];
  }

  supports(uri: URI) {
    return this.attachmentExtensions.includes(
      uri.getExtension().toLocaleLowerCase()
    );
  }

  async readAsMarkdown(uri: URI): Promise<string | null> {
    if (imageExtensions.includes(uri.getExtension())) {
      return `![${''}](${uri.toString()}|height=200)`;
    }
    return `### ${uri.getBasename()}`;
  }

  async fetch(uri: URI) {
    return asResource(uri);
  }

  resolveLink(w: FoamWorkspace, resource: Resource, l: ResourceLink) {
    throw new Error('not supported');
    // Silly workaround to make VS Code happy
    return resource.uri;
  }

  dispose() {
    this.disposables.forEach(d => d.dispose());
  }
}