gered/ollama
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

refactor default allow origins

Browse source 
this should be less error prone
This commit is contained in:
Michael Yang 2023-09-21 09:42:16 -07:00
parent 765770efdb
commit 1fabba474b
1 changed files with 17 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
server/routes.go
View file

@ -499,23 +499,25 @@ func CopyModelHandler(c *gin.Context) {
} }
} }
func Serve(ln net.Listener, origins []string) error { var defaultAllowOrigins = []string{
"localhost",
"127.0.0.1",
"0.0.0.0",
}
func Serve(ln net.Listener, allowOrigins []string) error {
config := cors.DefaultConfig() config := cors.DefaultConfig()
config.AllowWildcard = true config.AllowWildcard = true
config.AllowOrigins = append(origins, []string{
"http://localhost", config.AllowOrigins = allowOrigins
"http://localhost:*", for _, allowOrigin := range defaultAllowOrigins {
"https://localhost", config.AllowOrigins = append(config.AllowOrigins,
"https://localhost:*", fmt.Sprintf("http://%s", allowOrigin),
"http://127.0.0.1", fmt.Sprintf("https://%s", allowOrigin),
"http://127.0.0.1:*", fmt.Sprintf("http://%s:*", allowOrigin),
"https://127.0.0.1", fmt.Sprintf("https://%s:*", allowOrigin),
"https://127.0.0.1:*", )
"http://0.0.0.0", }
"http://0.0.0.0:*",
"https://0.0.0.0",
"https://0.0.0.0:*",
}...)
r := gin.Default() r := gin.Default()
r.Use(cors.New(config)) r.Use(cors.New(config))