import { Model, DataTypes, Optional } from 'sequelize';
import sequelize from '../storage/db';
import { WhereWhatPair } from 'maxun-core';

interface RobotMeta {
  name: string;
  id: string;
  createdAt: string;
  pairs: number;
  updatedAt: string;
  params: any[];
  type?: 'extract' | 'scrape' | 'crawl' | 'search';
  url?: string;
  formats?: ('markdown' | 'html' | 'screenshot-visible' | 'screenshot-fullpage')[];
  isLLM?: boolean;
}

interface RobotWorkflow {
  workflow: WhereWhatPair[];
}

interface WebhookConfig {
  id: string;
  url: string;
  events: string[];
  active: boolean;
  createdAt: string;
  updatedAt: string;
  lastCalledAt?: string | null;
  retryAttempts?: number;
  retryDelay?: number; 
  timeout?: number; 
}

interface RobotAttributes {
  id: string;
  userId?: number;
  recording_meta: RobotMeta;
  recording: RobotWorkflow;
  google_sheet_email?: string | null;
  google_sheet_name?: string | null;
  google_sheet_id?: string | null;
  google_access_token?: string | null;
  google_refresh_token?: string | null;
  airtable_base_id?: string | null; 
  airtable_base_name?: string | null; 
  airtable_table_name?: string | null; 
  airtable_access_token?: string | null; 
  airtable_refresh_token?: string | null; 
  schedule?: ScheduleConfig | null;
  airtable_table_id?: string | null;
  webhooks?: WebhookConfig[] | null; 
}

interface ScheduleConfig {
  runEvery: number;
  runEveryUnit: 'MINUTES' | 'HOURS' | 'DAYS' | 'WEEKS' | 'MONTHS';
  startFrom: 'SUNDAY' | 'MONDAY' | 'TUESDAY' | 'WEDNESDAY' | 'THURSDAY' | 'FRIDAY' | 'SATURDAY';
  atTimeStart?: string;
  atTimeEnd?: string;
  timezone: string;
  lastRunAt?: Date;
  nextRunAt?: Date;
  dayOfMonth?: string;
  cronExpression?: string;
}

interface RobotCreationAttributes extends Optional<RobotAttributes, 'id'> { }

class Robot extends Model<RobotAttributes, RobotCreationAttributes> implements RobotAttributes {
  public id!: string;
  public userId!: number;
  public recording_meta!: RobotMeta;
  public recording!: RobotWorkflow;
  public google_sheet_email!: string | null;
  public google_sheet_name!: string | null;
  public google_sheet_id!: string | null;
  public google_access_token!: string | null;
  public google_refresh_token!: string | null;
  public airtable_base_id!: string | null; 
  public airtable_base_name!: string | null; 
  public airtable_table_name!: string | null; 
  public airtable_access_token!: string | null; 
  public airtable_refresh_token!: string | null; 
  public airtable_table_id!: string | null; 
  public schedule!: ScheduleConfig | null;
  public webhooks!: WebhookConfig[] | null;
}

Robot.init(
  {
    id: {
      type: DataTypes.UUID,
      defaultValue: DataTypes.UUIDV4,
      primaryKey: true,
    },
    userId: {
      type: DataTypes.INTEGER,
      allowNull: false,
    },
    recording_meta: {
      type: DataTypes.JSONB,
      allowNull: false,
    },
    recording: {
      type: DataTypes.JSONB,
      allowNull: false,
    },
    google_sheet_email: {
      type: DataTypes.STRING,
      allowNull: true,
    },
    google_sheet_name: {
      type: DataTypes.STRING,
      allowNull: true,
    },
    google_sheet_id: {
      type: DataTypes.STRING,
      allowNull: true,
    },
    google_access_token: {
      type: DataTypes.STRING,
      allowNull: true,
    },
    google_refresh_token: {
      type: DataTypes.STRING,
      allowNull: true,
    },
    airtable_base_id: {
      type: DataTypes.STRING,
      allowNull: true,
    },
    airtable_base_name: {
      type: DataTypes.STRING,
      allowNull: true,
    },
    airtable_table_name: {
      type: DataTypes.STRING,
      allowNull: true,
    },
    airtable_table_id: {
      type: DataTypes.STRING,
      allowNull: true,
    },
    airtable_access_token: {
      type: DataTypes.TEXT,
      allowNull: true,
    },
    airtable_refresh_token: {
      type: DataTypes.TEXT,
      allowNull: true,
    },
    schedule: {
      type: DataTypes.JSONB,
      allowNull: true,
    },
    webhooks: {
      type: DataTypes.JSONB,
      allowNull: true,
      defaultValue: null,
    },
  },
  {
    sequelize,
    tableName: 'robot',
    timestamps: false,
  }
);

export default Robot;