Lbenedar/greenlight
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ch18.3

Browse Source
This commit is contained in:
lbenedar
2026-03-24 14:20:29 +03:00
parent 4ff2f16110
commit 974e7f151f
4 changed files with 74 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
cmd/api/main.go
View File

@@ -5,6 +5,7 @@ import (
"database/sql" "database/sql"
"flag" "flag"
"os" "os"
"strings"
"sync" "sync"
"time" "time"
@@ -45,6 +46,9 @@ type config struct {
password string password string
sender string sender string
} }
cors struct {
trustedOrigins []string
}
} }
func openDB(cfg config) (*sql.DB, error) { func openDB(cfg config) (*sql.DB, error) {
@@ -93,6 +97,11 @@ func main() {
flag.StringVar(&cfg.smtp.password, "smtp-password", "d8672aa2264bb5", "SMTP password") flag.StringVar(&cfg.smtp.password, "smtp-password", "d8672aa2264bb5", "SMTP password")
flag.StringVar(&cfg.smtp.sender, "smtp-sender", "Greenlight <no-reply@greenlight.net", "SMTP sender") flag.StringVar(&cfg.smtp.sender, "smtp-sender", "Greenlight <no-reply@greenlight.net", "SMTP sender")
flag.Func("cors-trusted-origins", "Trusted CORS origins (space separated)", func(s string) error {
cfg.cors.trustedOrigins = strings.Fields(s)
return nil
})
flag.Parse() flag.Parse()
logger := jsonlog.New(os.Stdout, jsonlog.LevelInfo) logger := jsonlog.New(os.Stdout, jsonlog.LevelInfo)

19
cmd/api/middleware.go
View File

@@ -170,3 +170,22 @@ func (app *application) requirePermissions(code string, next http.HandlerFunc) h
} }
return app.requireActivatedUser(fn) return app.requireActivatedUser(fn)
} }
func (app *application) enableCORS(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Vary", "Origin")
origin := r.Header.Get("Origin")
if origin != "" {
for i := range app.config.cors.trustedOrigins {
if origin == app.config.cors.trustedOrigins[i] {
w.Header().Set("Access-Control-Allow-Origin", origin)
break
}
}
}
next.ServeHTTP(w, r)
})
}

2
cmd/api/routes.go
View File

@@ -25,5 +25,5 @@ func (app *application) routes() http.Handler {
router.HandlerFunc(http.MethodPost, "/v1/tokens/authentication", app.createAuthenticationTokenHandler) router.HandlerFunc(http.MethodPost, "/v1/tokens/authentication", app.createAuthenticationTokenHandler)
return app.recoverPanic(app.rateLimit(app.authenticate(router))) return app.recoverPanic(app.enableCORS(app.rateLimit(app.authenticate(router))))
} }

45
cmd/examples/cors/simple/main.go Normal file
View File

@@ -0,0 +1,45 @@
package main
import (
"flag"
"log"
"net/http"
)
const html = `
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
</head>
<body>
<h1>Simple CORS</h1>
<div id="output"></div>
<script>
document.addEventListener('DOMContentLoaded', function() {
fetch("http://localhost:4000/v1/healthcheck").then(
function (response) {
response.text().then(function (text) {
document.getElementById("output").innerHTML = text;
});
},
function(err) {
document.getElementById("output").innerHTML = err;
}
);
});
</script>
</body>
</html>`
func main() {
addr := flag.String("addr", ":9000", "Server address")
flag.Parse()
log.Printf("starting server on %s", *addr)
err := http.ListenAndServe(*addr, http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Write([]byte(html))
}))
log.Fatal(err)
}