package comm

import (
	"context"
	cryptorand "crypto/rand"
	"yunyan/lego/base"
	"yunyan/lego/core"
	"yunyan/pb"
	"fmt"
	"reflect"
)

type IService interface {
	base.IRPCXService
	GetUserSession(ctx context.Context, mate map[string]string) (session IUserSession)
	PutUserSession(session IUserSession)
}

// ApiInterceptor API 请求拦截器函数
// 在 handler 执行前调用，返回 *pb.ErrorData 表示拦截（不继续执行handler），返回 nil 表示放行
type ApiInterceptor func(apiName string, session IUserSession, msg interface{}) *pb.ErrorData

// ApiAuditFunc API 请求审计回调函数
// apiName: 接口名称, session: 用户会话, reqBody: 请求体, respBody: 响应体, code: 状态码, costMs: 耗时(毫秒)
type ApiAuditFunc func(apiName string, session IUserSession, reqBody []byte, respBody []byte, code int32, costMs int64)

type ISC_HttpRouteComp interface {
	core.IServiceComp
	Rpc_GatewayHttpRoute(ctx context.Context, args *pb.Rpc_GatewayHttpRouteReq, reply *pb.Rpc_GatewayHttpRouteResp) error
	//注册路由
	RegisterRoute(methodName string, isEncrypt bool, comp reflect.Value, msg reflect.Type, handle reflect.Method)
	AddInterceptor(interceptor ApiInterceptor) // 添加 API 请求拦截器
	AddApiAuditHook(hook ApiAuditFunc)         // 添加 API 审计回调钩子
}

// 用户会话
type IUserSession interface {
	context.Context
	SetSession(ctx context.Context, mate map[string]string)
	GetMateToString(key string) string
	GetMateToInt64(key string) int64
	GetMateToUInt64(key string) uint64
	GetMateToInt32(key string) int32
	GetMateToFloat64(key string) float64
	SetMateForFloat64(key string, value float64)
	SetMateForInt64(key string, value int64)
	GetMateToBool(key string) bool
	GetIP() string
	GetUserId() string
	Reset()
	SetMate(name string, value string)
	SetMates(meta map[string]string)
	GetMate(name string) (value string, ok bool)
	SetCache(name string, value interface{})
	GetCache(name string) (value interface{}, ok bool)
	Clone() (session IUserSession) //克隆
	GetMetas() (meta map[string]string)
}

type HttpResult struct {
	// 错误代码
	// @Description 响应结果的错误码 0 表示成功
	// @example 0
	Code pb.ErrorCode `json:"code" example:"0" description:"请求回应Code 0表示成功 非0 请求异常"`
	// 消息说明
	// @Description 响应结果的消息 请求返回的描述信息
	// @example "Success"
	Message string `json:"msg" example:"Success" description:"code 对应的描述信息"`
	// 数据部分，可能是多种类型
	// @Description 响应的数据，类型可变
	// @oneOf User Product Order
	// @example {"id": 1, "name": "John Doe"}
	Data interface{} `json:"data" description:"返回的数据对象"`
}

// ai 工具结构
type IAITool struct {
}

// GeneratePublicCode 生成厂家公码（6 位短码，字符集去除易混的 0/O/I/1）
// 用于厂家公码场景：人眼可读、可手抄/可打印
func GeneratePublicCode() (string, error) {
	const charset = "ABCDEFGHJKLMNPQRSTUVWXYZ23456789" // 32 字符
	const length = 6
	buf := make([]byte, length)
	rb := make([]byte, length)
	if _, err := cryptorand.Read(rb); err != nil {
		return "", fmt.Errorf("生成公码失败: %v", err)
	}
	for i := 0; i < length; i++ {
		buf[i] = charset[int(rb[i])%len(charset)]
	}
	return string(buf), nil
}