dorfylegends/backend/server/loading.go

145 lines
3.1 KiB
Go
Raw Normal View History

2022-04-30 10:45:31 +03:00
package server
import (
"encoding/json"
"io/fs"
"io/ioutil"
"net/http"
"os"
"path/filepath"
"runtime"
"strings"
"github.com/robertjanetzko/LegendsBrowser2/backend/model"
"github.com/robertjanetzko/LegendsBrowser2/backend/util"
"github.com/shirou/gopsutil/disk"
)
type loadHandler struct {
server *DfServer
}
type loadProgress struct {
Msg string `json:"msg"`
Progress float64 `json:"progress"`
Done bool `json:"done"`
}
func (h loadHandler) Progress() *loadProgress {
percent := 0.0
p := h.server.context.progress
if p.ProgressBar != nil {
percent = float64(p.ProgressBar.Current()*100) / float64(p.ProgressBar.Total())
}
return &loadProgress{
Msg: h.server.context.progress.Message,
Progress: percent,
Done: h.server.context.world != nil,
}
}
func (h loadHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
if r.URL.Path == "/load/progress" {
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(http.StatusOK)
json.NewEncoder(w).Encode(h.Progress())
return
}
2022-05-09 18:20:31 +03:00
if h.server.context.config.ServerMode {
err := h.server.templates.Render(w, "serverMode.html", nil)
if err != nil {
httpError(w, err)
}
return
}
2022-04-30 10:45:31 +03:00
var partitions []string
if runtime.GOOS == "windows" {
ps, _ := disk.Partitions(false)
partitions = util.Map(ps, func(p disk.PartitionStat) string { return p.Mountpoint + `\` })
} else {
partitions = append(partitions, "/")
}
path := r.URL.Query().Get("p")
p := &paths{
Partitions: partitions,
Current: path,
}
if p.Current == "" {
2022-05-01 13:29:39 +03:00
p.Current = h.server.context.config.LastPath
2022-04-30 10:45:31 +03:00
}
var err error
p.Current, err = filepath.Abs(p.Current)
if err != nil {
httpError(w, err)
return
}
if f, err := os.Stat(p.Current); err == nil {
if f.IsDir() {
2022-05-01 13:29:39 +03:00
h.server.context.config.LastPath = p.Current
h.server.context.config.Save()
2022-04-30 10:45:31 +03:00
p.List, err = ioutil.ReadDir(p.Current)
if err != nil {
httpError(w, err)
return
}
err = h.server.templates.Render(w, "load.html", p)
if err != nil {
httpError(w, err)
}
return
} else {
2022-05-05 13:55:33 +03:00
h.server.context.config.LastFile = p.Current
h.server.context.config.Save()
2022-04-30 10:45:31 +03:00
h.server.context.isLoading = true
h.server.context.world = nil
go loadWorld(h.server, p.Current)
2022-05-09 18:20:31 +03:00
http.Redirect(w, r, h.server.context.config.SubUri+"/", http.StatusSeeOther)
2022-04-30 10:45:31 +03:00
return
}
}
2022-08-18 11:23:17 +03:00
if err != nil {
2023-04-27 15:55:32 +03:00
if !os.IsNotExist(err) {
httpError(w, err)
}
2022-08-18 11:23:17 +03:00
}
2023-04-27 15:55:32 +03:00
http.Redirect(w, r, h.server.context.config.SubUri+"/load?p=%2f", http.StatusSeeOther)
2022-04-30 10:45:31 +03:00
}
func isLegendsXml(f fs.FileInfo) bool {
return strings.HasSuffix(f.Name(), "-legends.xml")
}
func loadWorld(server *DfServer, file string) {
runtime.GC()
wrld, _ := model.Parse(file, server.context.progress)
server.context.world = wrld
server.context.isLoading = false
}
type paths struct {
Current string
List []fs.FileInfo
Partitions []string
}
func (srv *DfServer) renderLoading(w http.ResponseWriter, r *http.Request) {
if srv.context.isLoading {
err := srv.templates.Render(w, "loading.html", srv.loader.Progress())
if err != nil {
httpError(w, err)
}
} else {
2022-05-09 18:20:31 +03:00
http.Redirect(w, r, srv.context.config.SubUri+"/load", http.StatusSeeOther)
2022-04-30 10:45:31 +03:00
}
}