errors.go

// Package knowledgesdk errors.go
package knowledgesdk

import (
	"errors"
	"fmt"
)

// 定义SDK错误码
const (
	// 知识库相关错误
	ErrKnowledgeNotExist     = "ERR_KNOWLEDGE_NOT_EXIST"
	ErrKnowledgeCreateFailed = "ERR_KNOWLEDGE_CREATE_FAILED"
	ErrKnowledgeUpdateFailed = "ERR_KNOWLEDGE_UPDATE_FAILED"
	ErrKnowledgeDeleteFailed = "ERR_KNOWLEDGE_DELETE_FAILED"

	// 文档相关错误
	ErrDocumentNotExist      = "ERR_DOCUMENT_NOT_EXIST"
	ErrDocumentCreateFailed  = "ERR_DOCUMENT_CREATE_FAILED"
	ErrDocumentUpdateFailed  = "ERR_DOCUMENT_UPDATE_FAILED"
	ErrDocumentDeleteFailed  = "ERR_DOCUMENT_DELETE_FAILED"
	ErrDocumentUploadFailed  = "ERR_DOCUMENT_UPLOAD_FAILED"
	ErrDocumentProcessFailed = "ERR_DOCUMENT_PROCESS_FAILED"

	// 向量和嵌入相关错误
	ErrEmbeddingFailed    = "ERR_EMBEDDING_FAILED"
	ErrVectorSearchFailed = "ERR_VECTOR_SEARCH_FAILED"
	ErrVectorInsertFailed = "ERR_VECTOR_INSERT_FAILED"
	ErrVectorUpdateFailed = "ERR_VECTOR_UPDATE_FAILED"

	// 数据库相关错误
	ErrDatabaseConnectFailed  = "ERR_DATABASE_CONNECT_FAILED"
	ErrDatabaseQueryFailed    = "ERR_DATABASE_QUERY_FAILED"
	ErrDatabaseTransactFailed = "ERR_DATABASE_TRANSACT_FAILED"

	// 配置和初始化相关错误
	ErrConfigInvalid = "ERR_CONFIG_INVALID"
	ErrSDKInitFailed = "ERR_SDK_INIT_FAILED"

	// API请求相关错误
	ErrInvalidRequest = "ERR_INVALID_REQUEST"
	ErrUnauthorized   = "ERR_UNAUTHORIZED"

	// 其他通用错误
	ErrInternalServer = "ERR_INTERNAL_SERVER"
	ErrNotImplemented = "ERR_NOT_IMPLEMENTED"
)

// 定义SDK错误类型
type SDKError struct {
	Code    string
	Message string
	Err     error
}

// Error 实现error接口
func (e *SDKError) Error() string {
	if e.Err != nil {
		return fmt.Sprintf("[%s] %s: %v", e.Code, e.Message, e.Err)
	}
	return fmt.Sprintf("[%s] %s", e.Code, e.Message)
}

// Unwrap 支持错误链
func (e *SDKError) Unwrap() error {
	return e.Err
}

// Is 实现errors.Is接口
func (e *SDKError) Is(target error) bool {
	t, ok := target.(*SDKError)
	if !ok {
		return false
	}
	return e.Code == t.Code
}

// NewError 创建新的SDK错误
func NewError(code string, message string, err error) *SDKError {
	return &SDKError{
		Code:    code,
		Message: message,
		Err:     err,
	}
}

// 定义一些预定义错误实例，便于直接使用
var (
	// 知识库不存在
	ErrKnowledgeBaseNotFound = NewError(
		ErrKnowledgeNotExist,
		"知识库不存在",
		errors.New("knowledge base not found"),
	)

	// 文档不存在
	ErrDocumentNotFound = NewError(
		ErrDocumentNotExist,
		"文档不存在",
		errors.New("document not found"),
	)

	// 内部服务器错误
	ErrInternalServerError = NewError(
		ErrInternalServer,
		"内部服务器错误",
		errors.New("internal server error"),
	)
)

// IsKnowledgeNotExistError 检查是否为知识库不存在错误
func IsKnowledgeNotExistError(err error) bool {
	var sdkErr *SDKError
	if errors.As(err, &sdkErr) {
		return sdkErr.Code == ErrKnowledgeNotExist
	}
	return false
}

// IsDocumentNotExistError 检查是否为文档不存在错误
func IsDocumentNotExistError(err error) bool {
	var sdkErr *SDKError
	if errors.As(err, &sdkErr) {
		return sdkErr.Code == ErrDocumentNotExist
	}
	return false
}

// GetErrorMessage 获取错误消息
func GetErrorMessage(code string) string {
	messages := map[string]string{
		ErrKnowledgeNotExist:      "知识库不存在",
		ErrKnowledgeCreateFailed:  "创建知识库失败",
		ErrKnowledgeUpdateFailed:  "更新知识库失败",
		ErrKnowledgeDeleteFailed:  "删除知识库失败",
		ErrDocumentNotExist:       "文档不存在",
		ErrDocumentCreateFailed:   "创建文档失败",
		ErrDocumentUpdateFailed:   "更新文档失败",
		ErrDocumentDeleteFailed:   "删除文档失败",
		ErrDocumentUploadFailed:   "上传文档失败",
		ErrDocumentProcessFailed:  "处理文档失败",
		ErrEmbeddingFailed:        "生成嵌入向量失败",
		ErrVectorSearchFailed:     "向量搜索失败",
		ErrVectorInsertFailed:     "向量插入失败",
		ErrVectorUpdateFailed:     "更新向量失败",
		ErrDatabaseConnectFailed:  "数据库连接失败",
		ErrDatabaseQueryFailed:    "数据库查询失败",
		ErrDatabaseTransactFailed: "数据库事务失败",
		ErrConfigInvalid:          "配置无效",
		ErrSDKInitFailed:          "SDK初始化失败",
		ErrInvalidRequest:         "无效请求",
		ErrUnauthorized:           "未授权",
		ErrInternalServer:         "内部服务器错误",
		ErrNotImplemented:         "功能未实现",
	}

	if msg, ok := messages[code]; ok {
		return msg
	}
	return "未知错误"
}