mirror of
https://github.com/fumiama/paper-manager.git
synced 2026-06-24 04:27:09 +08:00
add /api/getUserRegex
This commit is contained in:
源文雨
231
backend/paper.go
231
backend/paper.go
@@ -7,7 +7,6 @@ import (
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
sql "github.com/FloatTech/sqlite"
|
||||
"github.com/FloatTech/ttl"
|
||||
"github.com/fumiama/paper-manager/backend/global"
|
||||
"github.com/fumiama/paper-manager/backend/utils"
|
||||
@@ -22,123 +21,10 @@ const (
|
||||
var analyzeper = ttl.NewCache[int, uint](time.Hour)
|
||||
|
||||
var (
|
||||
errNoAnalyzePermission = errors.New("no analyze permission")
|
||||
errNoDeletePermission = errors.New("no delete permission")
|
||||
errNoDownloadPermission = errors.New("no download permission")
|
||||
errNoAnalyzePermission = errors.New("no analyze permission")
|
||||
)
|
||||
|
||||
type filelist struct {
|
||||
ID int `json:"id"`
|
||||
Title string `json:"title"`
|
||||
Desc string `json:"description"`
|
||||
Size float64 `json:"size"`
|
||||
Ques int `json:"questions"`
|
||||
Auth string `json:"author"`
|
||||
Date string `json:"datetime"`
|
||||
Per uint `json:"percent"`
|
||||
}
|
||||
|
||||
type filestatus struct {
|
||||
Name string `json:"name"`
|
||||
Size float64 `json:"size"`
|
||||
Rate float64 `json:"rate"`
|
||||
Questions []question `json:"questions"`
|
||||
Duplications []duplication `json:"duplications"` // Duplications length == 10
|
||||
}
|
||||
|
||||
func init() {
|
||||
apimap["/api/getFileList"] = &apihandler{"GET", func(w http.ResponseWriter, r *http.Request) {
|
||||
token := r.Header.Get("Authorization")
|
||||
user := usertokens.Get(token)
|
||||
if user == nil {
|
||||
writeresult(w, codeError, nil, errInvalidToken.Error(), typeError)
|
||||
return
|
||||
}
|
||||
istemp := r.URL.Query().Get("permanent") != "true"
|
||||
count := -1
|
||||
var err error
|
||||
countstr := r.URL.Query().Get("count")
|
||||
if countstr != "" {
|
||||
count, err = strconv.Atoi(countstr)
|
||||
if err != nil {
|
||||
writeresult(w, codeError, nil, err.Error(), typeError)
|
||||
return
|
||||
}
|
||||
}
|
||||
lst, err := global.FileDB.ListUploadedFile(istemp)
|
||||
if err != nil && err != sql.ErrNullResult {
|
||||
writeresult(w, codeError, nil, err.Error(), typeError)
|
||||
return
|
||||
}
|
||||
if count > 0 && len(lst) > count {
|
||||
lst = lst[:count]
|
||||
}
|
||||
result := make([]filelist, len(lst))
|
||||
for i, v := range lst {
|
||||
result[i].ID = *v.ID
|
||||
j := strings.LastIndex(v.Path, "/")
|
||||
if j <= 0 {
|
||||
result[i].Title = v.Path
|
||||
} else {
|
||||
result[i].Title = v.Path[j+1:]
|
||||
}
|
||||
result[i].Desc = v.Desc
|
||||
result[i].Size = float64(v.Size) / 1024 / 1024 // MB
|
||||
result[i].Ques = v.QuesC
|
||||
result[i].Auth = v.UpName
|
||||
result[i].Date = time.Unix(v.UpTime, 0).Format(chineseYYMMDDLayout)
|
||||
if !v.HasntAnalyzed {
|
||||
result[i].Per = 100
|
||||
} else {
|
||||
result[i].Per = analyzeper.Get(*v.ID)
|
||||
}
|
||||
}
|
||||
writeresult(w, codeSuccess, &result, messageOk, typeSuccess)
|
||||
}}
|
||||
apimap["/api/getFileInfo"] = &apihandler{"GET", func(w http.ResponseWriter, r *http.Request) {
|
||||
token := r.Header.Get("Authorization")
|
||||
user := usertokens.Get(token)
|
||||
if user == nil {
|
||||
writeresult(w, codeError, nil, errInvalidToken.Error(), typeError)
|
||||
return
|
||||
}
|
||||
var err error
|
||||
idstr := r.URL.Query().Get("id")
|
||||
if idstr == "" {
|
||||
writeresult(w, codeError, nil, "empty id", typeError)
|
||||
return
|
||||
}
|
||||
id, err := strconv.Atoi(idstr)
|
||||
if err != nil {
|
||||
writeresult(w, codeError, nil, err.Error(), typeError)
|
||||
return
|
||||
}
|
||||
lst, err := global.FileDB.ListFileByID(id)
|
||||
if err != nil && err != sql.ErrNullResult {
|
||||
writeresult(w, codeError, nil, err.Error(), typeError)
|
||||
return
|
||||
}
|
||||
result := filelist{
|
||||
ID: id,
|
||||
Desc: lst.Desc,
|
||||
Size: float64(lst.Size) / 1024 / 1024, // MB
|
||||
Ques: lst.QuesC,
|
||||
Auth: lst.UpName,
|
||||
Date: time.Unix(lst.UpTime, 0).Format(chineseYYMMDDLayout),
|
||||
}
|
||||
j := strings.LastIndex(lst.Path, "/")
|
||||
if j <= 0 {
|
||||
result.Title = lst.Path
|
||||
} else {
|
||||
result.Title = lst.Path[j+1:]
|
||||
}
|
||||
if !lst.HasntAnalyzed {
|
||||
result.Per = 100
|
||||
} else {
|
||||
result.Per = analyzeper.Get(id)
|
||||
}
|
||||
writeresult(w, codeSuccess, &result, messageOk, typeSuccess)
|
||||
}}
|
||||
apimap["/api/getFilePercent"] = &apihandler{"GET", func(w http.ResponseWriter, r *http.Request) {
|
||||
token := r.Header.Get("Authorization")
|
||||
user := usertokens.Get(token)
|
||||
@@ -158,6 +44,7 @@ func init() {
|
||||
}
|
||||
writeresult(w, codeSuccess, analyzeper.Get(id), messageOk, typeSuccess)
|
||||
}}
|
||||
|
||||
apimap["/api/analyzeFile"] = &apihandler{"GET", func(w http.ResponseWriter, r *http.Request) {
|
||||
token := r.Header.Get("Authorization")
|
||||
user := usertokens.Get(token)
|
||||
@@ -211,120 +98,6 @@ func init() {
|
||||
writeresult(w, codeSuccess, &message{C: 0, M: "分析完成"}, messageOk, typeSuccess)
|
||||
}
|
||||
}}
|
||||
apimap["/api/delFile"] = &apihandler{"GET", func(w http.ResponseWriter, r *http.Request) {
|
||||
token := r.Header.Get("Authorization")
|
||||
user := usertokens.Get(token)
|
||||
if user == nil {
|
||||
writeresult(w, codeError, nil, errInvalidToken.Error(), typeError)
|
||||
return
|
||||
}
|
||||
istemp := r.URL.Query().Get("permanent") != "true"
|
||||
if !user.IsSuper() && !istemp {
|
||||
writeresult(w, codeError, nil, errNoDeletePermission.Error(), typeError)
|
||||
return
|
||||
}
|
||||
idstr := r.URL.Query().Get("id")
|
||||
if idstr == "" {
|
||||
writeresult(w, codeError, nil, "empty id", typeError)
|
||||
return
|
||||
}
|
||||
id, err := strconv.Atoi(idstr)
|
||||
if err != nil {
|
||||
writeresult(w, codeError, nil, err.Error(), typeError)
|
||||
return
|
||||
}
|
||||
err = global.FileDB.DelFile(id, *user.ID, istemp)
|
||||
if err != nil {
|
||||
writeresult(w, codeError, nil, err.Error(), typeError)
|
||||
return
|
||||
}
|
||||
writeresult(w, codeSuccess, "删除成功", messageOk, typeSuccess)
|
||||
}}
|
||||
apimap["/api/dlFile"] = &apihandler{"GET", func(w http.ResponseWriter, r *http.Request) {
|
||||
token := r.Header.Get("Authorization")
|
||||
user := usertokens.Get(token)
|
||||
if user == nil {
|
||||
writeresult(w, codeError, nil, errInvalidToken.Error(), typeError)
|
||||
return
|
||||
}
|
||||
idstr := r.URL.Query().Get("id")
|
||||
if idstr == "" {
|
||||
writeresult(w, codeError, nil, "empty id", typeError)
|
||||
return
|
||||
}
|
||||
id, err := strconv.Atoi(idstr)
|
||||
if err != nil {
|
||||
writeresult(w, codeError, nil, err.Error(), typeError)
|
||||
return
|
||||
}
|
||||
lst, err := global.FileDB.ListFileByID(id)
|
||||
if err != nil {
|
||||
writeresult(w, codeError, nil, err.Error(), typeError)
|
||||
return
|
||||
}
|
||||
type message struct {
|
||||
URL string `json:"url"`
|
||||
}
|
||||
if strings.HasPrefix(lst.Path, global.PaperFolder+"tmp/") {
|
||||
uidstr := lst.Path[17:]
|
||||
i := strings.Index(uidstr, "/")
|
||||
if i <= 0 {
|
||||
writeresult(w, codeError, nil, "extract uid error", typeError)
|
||||
return
|
||||
}
|
||||
uid, err := strconv.Atoi(uidstr[:i])
|
||||
if err != nil {
|
||||
writeresult(w, codeError, nil, err.Error(), typeError)
|
||||
return
|
||||
}
|
||||
if uid != *user.ID {
|
||||
writeresult(w, codeError, nil, errNoDownloadPermission.Error(), typeError)
|
||||
return
|
||||
}
|
||||
writeresult(w, codeSuccess, &message{URL: lst.Path[6:]}, messageOk, typeSuccess)
|
||||
return
|
||||
}
|
||||
if strings.HasPrefix(lst.Path, global.PaperFolder) {
|
||||
writeresult(w, codeSuccess, &message{URL: lst.Path[6:]}, messageOk, typeSuccess)
|
||||
return
|
||||
}
|
||||
writeresult(w, codeError, nil, "parse filepath error", typeError)
|
||||
}}
|
||||
apimap["/api/getFileStatus"] = &apihandler{"GET", func(w http.ResponseWriter, r *http.Request) {
|
||||
token := r.Header.Get("Authorization")
|
||||
user := usertokens.Get(token)
|
||||
if user == nil {
|
||||
writeresult(w, codeError, nil, errInvalidToken.Error(), typeError)
|
||||
return
|
||||
}
|
||||
idstr := r.URL.Query().Get("id")
|
||||
if idstr == "" {
|
||||
writeresult(w, codeError, nil, "empty id", typeError)
|
||||
return
|
||||
}
|
||||
id, err := strconv.Atoi(idstr)
|
||||
if err != nil {
|
||||
writeresult(w, codeError, nil, err.Error(), typeError)
|
||||
return
|
||||
}
|
||||
file, sz, istemp, err := global.FileDB.GetFile(id, *user.ID)
|
||||
if err != nil {
|
||||
writeresult(w, codeError, nil, err.Error(), typeError)
|
||||
return
|
||||
}
|
||||
qs, ds, filerate, err := parseFileQuestions(file.Questions, istemp)
|
||||
if err != nil {
|
||||
writeresult(w, codeError, nil, err.Error(), typeError)
|
||||
return
|
||||
}
|
||||
writeresult(w, codeSuccess, &filestatus{
|
||||
Name: file.Class + ".docx",
|
||||
Size: float64(sz) / 1024 / 1024, // MB
|
||||
Rate: filerate * 100,
|
||||
Questions: qs,
|
||||
Duplications: ds,
|
||||
}, messageOk, typeSuccess)
|
||||
}}
|
||||
}
|
||||
|
||||
// PaperHandler serves protected contents in global.PaperFolder
|
||||
|
||||
Reference in New Issue
Block a user