GithubProxy/coai
1
0
Fork 0
mirror of https://github.com/coaidev/coai.git synced 2025-05-19 21:10:18 +09:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: fix app protocol cors issue (#158)

Browse Source
This commit is contained in:
Minghan Zhang 2024-05-23 15:53:43 +08:00 committed by Zhang Minghan
parent 907362a9aa
commit 0bbdc68e08
1 changed files with 13 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
globals/variables.go
View File

@ -1,10 +1,9 @@
package globals package globals
import ( import (
"github.com/gin-gonic/gin"
"net/url" "net/url"
"strings" "strings"
"github.com/gin-gonic/gin"
) )
const ChatMaxThread = 5 const ChatMaxThread = 5
@ -23,43 +22,26 @@ var AcceptImageStore bool
var CloseRegistration bool var CloseRegistration bool
var CloseRelay bool var CloseRelay bool
var EpayBusinessId string
var EpayBusinessKey string
var EpayEndpoint string
var EpayEnabled bool
var EpayMethods []string
var SoftAuthPass byte
var SoftDomain []byte
var SoftName []byte
func OriginIsAllowed(uri string) bool { func OriginIsAllowed(uri string) bool {
instance, err := url.Parse(uri) if len(AllowedOrigins) == 0 {
if err != nil { // if allowed origins is empty, allow all origins
return true
}
instance, _ := url.Parse(uri)
if instance == nil {
return false return false
} }
if instance.Scheme == "file" { if instance.Hostname() == "localhost" || instance.Scheme == "file" {
return true return true
} }
if instance.Hostname() == "localhost" || strings.HasPrefix(instance.Hostname(), "localhost") || if strings.HasPrefix(instance.Host, "www.") {
instance.Hostname() == "127.0.0.1" || strings.HasPrefix(instance.Hostname(), "127.0.0.1") || instance.Host = instance.Host[4:]
strings.HasPrefix(instance.Hostname(), "192.168.") || strings.HasPrefix(instance.Hostname(), "10.") {
return true
} }
// get top level domain (example: sub.chatnio.net -> chatnio.net, chatnio.net -> chatnio.net) return in(instance.Host, AllowedOrigins)
// if the domain is in the allowed origins, return true
allow := string(SoftDomain)
domain := instance.Hostname()
if strings.HasSuffix(domain, allow) {
return true
}
return false
} }
func OriginIsOpen(c *gin.Context) bool { func OriginIsOpen(c *gin.Context) bool {
@ -145,7 +127,7 @@ var OpenAIDalleModels = []string{
var VisionModels = []string{ var VisionModels = []string{
GPT4VisionPreview, GPT41106VisionPreview, GPT4Turbo, GPT4Turbo20240409, GPT4O, GPT4O20240513, // openai GPT4VisionPreview, GPT41106VisionPreview, GPT4Turbo, GPT4Turbo20240409, GPT4O, GPT4O20240513, // openai
GeminiProVision, Gemini15ProLatest, Gemini15FlashLatest, // gemini GeminiProVision, Gemini15ProLatest, Gemini15FlashLatest, // gemini
Claude3, // anthropic Claude3, // anthropic
ZhiPuChatGLM4Vision, // chatglm ZhiPuChatGLM4Vision, // chatglm
} }
@ -171,4 +153,3 @@ func IsOpenAIDalleModel(model string) bool {
func IsVisionModel(model string) bool { func IsVisionModel(model string) bool {
return in(model, VisionModels) && !in(model, VisionSkipModels) return in(model, VisionModels) && !in(model, VisionSkipModels)
} }