import { Context, Kind, ObjectMap, Type } from "@apexlang/core/model";
import { isPrimitive, isRecursiveType } from "../../utils/index.js";
import {
  rustDoc,
  rustify,
  rustifyCaps,
  trimLines,
  deriveDirective,
  useSerde,
  visibility,
  types,
  customAttributes,
} from "../utils/index.js";

import { SourceGenerator } from "./base.js";

export class StructVisitor extends SourceGenerator<Type> {
  config: ObjectMap<any>;
  visibility: visibility;

  constructor(type: Type, context: Context) {
    super(type, context);
    this.config = context.config;
    this.visibility = visibility(this.root.name, this.config);
  }

  getSource(): string {
    let prefix = trimLines([
      rustDoc(this.root.description),
      deriveDirective(this.root.name, this.config),
      customAttributes(this.root.name, this.config),
    ]);

    return `
    ${prefix}
    ${this.visibility} struct ${rustifyCaps(this.root.name)}{
      ${this.source}
    }`;
  }

  visitTypeField(context: Context): void {
    const { field } = context;
    let isRecursive = isRecursiveType(field.type);
    let isHeapAllocated =
      field.type.kind === Kind.Map || field.type.kind === Kind.List;
    let baseType = types.apexToRustType(field.type, context.config);
    let typeString =
      isRecursive && !isHeapAllocated ? `Box<${baseType}>` : baseType;

    let serdeAnnotation = "";
    if (useSerde(context.config)) {
      let date_with = "";
      if (
        isPrimitive(field.type) &&
        field.type.name === "datetime" &&
        !this.config.datetime
      ) {
        date_with = ',with = "time::serde::rfc3339"';
      }
      serdeAnnotation = `#[serde(rename = "${field.name}"${date_with})]`;
    }

    const vis = visibility(this.root.name, this.config);

    this.append(
      `${trimLines([rustDoc(field.description), serdeAnnotation])}
      ${this.visibility} ${rustify(field.name)}: ${typeString},
      `.trim()
    );
  }
}