import { join, dirname } from 'path';
import { mkdirp } from 'mkdirp';
import { TelescopeBuilder } from '../builder';
import pkg from '../../package.json';
import { writeContentToFile } from '../utils/files';
import { external, internal, reactQuery, mobx, grpcGateway, grpcWeb, pinia, internalForBigInt, varint, utf8, binary } from '../helpers';

const version = process.env.NODE_ENV === 'test' ? 'latest' : pkg.version;
const header = `/**
* This file and any referenced files were automatically generated by ${pkg.name}@${version}
* DO NOT MODIFY BY HAND. Instead, download the latest proto files for your chain
* and run the transpile command or yarn proto command to regenerate this bundle.
*/
\n`;

const write = (
  builder: TelescopeBuilder,
  indexFile: string,
  content: string
) => {
  const indexOutFile = join(builder.outPath, indexFile);
  mkdirp.sync(dirname(indexOutFile));
  writeContentToFile(builder.outPath, builder.options, header + content, indexOutFile);
}

export const plugin = (
  builder: TelescopeBuilder
) => {
  write(builder, 'helpers.ts', builder.options.prototypes.typingsFormat.num64 === 'bigint' ? internalForBigInt : internal);

  // should be exported
  if (builder.options.includeExternalHelpers || builder.options.reactQuery?.enabled) {
    // also react-query needs these...
    builder.files.push('extern.ts');
    write(builder, 'extern.ts', external);
  }

  if (builder.options.reactQuery?.enabled) {
    builder.files.push('react-query.ts');
    write(builder, 'react-query.ts', reactQuery);
  }

  if (builder.options.mobx?.enabled) {
    builder.files.push('mobx.ts');
    write(builder, 'mobx.ts', mobx);
  }

  if (builder.options.pinia?.enabled) {
    builder.files.push('pinia-endpoint.ts');
    write(builder, 'pinia-endpoint.ts', pinia);
  }

  if (builder.options.rpcClients?.type === 'grpc-gateway') {
    builder.files.push('grpc-gateway.ts');
    write(builder, 'grpc-gateway.ts', grpcGateway);
  }

  if (builder.options.rpcClients?.type === 'grpc-web') {
    builder.files.push('grpc-web.ts');
    write(builder, 'grpc-web.ts', grpcWeb);
  }

  if (builder.options.prototypes.typingsFormat.num64 === "bigint") {
    builder.files.push("varint.ts");
    write(builder, "varint.ts", varint);

    builder.files.push("utf8.ts");
    write(builder, "utf8.ts", utf8);

    builder.files.push("binary.ts");
    write(builder, "binary.ts", binary);
  }

};