140 lines
2.9 KiB
Go
140 lines
2.9 KiB
Go
package service
|
||
|
||
import (
|
||
"apigo.cc/go/cast"
|
||
"apigo.cc/go/standard"
|
||
"io"
|
||
"mime/multipart"
|
||
"net"
|
||
"net/http"
|
||
"net/textproto"
|
||
"net/url"
|
||
"os"
|
||
"path/filepath"
|
||
)
|
||
|
||
// UploadFile 上传文件结构
|
||
type UploadFile struct {
|
||
fileHeader *multipart.FileHeader
|
||
Filename string
|
||
Header textproto.MIMEHeader
|
||
Size int64
|
||
}
|
||
|
||
// Open 打开上传文件
|
||
func (f *UploadFile) Open() (multipart.File, error) {
|
||
return f.fileHeader.Open()
|
||
}
|
||
|
||
// Save 保存上传文件到本地
|
||
func (f *UploadFile) Save(filename string) error {
|
||
dir := filepath.Dir(filename)
|
||
if _, err := os.Stat(dir); os.IsNotExist(err) {
|
||
_ = os.MkdirAll(dir, 0755)
|
||
}
|
||
|
||
dst, err := os.OpenFile(filename, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0644)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
defer dst.Close()
|
||
|
||
src, err := f.fileHeader.Open()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
defer src.Close()
|
||
|
||
_, err = io.Copy(dst, src)
|
||
return err
|
||
}
|
||
|
||
// Content 获取上传文件内容
|
||
func (f *UploadFile) Content() ([]byte, error) {
|
||
src, err := f.fileHeader.Open()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
defer src.Close()
|
||
return io.ReadAll(src)
|
||
}
|
||
|
||
// Request 封装 http.Request
|
||
type Request struct {
|
||
*http.Request
|
||
contextValues map[string]any
|
||
Id string
|
||
}
|
||
|
||
// NewRequest 创建 Request 包装
|
||
func NewRequest(httpRequest *http.Request) *Request {
|
||
return &Request{
|
||
Request: httpRequest,
|
||
contextValues: make(map[string]any),
|
||
}
|
||
}
|
||
|
||
// ResetPath 重写请求路径
|
||
func (r *Request) ResetPath(path string) {
|
||
r.RequestURI = path
|
||
if u, err := url.Parse(path); err == nil {
|
||
r.URL = u
|
||
}
|
||
}
|
||
|
||
// Set 设置请求上下文变量
|
||
func (r *Request) Set(key string, value any) {
|
||
r.contextValues[key] = value
|
||
}
|
||
|
||
// Get 获取请求上下文变量
|
||
func (r *Request) Get(key string) any {
|
||
return r.contextValues[key]
|
||
}
|
||
|
||
// MakeUrl 根据当前请求构建完整 URL
|
||
func (r *Request) MakeUrl(path string) string {
|
||
scheme := r.Header.Get(standard.DiscoverHeaderScheme)
|
||
if scheme == "" {
|
||
scheme = "http"
|
||
}
|
||
host := r.Header.Get(standard.DiscoverHeaderHost)
|
||
if host == "" {
|
||
host = r.Host
|
||
}
|
||
return scheme + "://" + host + path
|
||
}
|
||
|
||
// GetSessionId 获取会话 ID
|
||
func (r *Request) GetSessionId() string {
|
||
sessionId := r.Header.Get(Config.Listen) // Wait, this should be usedSessionIdKey
|
||
// TODO: Fix dependency on global usedSessionIdKey
|
||
return sessionId
|
||
}
|
||
|
||
// SetUserId 设置用户 ID（传递给下游）
|
||
func (r *Request) SetUserId(userId string) {
|
||
r.Header.Set(standard.DiscoverHeaderUserId, userId)
|
||
}
|
||
|
||
// GetRealIp 获取真实 IP
|
||
func (r *Request) GetRealIp() string {
|
||
ip := r.Header.Get(standard.DiscoverHeaderClientIp)
|
||
if ip == "" {
|
||
ip = r.Header.Get(standard.DiscoverHeaderForwardedFor)
|
||
}
|
||
if ip == "" {
|
||
host, _, err := net.SplitHostPort(r.RemoteAddr)
|
||
if err == nil {
|
||
return host
|
||
}
|
||
return r.RemoteAddr
|
||
}
|
||
return ip
|
||
}
|
||
|
||
// GetLowerName (Aliased from cast)
|
||
func GetLowerName(s string) string {
|
||
return cast.GetLowerName(s)
|
||
}
|