aykhans/oh-my-url
1
0
Fork 0
mirror of https://github.com/aykhans/oh-my-url.git synced 2025-04-17 02:23:13 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

🔨 Refactor URL handling logic and remove unused variables

Browse Source
This commit is contained in:
Aykhan Shahsuvarov 2024-03-07 15:32:38 +04:00
parent f7c4239163
commit a947b98acf
5 changed files with 16 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
app/config/settings.go
View File

@ -37,6 +37,14 @@ func GetAppConfig() *AppConfig {
}
}
func GetForwardDomain() string {
return GetEnvOrPanic("FORWARD_DOMAIN")
}
func GetCreateDomain() string {
return GetEnvOrPanic("CREATE_DOMAIN")
}
func GetPostgresConfig() *PostgresConfig {
return &PostgresConfig{
USER: GetEnvOrPanic("POSTGRES_USER"),

6
app/http_handlers/base.go
View File

@ -8,13 +8,11 @@ import (
)
type HandlerCreate struct {
DB db.DB
ForwardDomain string
DB db.DB
}
type HandlerForward struct {
DB db.DB
CreateDomain string
DB db.DB
}
func FaviconHandler(w http.ResponseWriter, r *http.Request) {

3
app/http_handlers/url_create.go
View File

@ -2,6 +2,7 @@ package httpHandlers
import (
"github.com/aykhans/oh-my-url/app/utils"
"github.com/aykhans/oh-my-url/app/config"
"html/template"
"net/http"
netUrl "net/url"
@ -54,7 +55,7 @@ func (hl *HandlerCreate) UrlCreate(w http.ResponseWriter, r *http.Request) {
InternalServerError(w, err)
return
}
shortedURL, err := netUrl.JoinPath(hl.ForwardDomain, key)
shortedURL, err := netUrl.JoinPath(config.GetForwardDomain(), key)
if err != nil {
InternalServerError(w, err)
return

3
app/http_handlers/url_forward.go
View File

@ -1,6 +1,7 @@
package httpHandlers
import (
"github.com/aykhans/oh-my-url/app/config"
"net/http"
"strings"
)
@ -12,7 +13,7 @@ func (hl *HandlerForward) UrlForward(w http.ResponseWriter, r *http.Request) {
http.NotFound(w, r)
return
} else if segments[1] == "" {
http.Redirect(w, r, hl.CreateDomain, http.StatusMovedPermanently)
http.Redirect(w, r, config.GetCreateDomain(), http.StatusMovedPermanently)
return
}

4
app/main.go
View File

@ -12,13 +12,13 @@ func main() {
config := config.GetAppConfig()
dbCreate := db.GetDB()
dbCreate.Init()
handlerCreate := httpHandlers.HandlerCreate{DB: dbCreate, ForwardDomain: config.FORWARD_DOMAIN}
handlerCreate := httpHandlers.HandlerCreate{DB: dbCreate}
urlCreateMux := http.NewServeMux()
urlCreateMux.HandleFunc("/", handlerCreate.UrlCreate)
urlCreateMux.HandleFunc("/favicon.ico", httpHandlers.FaviconHandler)
dbRead := db.GetDB()
handlerForward := httpHandlers.HandlerForward{DB: dbRead, CreateDomain: config.CREATE_DOMAIN}
handlerForward := httpHandlers.HandlerForward{DB: dbRead}
urlReadMux := http.NewServeMux()
urlReadMux.HandleFunc("/", handlerForward.UrlForward)
urlReadMux.HandleFunc("/favicon.ico", httpHandlers.FaviconHandler)