some init website stuff not working header does not get set propperly
This commit is contained in:
parent
e4810bf1be
commit
22cdce32bc
@ -34,6 +34,9 @@ func init() {
|
|||||||
func main() {
|
func main() {
|
||||||
defer internal.CloseLog()
|
defer internal.CloseLog()
|
||||||
|
|
||||||
|
http.HandleFunc("/", internal.LandingHandler)
|
||||||
|
http.HandleFunc("/login", internal.LoginHandler)
|
||||||
|
http.HandleFunc("/dash", internal.AuthClientMiddleware(internal.DashboardHandler))
|
||||||
http.HandleFunc("/upload", internal.AuthMiddleware(internal.UploadHandler))
|
http.HandleFunc("/upload", internal.AuthMiddleware(internal.UploadHandler))
|
||||||
http.HandleFunc("/health", internal.HealthHandler)
|
http.HandleFunc("/health", internal.HealthHandler)
|
||||||
|
|
||||||
|
|||||||
@ -9,6 +9,7 @@ import (
|
|||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
|
"text/template"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -30,6 +31,106 @@ func SetShit() {
|
|||||||
UploadDir = os.Getenv("CCTV_UploadDir")
|
UploadDir = os.Getenv("CCTV_UploadDir")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func LandingHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
|
content, err := os.ReadFile("views/index.html")
|
||||||
|
if err != nil {
|
||||||
|
http.Error(w, "Error reading file", http.StatusInternalServerError)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
fmt.Fprint(w, string(content))
|
||||||
|
}
|
||||||
|
|
||||||
|
func LoginHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
|
ip := getClientIP(r)
|
||||||
|
|
||||||
|
mu.Lock()
|
||||||
|
|
||||||
|
if banTime, banned := bannedIPs[ip]; banned {
|
||||||
|
fmt.Print("Banned IP: ", ip, " until ", banTime, "\n")
|
||||||
|
if time.Now().Before(banTime) {
|
||||||
|
bannedIPs[ip] = time.Now().Add(BanDuration)
|
||||||
|
http.Error(w, "You are banned", http.StatusForbidden)
|
||||||
|
mu.Unlock()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
delete(bannedIPs, ip)
|
||||||
|
}
|
||||||
|
|
||||||
|
mu.Unlock()
|
||||||
|
|
||||||
|
username := r.FormValue("username")
|
||||||
|
password := r.FormValue("password")
|
||||||
|
|
||||||
|
if username == "admin" && password == "admin" {
|
||||||
|
w.Header().Set("Authorization", "Bearer "+"hihi")
|
||||||
|
http.Redirect(w, r, "/dash", http.StatusSeeOther)
|
||||||
|
return
|
||||||
|
} else {
|
||||||
|
LogFailedAttempt(ip)
|
||||||
|
|
||||||
|
mu.Lock()
|
||||||
|
failedAttempts[ip]++
|
||||||
|
if failedAttempts[ip] >= MaxFailedAttempts {
|
||||||
|
bannedIPs[ip] = time.Now().Add(BanDuration)
|
||||||
|
delete(failedAttempts, ip)
|
||||||
|
|
||||||
|
LogBan(ip)
|
||||||
|
}
|
||||||
|
mu.Unlock()
|
||||||
|
|
||||||
|
http.Error(w, "Unauthorized", http.StatusUnauthorized)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func AuthClientMiddleware(next http.HandlerFunc) http.HandlerFunc {
|
||||||
|
return func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
authHeader := r.Header.Get("Authorization")
|
||||||
|
fmt.Print(authHeader)
|
||||||
|
if !strings.HasPrefix(authHeader, "Bearer ") || strings.TrimPrefix(authHeader, "Bearer ") != "hihi" {
|
||||||
|
http.Error(w, "Unauthorized", http.StatusUnauthorized)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
next(w, r)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func DashboardHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
|
// get the newest picture name
|
||||||
|
pictures, err := os.ReadDir(UploadDir)
|
||||||
|
if err != nil {
|
||||||
|
http.Error(w, "Error reading directory", http.StatusInternalServerError)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(pictures) == 0 {
|
||||||
|
http.Error(w, "No pictures found", http.StatusNotFound)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// get the newest picture
|
||||||
|
newestPicture := pictures[0]
|
||||||
|
for _, picture := range pictures {
|
||||||
|
pictureInfo, err := picture.Info()
|
||||||
|
if err != nil {
|
||||||
|
http.Error(w, "Error getting file info", http.StatusInternalServerError)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
newestPictureInfo, err := newestPicture.Info()
|
||||||
|
if err != nil {
|
||||||
|
http.Error(w, "Error getting file info", http.StatusInternalServerError)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if pictureInfo.ModTime().After(newestPictureInfo.ModTime()) {
|
||||||
|
newestPicture = picture
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
tmpl := template.Must(template.ParseFiles("views/dash.html"))
|
||||||
|
tmpl.Execute(w, newestPicture.Name())
|
||||||
|
}
|
||||||
|
|
||||||
func UploadHandler(w http.ResponseWriter, r *http.Request) {
|
func UploadHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
err := r.ParseMultipartForm(10 << 20)
|
err := r.ParseMultipartForm(10 << 20)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|||||||
10
views/dash.html
Normal file
10
views/dash.html
Normal file
@ -0,0 +1,10 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<html>
|
||||||
|
<head>
|
||||||
|
<title>CCTV</title>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<h1>Latest Picture</h1>
|
||||||
|
<img src="{{.picture}}" alt="Latest Picture">
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
17
views/index.html
Normal file
17
views/index.html
Normal file
@ -0,0 +1,17 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<html>
|
||||||
|
<head>
|
||||||
|
<title>CCTV</title>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<h1>CCTV</h1>
|
||||||
|
<h2>Login</h2>
|
||||||
|
<form action="/login" method="post">
|
||||||
|
<label for="username">Username:</label>
|
||||||
|
<input type="text" id="username" name="username" required>
|
||||||
|
<label for="password">Password:</label>
|
||||||
|
<input type="password" id="password" name="password" required>
|
||||||
|
<button type="submit">Login</button>
|
||||||
|
</form>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
Loading…
Reference in New Issue
Block a user