diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..3af0ccb --- /dev/null +++ b/.gitignore @@ -0,0 +1 @@ +/data diff --git a/Web/info.html b/Web/info.html new file mode 100644 index 0000000..11a7aea --- /dev/null +++ b/Web/info.html @@ -0,0 +1,138 @@ + + + + + + + 文件详细 + + + + + + +
+

Pixel

+ +
+ +
+
+ 缩略图 +
+
+ + + + + + + + + + + + + + + + + +
URL
MarkDown![alt]()
HTML<img src="" alt="img" /> +
HTML[img][/img] +
+
+
+
+ + + + + + \ No newline at end of file diff --git a/Web/upload.html b/Web/upload.html index 18d2f1d..7302f8a 100644 --- a/Web/upload.html +++ b/Web/upload.html @@ -1,5 +1,6 @@ + @@ -36,7 +37,8 @@ padding: 20px; border-radius: 8px; box-shadow: 0 0 10px rgba(0, 0, 0, 0.1); - margin-top: 80px; /* 顶栏高度 + 20px 的间距 */ + margin-top: 80px; + /* 顶栏高度 + 20px 的间距 */ text-align: center; } @@ -54,7 +56,6 @@ background-color: #4caf50; color: #fff; border: none; - padding: 10px; border-radius: 4px; cursor: pointer; transition: background-color 0.3s; @@ -71,7 +72,6 @@ background-color: #4caf50; color: #fff; border: none; - padding: 10px; border-radius: 4px; cursor: pointer; transition: background-color 0.3s; @@ -99,7 +99,8 @@ margin-top: 20px; max-width: 100%; height: auto; - display: none; /* 默认隐藏 */ + display: none; + /* 默认隐藏 */ } #thumbnail { @@ -108,6 +109,7 @@ } +
@@ -158,7 +160,8 @@ xhr.onreadystatechange = function () { if (xhr.readyState === XMLHttpRequest.DONE) { if (xhr.status === 200) { - alert('文件上传成功！'); + gourl = "info?id=" + xhr.responseText + window.location.href = gourl } else { alert('文件上传失败！'); } @@ -198,4 +201,5 @@ - + + \ No newline at end of file diff --git a/database/db.go b/database/db.go index 2f35c9a..6e52890 100644 --- a/database/db.go +++ b/database/db.go @@ -18,7 +18,8 @@ func Initdb() { createTableSQL := ` CREATE TABLE IF NOT EXISTS mytable ( link TEXT PRIMARY KEY, - md5 TEXT NOT NULL + md5 TEXT NOT NULL, + ext TEXT NOT NULL ); ` @@ -28,7 +29,7 @@ func Initdb() { } } -func NewFile(linkID string, md5 string) { +func NewFile(linkID string, md5 string, ext string) { db, err := sql.Open("sqlite3", "./data/database.db") if err != nil { log.Fatal(err) @@ -37,7 +38,7 @@ func NewFile(linkID string, md5 string) { // SQL语句 SQL := ` - INSERT INTO "main"."mytable" ("link", "md5") VALUES (?, ?) + INSERT INTO "main"."mytable" ("link", "md5" ,"ext") VALUES (?, ? , ?) ` stmt, err := db.Prepare(SQL) @@ -45,13 +46,13 @@ func NewFile(linkID string, md5 string) { log.Fatal(err) } - _, err = stmt.Exec(linkID,md5) //插入记录 + _, err = stmt.Exec(linkID,md5,ext) //插入记录 if err != nil { log.Fatal(err) } } -func GetFileInfo(linkID string) string { +func GetFileName(linkID string) string { db, err := sql.Open("sqlite3", "./data/database.db") if err != nil { log.Fatal(err) @@ -71,7 +72,8 @@ func GetFileInfo(linkID string) string { // 扫描查询结果 var md5 string var linkIDx string - err = row.Scan(&linkIDx,&md5) + var ext string + err = row.Scan(&linkIDx,&md5,&ext) if err != nil { if err == sql.ErrNoRows { return "" @@ -80,5 +82,5 @@ func GetFileInfo(linkID string) string { } return "" } - return md5 + return md5+ext } \ No newline at end of file diff --git a/main.go b/main.go index 64146f0..d943fa5 100644 --- a/main.go +++ b/main.go @@ -11,6 +11,7 @@ import ( "net/http" "os" "path" + "path/filepath" "strconv" "time" ) @@ -31,7 +32,6 @@ func init() { ` fmt.Println(appinfo) - database.Initdb() //初始化数据库 dirPath := "./data/img" // 使用 os.Stat 检查目录是否存在 @@ -39,7 +39,7 @@ func init() { if os.IsNotExist(err) { // 目录不存在，可以调用 os.Mkdir 创建 - err := os.Mkdir(dirPath, 0755) + err := os.MkdirAll(dirPath, 0755) if err != nil { fmt.Println("无法创建目录:", err) return @@ -52,18 +52,28 @@ func init() { // 发生其他错误 fmt.Println("发生错误:", err) } + + database.Initdb() //初始化数据库 + fmt.Println("数据库初始化完成") } func main() { - - http.HandleFunc("/upload", upload) //设置访问的路由 + http.HandleFunc("/info", showimg) + http.HandleFunc("/upload", upload) + http.HandleFunc("/img/",downloadHandler)//设置访问的路由 + fmt.Println("Web服务器已启动") err := http.ListenAndServe(":9090", nil) //设置监听的端口 if err != nil { log.Fatal("ListenAndServe: ", err) } } +func showimg(w http.ResponseWriter, r *http.Request) { + t, _ := template.ParseFiles("Web/info.html") + t.Execute(w, "Hello") +} + // 处理/upload 逻辑 func upload(w http.ResponseWriter, r *http.Request) { @@ -116,8 +126,9 @@ func upload(w http.ResponseWriter, r *http.Request) { io.Copy(f, file) // 存入数据库 - database.NewFile(RandomString(10),md5sum) - + var linkid = RandomString(10) + database.NewFile(linkid,md5sum,ext) + w.Write([]byte(linkid)) } } @@ -131,3 +142,46 @@ func RandomString(n int) string { } return string(bytes) } + +func downloadHandler(w http.ResponseWriter, r *http.Request) { + // 获取请求参数，例如文件名 + filename := r.FormValue("id") + if filename == "" { + http.Error(w, "未提供文件名", http.StatusBadRequest) + return + } + + // 拼接文件路径，确保路径安全性 + filePath := filepath.Join("./data/img", database.GetFileName(filename)) + + // 打开文件 + file, err := os.Open(filePath) + if err != nil { + http.Error(w, "文件未找到", http.StatusNotFound) + return + } + defer file.Close() + + // 获取文件信息 + fileInfo, err := file.Stat() + if err != nil { + http.Error(w, "无法获取文件信息", http.StatusInternalServerError) + return + } + + // 设置响应头，告诉浏览器直接显示文件 + w.Header().Set("Content-Disposition", "inline; filename="+filename) + w.Header().Set("Content-Type", "image/jpeg") // 适用于 JPEG 图片，根据实际文件类型设置 + + // 将文件内容拷贝到响应体中 + http.ServeContent(w, r, filename, fileInfo.ModTime(), file) +} + +// 辅助函数，获取文件大小 +func fileSize(file *os.File) int64 { + stat, err := file.Stat() + if err != nil { + return 0 + } + return stat.Size() +}