jan/CCTV
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

some init website stuff not working header does not get set propperly

Browse Source
This commit is contained in:
Jan 2025-02-24 23:40:36 +01:00
parent e4810bf1be
commit 22cdce32bc
4 changed files with 131 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
cmd/main.go
View File

@ -34,6 +34,9 @@ func init() {
func main() {
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("/health", internal.HealthHandler)

101
internal/server.go
View File

@ -9,6 +9,7 @@ import (
"path/filepath"
"strconv"
"strings"
"text/template"
"time"
)
@ -30,6 +31,106 @@ func SetShit() {
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) {
err := r.ParseMultipartForm(10 << 20)
if err != nil {

10
views/dash.html Normal file
View 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
View 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>