#!/bin/bash
set -e

source ~/cloud-env.sh

mysql \
   --comments -u "${SERVERLESS_CLUSTER_USERNAME}" -p"${SERVERLESS_CLUSTER_PASSWORD}" \
   -h ${SERVERLESS_CLUSTER_HOST} -P ${SERVERLESS_CLUSTER_PORT} \
   --ssl-mode=VERIFY_IDENTITY --ssl-ca=/etc/ssl/certs/ca-certificates.crt\
   -D ${SERVERLESS_CLUSTER_DATABASE_NAME} <<EOF
USE ${SERVERLESS_CLUSTER_DATABASE_NAME};
DROP TABLE IF EXISTS books_embedding;
CREATE TABLE books_embedding (
    id int AUTO_INCREMENT,
    title varchar(100) COLLATE utf8mb4_unicode_ci NOT NULL,
    type enum(
        'Magazine',
        'Novel',
        'Life',
        'Arts',
        'Comics',
        'Education & Reference',
        'Humanities & Social Sciences',
        'Science & Technology',
        'Kids',
        'Sports'
    ) COLLATE utf8mb4_unicode_ci NOT NULL,
    published_at datetime NOT NULL,
    stock int DEFAULT '0',
    price decimal(15, 2) DEFAULT '0.0',
    title_vector VECTOR(1024) GENERATED ALWAYS AS (
        EMBED_TEXT("tidbcloud_free/amazon/titan-embed-text-v2", title)
    ) STORED,
    PRIMARY KEY (id)
);
INSERT INTO books_embedding (title, type, published_at, stock, price)
SELECT title, type, published_at, stock, price FROM books WHERE id BETWEEN 1 AND 100;
EOF