const getTableSchema = async (tableName: string) => {
  console.log("Getting schema");
  const schemaQuery = `SELECT column_name, data_type, is_nullable FROM information_schema.columns WHERE table_name = '${tableName}' ORDER BY ordinal_position;`;

  const command = new ExecuteStatementCommand({
    database: "root" /* required */,
    formatRecordsAs: "JSON",
    includeResultMetadata: true,
    resourceArn: dbArn,
    secretArn: secretArn,
    sql: schemaQuery,
  });

  const response = await rdsdataservice.send(command);
  console.log(response);
  if (response.formattedRecords) {
    return JSON.parse(response.formattedRecords);
  }
  return [];
};

// Function to validate and identify incorrect TRUE values
const validateRecord = (record: any, schema: any[]) => {
  console.log("Validating record:", record);
  console.log("Schema:", schema);
  for (const column of schema) {
    if (record[column.column_name]) {
      const value = record[column.column_name];
      // Check for incorrect TRUE values in non-boolean fields
      if (
        column.is_nullable &&
        column.data_type !== "boolean" &&
        value === true
      ) {
        console.info(
          `Incorrect TRUE value in column ${column.column_name} with data type ${column.data_type}`
        );
        record[column.column_name] = null;
      }
    }
  }

  return record;
};

//Take the last table name that follows the key word FROM.
//Works for reasonably complicated queries, may need modification to suit your case.
const extractLastTableName = (query: string): string | null => {
  const pattern = /\bfrom\s+([a-zA-Z_][a-zA-Z0-9_\.]*)/gi; // Global, case-insensitive regex
  const matches = [...query.matchAll(pattern)];
  return matches.length > 0 ? matches[matches.length - 1][1] : null; // Return the last match or null
};

const isSelectQuery = (query: string): boolean => {
  const pattern = /^\s*select\b/i; // Case-insensitive regex
  return pattern.test(query);
};