114 lines
2.4 KiB
Go
114 lines
2.4 KiB
Go
package web
|
|
|
|
import (
|
|
"archive/zip"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"path"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/dustin/go-humanize"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/go-git/go-git/v5"
|
|
"github.com/go-git/go-git/v5/plumbing"
|
|
"github.com/go-git/go-git/v5/plumbing/object"
|
|
)
|
|
|
|
func (web *Web) mapVersions(c *gin.Context) {
|
|
userId := c.Param("userId")
|
|
mapId := c.Param("mapId")
|
|
|
|
id, _ := strconv.Atoi(mapId)
|
|
bs, _ := web.api.GetBeatmapSet(id)
|
|
|
|
repoDir := path.Join(web.config.Repos, userId, mapId)
|
|
repo, _ := git.PlainOpen(repoDir)
|
|
|
|
type Revision struct {
|
|
Date time.Time
|
|
HumanDate string
|
|
Summary string
|
|
Hash string
|
|
HasParent bool
|
|
}
|
|
|
|
versions := make([]Revision, 0)
|
|
logIter, _ := repo.Log(&git.LogOptions{})
|
|
for i := 0; i < 20; i++ {
|
|
commit, err := logIter.Next()
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
|
|
stats, _ := commit.Stats()
|
|
_, err = commit.Parent(0)
|
|
hasParent := !errors.Is(err, object.ErrParentNotFound)
|
|
|
|
versions = append(versions, Revision{
|
|
Date: commit.Author.When,
|
|
HumanDate: humanize.Time(commit.Author.When),
|
|
Summary: stats.String(),
|
|
Hash: commit.Hash.String(),
|
|
HasParent: hasParent,
|
|
})
|
|
}
|
|
|
|
c.HTML(http.StatusOK, "map-version.html", gin.H{
|
|
"Beatmapset": bs,
|
|
"LoggedIn": isLoggedIn(c),
|
|
"Versions": versions,
|
|
})
|
|
}
|
|
|
|
func (web *Web) mapPatch(c *gin.Context) {
|
|
userId := c.Param("userId")
|
|
mapId := c.Param("mapId")
|
|
hash := c.Param("hash")
|
|
|
|
repoDir := path.Join(web.config.Repos, userId, mapId)
|
|
repo, _ := git.PlainOpen(repoDir)
|
|
|
|
hashObj := plumbing.NewHash(hash)
|
|
commit, _ := repo.CommitObject(hashObj)
|
|
parent, _ := commit.Parent(0)
|
|
patch, _ := parent.Patch(commit)
|
|
|
|
c.String(http.StatusOK, "text/plain", patch.String())
|
|
}
|
|
|
|
func (web *Web) mapZip(c *gin.Context) {
|
|
userId := c.Param("userId")
|
|
mapId := c.Param("mapId")
|
|
hash := c.Param("hash")
|
|
|
|
repoDir := path.Join(web.config.Repos, userId, mapId)
|
|
repo, _ := git.PlainOpen(repoDir)
|
|
|
|
hashObj := plumbing.NewHash(hash)
|
|
commit, _ := repo.CommitObject(hashObj)
|
|
tree, _ := commit.Tree()
|
|
|
|
files := tree.Files()
|
|
|
|
c.Writer.Header().Set("Content-type", "application/octet-stream")
|
|
c.Writer.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=%s-%s.zip", mapId, hash))
|
|
c.Stream(func(w io.Writer) bool {
|
|
ar := zip.NewWriter(w)
|
|
for {
|
|
file, err := files.Next()
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
|
|
reader, _ := file.Reader()
|
|
fdest, _ := ar.Create(file.Name)
|
|
io.Copy(fdest, reader)
|
|
}
|
|
ar.Close()
|
|
|
|
return false
|
|
})
|
|
}
|