64 lines
1.4 KiB
Go
64 lines
1.4 KiB
Go
package service
|
|
|
|
import (
|
|
"apigo.cc/go/jsmod"
|
|
)
|
|
|
|
func init() {
|
|
jsmod.Register("service", map[string]any{
|
|
// 类型占位工厂 (用于 AI 发现类型结构，生成文档时隐藏)
|
|
"__exportRequest": jsExportRequest,
|
|
"__exportResponse": jsExportResponse,
|
|
"__exportWebSocket": jsExportWebSocket,
|
|
"__exportSession": jsExportSession,
|
|
"__exportFile": jsExportFile,
|
|
|
|
// 功能函数
|
|
"Upgrade": jsUpgrade,
|
|
})
|
|
}
|
|
|
|
func jsExportRequest() *Request { return &Request{} }
|
|
func jsExportResponse() *Response { return &Response{} }
|
|
func jsExportWebSocket() *WebSocketConn { return &WebSocketConn{} }
|
|
func jsExportSession() *Session { return &Session{} }
|
|
func jsExportFile() *jsUploadFile { return &jsUploadFile{} }
|
|
|
|
func jsUpgrade(response *Response, request *Request) (*WebSocketConn, error) {
|
|
conn, err := Upgrade(response, request)
|
|
if err != nil {
|
|
return nil, jsmod.MakeError(err)
|
|
}
|
|
return conn, nil
|
|
}
|
|
|
|
// jsUploadFile 包装 UploadFile 以隐藏敏感方法
|
|
type jsUploadFile struct {
|
|
f *UploadFile
|
|
}
|
|
|
|
func (j *jsUploadFile) Filename() string {
|
|
if j.f == nil {
|
|
return ""
|
|
}
|
|
return j.f.Filename
|
|
}
|
|
|
|
func (j *jsUploadFile) Size() int64 {
|
|
if j.f == nil {
|
|
return 0
|
|
}
|
|
return j.f.Size
|
|
}
|
|
|
|
func (j *jsUploadFile) Content() ([]byte, error) {
|
|
if j.f == nil {
|
|
return nil, nil
|
|
}
|
|
res, err := j.f.Content()
|
|
if err != nil {
|
|
return nil, jsmod.MakeError(err)
|
|
}
|
|
return res, nil
|
|
}
|