# Copyright 2025 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import os

from google.cloud.bigquery import Client as BigQueryClient
from google.cloud.storage import Client as GCSClient

from deploy_agent_engine import setup_remote_agent
from scripts.big_query_setup import create_dataset
from scripts.big_query_setup import create_table
from scripts.big_query_setup import insert_intent
from scripts.gcs_setup import create_bucket
from src.model.chats import Chat
from src.model.embedding import Embedding
from src.model.intent import Intent
from src.repository.big_query import CHATS_TABLE
from src.repository.big_query import EMBEDDINGS_TABLE
from src.service.intent import INTENTS_TABLE

try:
    print("Setting up GCS... \n")
    project_id = os.getenv("_PROJECT_ID")
    location = os.getenv("_REGION")
    storage_client = GCSClient()

    bucket = create_bucket(f"quick-bot-{project_id}-travel-concierge-bucket", location, storage_client)

    print("Setting up Remote Agent... \n")
    remote_agent_resource_id = setup_remote_agent(bucket)

    DEFAULT_INTENTS = [
        Intent(
            name="Travel concierge",
            ai_model="gemini-2.0-flash",
            ai_temperature=1,
            description="",
            prompt="",
            questions=[],
            status="5",
            remote_agent_resource_id=remote_agent_resource_id,
        ),
    ]

    print(f"Successfully setted Agent. Remote agent resource ID: {remote_agent_resource_id} \n")

    print("Setting up BigQuery... \n")
    bigquery_client = BigQueryClient()
    BIG_QUERY_DATASET = os.getenv("BIG_QUERY_DATASET") if os.getenv("BIG_QUERY_DATASET") else os.getenv("_CHATBOT_NAME", "quickbot").lower().replace(" ", "_").replace("-","_")
    print(f"Setting up BigQuery Dataset {BIG_QUERY_DATASET}... \n")
    create_dataset(BIG_QUERY_DATASET, bigquery_client)

    print(f"Setting up BigQuery Table {CHATS_TABLE}... \n")
    create_table(
        BIG_QUERY_DATASET, CHATS_TABLE, Chat.__schema__(), project_id, bigquery_client
    )
    print(f"Setting up BigQuery Table {EMBEDDINGS_TABLE}... \n")
    create_table(
        BIG_QUERY_DATASET,
        EMBEDDINGS_TABLE,
        Embedding.__schema__(),
        project_id,
        bigquery_client,
    )
    print(f"Setting up BigQuery Table {INTENTS_TABLE}... \n")
    create_table(
        BIG_QUERY_DATASET, INTENTS_TABLE, Intent.__schema__(), project_id, bigquery_client
    )

    for intent in DEFAULT_INTENTS:
        try:
            insert_intent(
                BIG_QUERY_DATASET, INTENTS_TABLE, intent.to_insert_string(), bigquery_client
            )
            print(f"Intent 'Travel concierge' created successfully \n")
        except Exception as e:
            print(e)
            print(f"An error occurred during the setup process: {e}")
            print(f"ERROR: A command in the backend setup block failed with code {e.returncode}.")
            if e.stdout: print(f"Stdout from failed command:\n{e.stdout}")
            else: print("Stdout from failed command: <empty>")
            if e.stderr: print(f"Stderr from failed command:\n{e.stderr}")
            else: print("Stderr from failed command: <empty>")

    print("\nSuccess!\n")
except Exception as e:
    print(f"A critical error occurred during the setup process: {e}")
    import traceback
    print("Detailed traceback:")
    print(traceback.format_exc())
    # If running in Docker build, exiting with non-zero will fail the build
    import sys
    sys.exit(1)