fcrun/x
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Wildcard domain support

Browse Source
This commit is contained in:
Ian Gulliver
2024-12-02 22:54:22 -08:00
parent 695e947e4c
commit 9648518a69
1 changed files with 25 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
main.go
View File

@@ -72,7 +72,19 @@ func (sl *ShortLinks) ServeHTTP(w http.ResponseWriter, r *http.Request) {
func (sl *ShortLinks) serveRoot(w http.ResponseWriter, r *http.Request) {
log.Printf("%s %s %s %s %s", r.RemoteAddr, r.Method, r.Host, sl.getDomain(r.Host), r.URL)
sl.serveRootWithPath(w, r, "")
parts := strings.SplitN(r.Host, ".", 2)
if len(parts) != 2 {
sl.serveRootWithPath(w, r, "")
return
}
long, err := sl.getLong(parts[0], sl.getDomain(parts[1]))
if err != nil {
sl.serveRootWithPath(w, r, "")
return
}
http.Redirect(w, r, long, http.StatusTemporaryRedirect)
}
func (sl *ShortLinks) serveRootWithPath(w http.ResponseWriter, r *http.Request, path string) {
@@ -81,6 +93,7 @@ func (sl *ShortLinks) serveRootWithPath(w http.ResponseWriter, r *http.Request,
sendError(w, http.StatusBadRequest, "Parse form: %s", err)
return
}
log.Printf("%s %s %s %s %s", r.RemoteAddr, r.Method, r.Host, sl.getDomain(r.Host), r.URL)
if !sl.isWritable(r.Host) {
@@ -104,9 +117,7 @@ func (sl *ShortLinks) serveShort(w http.ResponseWriter, r *http.Request) {
short := r.PathValue("short")
row := sl.db.QueryRow(`SELECT long FROM links WHERE short = $1 AND domain = $2`, short, sl.getDomain(r.Host))
var long string
err := row.Scan(&long)
long, err := sl.getLong(short, sl.getDomain(r.Host))
if err != nil {
sl.serveRootWithPath(w, r, short)
return
@@ -237,6 +248,16 @@ func (sl *ShortLinks) isWritable(host string) bool {
return sl.writableDomains[host]
}
func (sl *ShortLinks) getLong(short, domain string) (string, error) {
var long string
err := sl.db.QueryRow("SELECT long FROM links WHERE short = $1 AND domain = $2", short, domain).Scan(&long)
if err != nil {
return "", err
}
return long, nil
}
func main() {
port := os.Getenv("PORT")
if port == "" {