fix_postgres_auth.sh

#!/bin/bash

# Скрипт для исправления аутентификации PostgreSQL и загрузки схемы

set -e

GREEN='\033[0;32m'
YELLOW='\033[0;33m'
RED='\033[0;31m'
NC='\033[0m'

echo "==========================================="
echo "   Настройка PostgreSQL для Biotact       "
echo "==========================================="
echo ""

# Проверяем что мы в правильной директории
if [ ! -f "infrastructure/postgres/init/002_create_schema_fixed.sql" ]; then
    echo -e "${RED}Ошибка: Не найден файл схемы!${NC}"
    echo "Запустите скрипт из директории /home/temrjan/project/biotact-production"
    exit 1
fi

echo -e "${YELLOW}Шаг 1: Сброс пароля для biotact_admin...${NC}"

# Создаем временный SQL файл для выполнения команд
cat > /tmp/fix_biotact_auth.sql << EOF
-- Сброс пароля для пользователя biotact_admin
ALTER USER biotact_admin WITH PASSWORD 'biotact_password';

-- Проверяем наличие базы данных
SELECT datname FROM pg_database WHERE datname = 'biotact_production';

-- Даём все права пользователю
GRANT ALL PRIVILEGES ON DATABASE biotact_production TO biotact_admin;

-- Проверяем пользователя
SELECT usename, usesysid FROM pg_user WHERE usename = 'biotact_admin';
EOF

# Выполняем SQL команды от имени postgres
sudo -u postgres psql < /tmp/fix_biotact_auth.sql

if [ $? -eq 0 ]; then
    echo -e "${GREEN}✓ Пароль успешно сброшен${NC}"
else
    echo -e "${RED}✗ Ошибка при сбросе пароля${NC}"
    exit 1
fi

echo ""
echo -e "${YELLOW}Шаг 2: Проверка подключения...${NC}"

# Проверяем подключение
PGPASSWORD=biotact_password psql -h 127.0.0.1 -U biotact_admin -d biotact_production -c "SELECT version();" > /dev/null 2>&1

if [ $? -eq 0 ]; then
    echo -e "${GREEN}✓ Подключение успешно${NC}"
else
    echo -e "${RED}✗ Не удалось подключиться${NC}"
    echo ""
    echo "Попробуем альтернативный способ..."

    # Альтернативный способ - создаём пользователя заново
    cat > /tmp/recreate_user.sql << EOF
-- Удаляем пользователя если существует
DROP USER IF EXISTS biotact_admin;

-- Создаём заново
CREATE USER biotact_admin WITH PASSWORD 'biotact_password';

-- Даём права
ALTER USER biotact_admin CREATEDB;
GRANT ALL PRIVILEGES ON DATABASE biotact_production TO biotact_admin;
ALTER DATABASE biotact_production OWNER TO biotact_admin;
EOF

    sudo -u postgres psql < /tmp/recreate_user.sql

    # Проверяем снова
    PGPASSWORD=biotact_password psql -h 127.0.0.1 -U biotact_admin -d biotact_production -c "SELECT version();" > /dev/null 2>&1

    if [ $? -ne 0 ]; then
        echo -e "${RED}✗ Всё ещё не удаётся подключиться${NC}"
        echo "Возможно, проблема в pg_hba.conf. Проверьте настройки аутентификации."
        exit 1
    fi
fi

echo ""
echo -e "${YELLOW}Шаг 3: Загрузка схемы базы данных...${NC}"

# Сначала включаем расширения
cat > /tmp/enable_extensions.sql << EOF
-- Включаем необходимые расширения
CREATE EXTENSION IF NOT EXISTS "uuid-ossp";
CREATE EXTENSION IF NOT EXISTS "pgcrypto";

-- Создаём ENUM типы если не существуют
DO \$\$
BEGIN
    IF NOT EXISTS (SELECT 1 FROM pg_type WHERE typname = 'user_role') THEN
        CREATE TYPE user_role AS ENUM ('admin', 'manager', 'user', 'viewer');
    END IF;

    IF NOT EXISTS (SELECT 1 FROM pg_type WHERE typname = 'severity_level') THEN
        CREATE TYPE severity_level AS ENUM ('debug', 'info', 'warning', 'error', 'critical');
    END IF;
END
\$\$;
EOF

# Загружаем расширения
PGPASSWORD=biotact_password psql -h 127.0.0.1 -U biotact_admin -d biotact_production < /tmp/enable_extensions.sql

if [ $? -eq 0 ]; then
    echo -e "${GREEN}✓ Расширения включены${NC}"
else
    echo -e "${YELLOW}⚠ Предупреждение: Некоторые расширения уже включены${NC}"
fi

# Теперь загружаем основную схему
echo "Загружаем основную схему..."
PGPASSWORD=biotact_password psql -h 127.0.0.1 -U biotact_admin -d biotact_production -f infrastructure/postgres/init/002_create_schema_fixed.sql

if [ $? -eq 0 ]; then
    echo -e "${GREEN}✓ Схема успешно загружена${NC}"
else
    echo -e "${RED}✗ Ошибка при загрузке схемы${NC}"
    echo "Проверьте файл infrastructure/postgres/init/002_create_schema_fixed.sql на ошибки"
    exit 1
fi

echo ""
echo -e "${YELLOW}Шаг 4: Проверка таблиц...${NC}"

# Проверяем количество созданных таблиц
TABLE_COUNT=$(PGPASSWORD=biotact_password psql -h 127.0.0.1 -U biotact_admin -d biotact_production -t -c "SELECT COUNT(*) FROM information_schema.tables WHERE table_schema = 'public' AND table_type = 'BASE TABLE';")

echo -e "Создано таблиц: ${GREEN}$TABLE_COUNT${NC}"

# Показываем список таблиц
echo ""
echo "Список таблиц:"
PGPASSWORD=biotact_password psql -h 127.0.0.1 -U biotact_admin -d biotact_production -c "\dt"

# Очищаем временные файлы
rm -f /tmp/fix_biotact_auth.sql
rm -f /tmp/recreate_user.sql
rm -f /tmp/enable_extensions.sql

echo ""
echo "==========================================="
echo -e "${GREEN}✅ Настройка PostgreSQL завершена!${NC}"
echo "==========================================="
echo ""
echo "База данных готова к использованию:"
echo "  Host: 127.0.0.1"
echo "  Port: 5432"
echo "  Database: biotact_production"
echo "  User: biotact_admin"
echo "  Password: biotact_password"
echo ""
echo "Для подключения используйте:"
echo "  PGPASSWORD=biotact_password psql -h 127.0.0.1 -U biotact_admin -d biotact_production"