import { BaseModel } from "./BaseModel";
import { User } from "./User";
import { Query } from "./Queries";
import { PricingPlan } from "./PricingPlan";

export class Subscription extends BaseModel {
  static tableName = "subscriptions";

  id!: number;
  user_id!: number;
  queries_id?: number;
  plan_id?: number;
  used_token?: string;
  left_token?: string;
  used_words?: string;
  left_words?: string;
  created_at?: string;
  updated_at?: string;
  deleted_at?: string | null;

  query?: Query | null;
  pricingPlans?: PricingPlan | null;

  $formatDatabaseJson(json: any) {
    json = super.$formatDatabaseJson(json);

    const formatDate = (d: Date | string) => {
      if (!d) return null;
      const date = new Date(d);
      const pad = (n: number) => n.toString().padStart(2, "0");
      return `${date.getFullYear()}-${pad(date.getMonth() + 1)}-${pad(
        date.getDate()
      )} ${pad(date.getHours())}:${pad(date.getMinutes())}:${pad(
        date.getSeconds()
      )}`;
    };

    if (json.created_at) json.created_at = formatDate(json.created_at);
    if (json.updated_at) json.updated_at = formatDate(json.updated_at);
    if (json.deleted_at) json.deleted_at = formatDate(json.deleted_at);

    return json;
  }

  // Relationship to User
  static relationMappings = {
    user: {
      relation: BaseModel.BelongsToOneRelation,
      modelClass: User,
      join: {
        from: "subscriptions.user_id",
        to: "users.id",
      },
    },
    query: {
      relation: BaseModel.BelongsToOneRelation,
      modelClass: Query,
      join: {
        from: "subscriptions.queries_id",
        to: "queries.id",
      },
    },
    pricingPlans: {
      relation: BaseModel.BelongsToOneRelation,
      modelClass: PricingPlan,
      join: {
        from: "subscriptions.plan_id",
        to: "pricing_plans.id",
      },
    },
  };
}