package store

import (
	"context"
	"encoding/json"
	"fmt"

	"aiaa-notification-service/internal/model"
)

func (s *Store) CreateChannel(ctx context.Context, ch *model.Channel) error {
	configJSON, err := json.Marshal(ch.Config)
	if err != nil {
		return fmt.Errorf("marshal channel config: %w", err)
	}
	query := `INSERT INTO channel (name, type, config, status) VALUES (?, ?, ?, ?)`
	result, err := s.DB.ExecContext(ctx, query, ch.Name, ch.Type, configJSON, ch.Status)
	if err != nil {
		return fmt.Errorf("create channel: %w", err)
	}
	id, _ := result.LastInsertId()
	ch.ID = int(id)
	return nil
}

func (s *Store) GetChannel(ctx context.Context, id int) (*model.Channel, error) {
	var ch model.Channel
	var configBytes []byte
	row := s.DB.QueryRowContext(ctx, `SELECT id, name, type, config, status, created_at, updated_at FROM channel WHERE id = ?`, id)
	if err := row.Scan(&ch.ID, &ch.Name, &ch.Type, &configBytes, &ch.Status, &ch.CreatedAt, &ch.UpdatedAt); err != nil {
		return nil, fmt.Errorf("get channel %d: %w", id, err)
	}
	raw := json.RawMessage(configBytes)
	ch.Config = &raw
	return &ch, nil
}

func (s *Store) GetChannelByName(ctx context.Context, name string) (*model.Channel, error) {
	var ch model.Channel
	var configBytes []byte
	row := s.DB.QueryRowContext(ctx, `SELECT id, name, type, config, status, created_at, updated_at FROM channel WHERE name = ?`, name)
	if err := row.Scan(&ch.ID, &ch.Name, &ch.Type, &configBytes, &ch.Status, &ch.CreatedAt, &ch.UpdatedAt); err != nil {
		return nil, fmt.Errorf("get channel by name %s: %w", name, err)
	}
	raw := json.RawMessage(configBytes)
	ch.Config = &raw
	return &ch, nil
}

func (s *Store) ListChannels(ctx context.Context) ([]model.Channel, error) {
	rows, err := s.DB.QueryContext(ctx, `SELECT id, name, type, config, status, created_at, updated_at FROM channel ORDER BY id`)
	if err != nil {
		return nil, fmt.Errorf("list channels: %w", err)
	}
	defer rows.Close()

	var channels []model.Channel
	for rows.Next() {
		var ch model.Channel
		var configBytes []byte
		if err := rows.Scan(&ch.ID, &ch.Name, &ch.Type, &configBytes, &ch.Status, &ch.CreatedAt, &ch.UpdatedAt); err != nil {
			return nil, fmt.Errorf("scan channel: %w", err)
		}
		raw := json.RawMessage(configBytes)
		ch.Config = &raw
		channels = append(channels, ch)
	}
	return channels, rows.Err()
}

func (s *Store) UpdateChannel(ctx context.Context, id int, ch *model.Channel) error {
	configJSON, err := json.Marshal(ch.Config)
	if err != nil {
		return fmt.Errorf("marshal channel config: %w", err)
	}
	query := `UPDATE channel SET name=?, type=?, config=?, status=? WHERE id=?`
	_, err = s.DB.ExecContext(ctx, query, ch.Name, ch.Type, configJSON, ch.Status, id)
	if err != nil {
		return fmt.Errorf("update channel %d: %w", id, err)
	}
	return nil
}

func (s *Store) DeleteChannel(ctx context.Context, id int) error {
	_, err := s.DB.ExecContext(ctx, `DELETE FROM channel WHERE id = ?`, id)
	if err != nil {
		return fmt.Errorf("delete channel %d: %w", id, err)
	}
	return nil
}