Function handler

src/pages/api/connection/db_schema.ts:10–35

(req: NextApiRequest, res: NextApiResponse)

// req body: { connection: Connection, db: string }
// It's mainly used to get the database list.
const handler = async (req: NextApiRequest, res: NextApiResponse) => {
  if (req.method !== "POST") {
    return res.status(405).json([]);
  }

  let connection = req.body.connection as Connection;
  if (connection.engineType === Engine.TiDB) {
    connection = changeTiDBConnectionToMySQL(connection);
  }

  const db = req.body.db as string;

  try {
    const connector = newConnector(connection);
    const schemaList: Schema[] = await connector.getTableSchema(db);

    res.status(200).json({
      data: schemaList,
    });
  } catch (error: any) {
    res.status(400).json({
      message: error.message || "Failed to get database schema.",
      code: error.code || "UNKNOWN",
    });
  }
};

export default handler;