mirror of
https://github.com/NotAShelf/go-cdn.git
synced 2024-11-22 16:00:43 +00:00
feat: make the CDN more verbose
This commit is contained in:
parent
399892a5e6
commit
390606d7b3
1 changed files with 121 additions and 0 deletions
121
main.go
Normal file
121
main.go
Normal file
|
@ -0,0 +1,121 @@
|
|||
// main.go
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
"io"
|
||||
"path/filepath"
|
||||
)
|
||||
|
||||
const (
|
||||
uploadPath = "uploads/" // Directory to store uploaded files
|
||||
username = "admin" // Username for authentication
|
||||
password = "password" // Password for authentication
|
||||
)
|
||||
|
||||
func main() {
|
||||
http.HandleFunc("/", serveCDN)
|
||||
http.HandleFunc("/upload", handleUpload)
|
||||
|
||||
log.Println("Starting CDN server on port 8080...")
|
||||
log.Fatal(http.ListenAndServe(":8080", nil))
|
||||
}
|
||||
|
||||
func serveCDN(w http.ResponseWriter, r *http.Request) {
|
||||
// Log the request information
|
||||
log.Printf("Received request: %s %s", r.Method, r.URL.Path)
|
||||
|
||||
// Check if the request has valid authentication
|
||||
if !checkAuthentication(r) {
|
||||
log.Println("Authentication failed")
|
||||
w.Header().Set("WWW-Authenticate", `Basic realm="CDN Authentication"`)
|
||||
w.WriteHeader(http.StatusUnauthorized)
|
||||
fmt.Fprintf(w, "401 Unauthorized\n")
|
||||
return
|
||||
}
|
||||
|
||||
// Get the requested file path
|
||||
filePath := filepath.Join("src", r.URL.Path)
|
||||
|
||||
// Check if the file exists
|
||||
_, err := os.Stat(filePath)
|
||||
if err != nil {
|
||||
if os.IsNotExist(err) {
|
||||
log.Printf("File not found: %s", r.URL.Path)
|
||||
http.NotFound(w, r)
|
||||
} else {
|
||||
log.Println("Internal Server Error")
|
||||
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// Serve the file
|
||||
log.Printf("Serving file: %s", r.URL.Path)
|
||||
http.ServeFile(w, r, filePath)
|
||||
}
|
||||
|
||||
func handleUpload(w http.ResponseWriter, r *http.Request) {
|
||||
// Log the request information
|
||||
log.Printf("Received upload request: %s %s", r.Method, r.URL.Path)
|
||||
|
||||
// Check if the request has valid authentication
|
||||
if !checkAuthentication(r) {
|
||||
log.Println("Authentication failed")
|
||||
w.Header().Set("WWW-Authenticate", `Basic realm="CDN Authentication"`)
|
||||
w.WriteHeader(http.StatusUnauthorized)
|
||||
fmt.Fprintf(w, "401 Unauthorized\n")
|
||||
return
|
||||
}
|
||||
|
||||
// Parse the uploaded file
|
||||
file, header, err := r.FormFile("file")
|
||||
if err != nil {
|
||||
log.Println("Bad Request")
|
||||
http.Error(w, "Bad Request", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
// Create the uploads directory if it doesn't exist
|
||||
err = os.MkdirAll(uploadPath, os.ModePerm)
|
||||
if err != nil {
|
||||
log.Println("Internal Server Error")
|
||||
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
// Create a new file in the uploads directory
|
||||
dst, err := os.Create(filepath.Join(uploadPath, header.Filename))
|
||||
if err != nil {
|
||||
log.Println("Internal Server Error")
|
||||
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
defer dst.Close()
|
||||
|
||||
// Copy the uploaded file to the destination
|
||||
_, err = io.Copy(dst, file)
|
||||
if err != nil {
|
||||
log.Println("Internal Server Error")
|
||||
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
log.Printf("File uploaded successfully: %s", header.Filename)
|
||||
fmt.Fprintf(w, "File uploaded successfully!")
|
||||
}
|
||||
|
||||
func checkAuthentication(r *http.Request) bool {
|
||||
username, password, ok := r.BasicAuth()
|
||||
if ok && username == "admin" && password == "password" {
|
||||
log.Println("Authentication successful")
|
||||
return true
|
||||
}
|
||||
log.Println("Authentication failed")
|
||||
return false
|
||||
}
|
Loading…
Reference in a new issue