package model

import (
	"context"
	"errors"
	"fmt"
	"github.com/zeromicro/go-zero/core/stores/sqlx"
)

var _ MgServersModel = (*customMgServersModel)(nil)

type (
	// MgServersModel is an interface to be customized, add more methods here,
	// and implement the added methods in customMgServersModel.
	MgServersModel interface {
		mgServersModel
		withSession(session sqlx.Session) MgServersModel
		GetGameServer(ctx context.Context, agentId, serverId int64) (*GameServer, error)
	}

	customMgServersModel struct {
		*defaultMgServersModel
	}

	GameServer struct {
		AgentID        int    `db:"agent_id"`         // 代理ID
		ServerID       int    `db:"server_id"`        // 服务器ID
		WebPort        int    `db:"web_port"`         // web访问端口
		ServerIP       string `db:"server_ip"`        // 内网IP
		PublicServerIP string `db:"public_server_ip"` // 公网IP
	}
)

func (m *customMgServersModel) GetGameServer(ctx context.Context, agentId, serverId int64) (*GameServer, error) {
	query := fmt.Sprintf("SELECT g.agent_id, g.server_id, g.web_port, s.ip AS server_ip, s.public_ip AS public_server_ip FROM ms_game_server AS g LEFT JOIN mg_servers AS s ON s.id = g.server_server_id WHERE g.server_id = ? AND g.agent_id = ?")
	var result GameServer
	err := m.conn.QueryRowCtx(ctx, &result, query, serverId, agentId)
	switch {
	case err == nil:
		return &result, nil
	case errors.Is(err, sqlx.ErrNotFound):
		return nil, ErrNotFound
	default:
		return nil, err
	}
}

// NewMgServersModel returns a model for the database table.
func NewMgServersModel(conn sqlx.SqlConn) MgServersModel {
	return &customMgServersModel{
		defaultMgServersModel: newMgServersModel(conn),
	}
}

func (m *customMgServersModel) withSession(session sqlx.Session) MgServersModel {
	return NewMgServersModel(sqlx.NewSqlConnFromSession(session))
}