kaku/Pixel
1
0
Fork 0
mirror of https://github.com/Kakune55/Pixel.git synced 2025-05-06 18:29:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

添加配置文件功能

Browse Source
This commit is contained in:
Kakune55 2023-12-21 09:22:22 +08:00
parent 4dde2ba1ab
commit d701f59a4e
3 changed files with 114 additions and 65 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@ -1 +1,2 @@
/data /data
user.json

44
config/load.go Normal file
View File

@ -0,0 +1,44 @@
package config
import (
"encoding/json"
"os"
)
type User struct {
Listen string `json:"listen"`
}
func ReadConfig(filePath string) (User, error) {
// 检查文件是否存在
if _, err := os.Stat(filePath); os.IsNotExist(err) {
// 文件不存在，创建一个新的JSON文件
user := User{Listen: ":9090"}
data, err := json.Marshal(user)
if err != nil {
return User{}, err
}
err = os.WriteFile(filePath, data, 0644)
if err != nil {
return User{}, err
}
}
// 读取文件内容
data, err := os.ReadFile(filePath)
if err != nil {
return User{}, err
}
// 解析JSON数据
var user User
err = json.Unmarshal(data, &user)
if err != nil {
return User{}, err
}
return user, nil
}
//如何调用
//---------------------------------------
// filePath := "user.json"
// // 读取配置文件
// user, err := ReadConfig(filePath)
// if err != nil {
// fmt.Println("读取配置文件失败：", err)
// }

130
main.go
View File

@ -1,6 +1,7 @@
package main package main
import ( import (
"Pixel/config"
"Pixel/database" "Pixel/database"
"crypto/md5" "crypto/md5"
"encoding/json" "encoding/json"
@ -20,7 +21,6 @@ import (
"github.com/disintegration/imaging" "github.com/disintegration/imaging"
) )
func init() { func init() {
const appinfo string = ` const appinfo string = `
@ -61,18 +61,24 @@ func init() {
fmt.Println("数据库初始化完成") fmt.Println("数据库初始化完成")
} }
func main() { func main() {
filePath := "user.json"
// 读取配置文件
config, err := config.ReadConfig(filePath)
if err != nil {
fmt.Println("读取配置文件失败：", err)
}
http.HandleFunc("/info", showimg) http.HandleFunc("/info", showimg)
http.HandleFunc("/info/list", showlist)// http.HandleFunc("/info/list", showlist) //
http.HandleFunc("/upload", upload)//上传图片 http.HandleFunc("/upload", upload) //上传图片
http.HandleFunc("/img/",downloadHandler)//图片接口 http.HandleFunc("/img/", downloadHandler) //图片接口
http.HandleFunc("/img/mini",displayThumbnailHandler)//缩略图接口 http.HandleFunc("/img/mini", displayThumbnailHandler) //缩略图接口
http.HandleFunc("/idlist",arrayHandler)//获取现有图片id http.HandleFunc("/idlist", arrayHandler) //获取现有图片id
http.HandleFunc("/img/del",deleteImagesHandler)//删除相应图片 http.HandleFunc("/img/del", deleteImagesHandler) //删除相应图片
http.HandleFunc("/login",login)//登录页 http.HandleFunc("/login", login) //登录页
fmt.Println("Web服务器已启动") fmt.Println("Web服务器已启动")
err := http.ListenAndServe(":9090", nil) //设置监听的端口 err = http.ListenAndServe(config.Listen, nil) //设置监听的端口
if err != nil { if err != nil {
log.Fatal("ListenAndServe: ", err) log.Fatal("ListenAndServe: ", err)
} }
@ -85,7 +91,7 @@ func showimg(w http.ResponseWriter, r *http.Request) {
func showlist(w http.ResponseWriter, r *http.Request) { func showlist(w http.ResponseWriter, r *http.Request) {
cookie, _ := r.Cookie("login") cookie, _ := r.Cookie("login")
if cookie == nil{ //未授权禁止访问 if cookie == nil { //未授权禁止访问
w.WriteHeader(401) w.WriteHeader(401)
w.Write([]byte(`<html><a href="/login">验证失败 点此登录</a><html>`)) w.Write([]byte(`<html><a href="/login">验证失败 点此登录</a><html>`))
return return
@ -95,65 +101,63 @@ func showlist(w http.ResponseWriter, r *http.Request) {
t.Execute(w, "Hello") t.Execute(w, "Hello")
} }
// 处理/upload 逻辑 // 处理/upload 逻辑
func upload(w http.ResponseWriter, r *http.Request) { func upload(w http.ResponseWriter, r *http.Request) {
fmt.Println("method:", r.Method) // 获取请求的方法 fmt.Println("method:", r.Method) // 获取请求的方法
if r.Method == "GET" { // 前端页面渲染 if r.Method == "GET" { // 前端页面渲染
crutime := time.Now().Unix() crutime := time.Now().Unix()
h := md5.New() h := md5.New()
io.WriteString(h, strconv.FormatInt(crutime, 10)) io.WriteString(h, strconv.FormatInt(crutime, 10))
token := fmt.Sprintf("%x", h.Sum(nil)) token := fmt.Sprintf("%x", h.Sum(nil))
t, _ := template.ParseFiles("Web/upload.html") t, _ := template.ParseFiles("Web/upload.html")
t.Execute(w, token) t.Execute(w, token)
} else { // 后端POST接收逻辑 } else { // 后端POST接收逻辑
r.ParseMultipartForm(32 << 20) r.ParseMultipartForm(32 << 20)
file, handler, err := r.FormFile("file") file, handler, err := r.FormFile("file")
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
return return
} }
defer file.Close() defer file.Close()
// 生成文件的MD5哈希 // 生成文件的MD5哈希
h := md5.New() h := md5.New()
if _, err := io.Copy(h, file); err != nil { if _, err := io.Copy(h, file); err != nil {
fmt.Println(err) fmt.Println(err)
return return
} }
md5sum := fmt.Sprintf("%x", h.Sum(nil)) md5sum := fmt.Sprintf("%x", h.Sum(nil))
// 获取文件扩展名 // 获取文件扩展名
fname := handler.Filename fname := handler.Filename
ext := path.Ext(fname) ext := path.Ext(fname)
// 创建新文件，使用MD5哈希和原始扩展名 // 创建新文件，使用MD5哈希和原始扩展名
newFilename := md5sum + ext newFilename := md5sum + ext
f, err := os.OpenFile("./data/img/"+newFilename, os.O_WRONLY|os.O_CREATE, 0666) f, err := os.OpenFile("./data/img/"+newFilename, os.O_WRONLY|os.O_CREATE, 0666)
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
return return
} }
defer f.Close() defer f.Close()
// 将文件内容拷贝到新文件 // 将文件内容拷贝到新文件
_, err = file.Seek(0, 0) _, err = file.Seek(0, 0)
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
return return
} }
io.Copy(f, file) io.Copy(f, file)
// 存入数据库 // 存入数据库
var linkid = RandomString(10) var linkid = RandomString(10)
database.NewFile(linkid,md5sum,ext) database.NewFile(linkid, md5sum, ext)
w.Write([]byte(linkid)) w.Write([]byte(linkid))
} }
} }
func RandomString(n int) string { func RandomString(n int) string {
const letters = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz" const letters = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"
bytes := make([]byte, n) bytes := make([]byte, n)
@ -248,9 +252,9 @@ func displayThumbnailHandler(w http.ResponseWriter, r *http.Request) {
} }
} }
func arrayHandler(w http.ResponseWriter, r *http.Request) { //获取全部图片ID func arrayHandler(w http.ResponseWriter, r *http.Request) { //获取全部图片ID
cookie, _ := r.Cookie("login") cookie, _ := r.Cookie("login")
if cookie == nil{ //未授权禁止访问 if cookie == nil { //未授权禁止访问
w.WriteHeader(401) w.WriteHeader(401)
w.Write([]byte(`<html><a href="/login">验证失败 点此登录</a><html>`)) w.Write([]byte(`<html><a href="/login">验证失败 点此登录</a><html>`))
return return
@ -277,7 +281,7 @@ func arrayHandler(w http.ResponseWriter, r *http.Request) { //获取全部图
func deleteImagesHandler(w http.ResponseWriter, r *http.Request) { func deleteImagesHandler(w http.ResponseWriter, r *http.Request) {
cookie, _ := r.Cookie("login") cookie, _ := r.Cookie("login")
if cookie == nil{ //未授权禁止访问 if cookie == nil { //未授权禁止访问
w.WriteHeader(401) w.WriteHeader(401)
w.Write([]byte(`<html><a href="/login">验证失败 点此登录</a><html>`)) w.Write([]byte(`<html><a href="/login">验证失败 点此登录</a><html>`))
return return
@ -311,15 +315,15 @@ func login(w http.ResponseWriter, r *http.Request) {
if r.Method == "GET" { if r.Method == "GET" {
t, _ := template.ParseFiles("Web/login.html") t, _ := template.ParseFiles("Web/login.html")
w.Header().Set("Content-Type", "text/html") w.Header().Set("Content-Type", "text/html")
t.Execute(w,"") t.Execute(w, "")
} else { } else {
userlist,_:= database.QueryUser() userlist, _ := database.QueryUser()
fmt.Println(userlist) fmt.Println(userlist)
if len(userlist) == 0 { if len(userlist) == 0 {
database.NewUser("admin", r.FormValue("passwd")) database.NewUser("admin", r.FormValue("passwd"))
} else { } else {
if !database.CheckUserPasswd("admin", r.FormValue("passwd")) { if !database.CheckUserPasswd("admin", r.FormValue("passwd")) {
http.Redirect(w, r, "/login",http.StatusFound) http.Redirect(w, r, "/login", http.StatusFound)
fmt.Println("密码错误") fmt.Println("密码错误")
return return
} }
@ -327,6 +331,6 @@ func login(w http.ResponseWriter, r *http.Request) {
cookie := http.Cookie{Name: "login", Value: "yes"} cookie := http.Cookie{Name: "login", Value: "yes"}
http.SetCookie(w, &cookie) http.SetCookie(w, &cookie)
fmt.Println("密码正确") fmt.Println("密码正确")
http.Redirect(w, r, "/info/list",http.StatusFound) http.Redirect(w, r, "/info/list", http.StatusFound)
} }
} }