This commit is contained in:
2025-05-15 19:38:13 +02:00
parent 35ce8c4707
commit 130af90e03
9 changed files with 360 additions and 62 deletions

View File

@@ -1,12 +1,16 @@
package sync package sync
import ( import (
"cloudsave/pkg/game"
"cloudsave/pkg/remote" "cloudsave/pkg/remote"
"cloudsave/pkg/remote/client"
"cloudsave/pkg/tools/prompt/credentials"
"context" "context"
"crypto/md5"
"flag" "flag"
"fmt" "fmt"
"io" "log/slog"
"net/http"
"net/url"
"os" "os"
"github.com/google/subcommands" "github.com/google/subcommands"
@@ -29,37 +33,99 @@ func (p *SyncCmd) SetFlags(f *flag.FlagSet) {
} }
func (p *SyncCmd) Execute(_ context.Context, f *flag.FlagSet, _ ...interface{}) subcommands.ExitStatus { func (p *SyncCmd) Execute(_ context.Context, f *flag.FlagSet, _ ...interface{}) subcommands.ExitStatus {
_, err := remote.All() remotes, err := remote.All()
if err != nil { if err != nil {
fmt.Fprintln(os.Stderr, "error: failed to load datastore:", err) fmt.Fprintln(os.Stderr, "error: failed to load datastore:", err)
return subcommands.ExitFailure return subcommands.ExitFailure
} }
for _, remote := range remotes { username, password, err := credentials.Read()
if err != nil {
fmt.Fprintln(os.Stderr, "error: failed to read std output:", err)
return subcommands.ExitFailure
}
for _, r := range remotes {
if !ping(r.URL, username, password) {
slog.Warn("remote is unavailable", "url", r.URL)
continue
}
client := client.New(r.URL, username, password)
hlocal, err := game.Hash(r.GameID)
if err != nil {
slog.Error(err.Error())
continue
}
hremote, _ := client.Hash(r.GameID)
vlocal, err := game.Version(r.GameID)
if err != nil {
slog.Error(err.Error())
continue
}
vremote, _ := client.Version(r.GameID)
if hlocal == hremote {
fmt.Println("already up-to-date")
continue
}
if vremote == 0 {
fmt.Println("push")
continue
}
if vlocal > vremote {
fmt.Println("push")
continue
}
if vlocal < vremote {
fmt.Println("pull")
continue
}
if vlocal == vremote {
fmt.Println("conflict")
continue
}
} }
return subcommands.ExitSuccess return subcommands.ExitSuccess
} }
func hash(path string) string { func ping(remote, username, password string) bool {
f, err := os.OpenFile(path, os.O_RDONLY, 0) cli := http.Client{}
hburl, err := url.JoinPath(remote, "heartbeat")
if err != nil { if err != nil {
notFound("id not found", w, r) fmt.Fprintln(os.Stderr, "cannot connect to remote:", err)
return return false
}
defer f.Close()
// Create MD5 hasher
hasher := md5.New()
// Copy file content into hasher
if _, err := io.Copy(hasher, f); err != nil {
fmt.Fprintln(os.Stderr, "error: an error occured while reading data:", err)
internalServerError(w, r)
return
} }
// Get checksum result req, err := http.NewRequest("GET", hburl, nil)
sum := hasher.Sum(nil) if err != nil {
fmt.Fprintln(os.Stderr, "cannot connect to remote:", err)
return false
} }
req.SetBasicAuth(username, password)
res, err := cli.Do(req)
if err != nil {
fmt.Fprintln(os.Stderr, "cannot connect to remote:", err)
return false
}
if res.StatusCode != http.StatusOK {
fmt.Fprintln(os.Stderr, "cannot connect to remote: server return code", res.StatusCode)
return false
}
return true
}

View File

@@ -57,6 +57,7 @@ func NewServer(documentRoot string, creds map[string]string, port int) *HTTPServ
saveRouter.Post("/{id}/data", s.upload) saveRouter.Post("/{id}/data", s.upload)
saveRouter.Get("/{id}/data", s.download) saveRouter.Get("/{id}/data", s.download)
saveRouter.Get("/{id}/hash", s.hash) saveRouter.Get("/{id}/hash", s.hash)
saveRouter.Get("/{id}/version", s.version)
}) })
}) })
}) })
@@ -220,3 +221,39 @@ func (s HTTPServer) hash(w http.ResponseWriter, r *http.Request) {
sum := hasher.Sum(nil) sum := hasher.Sum(nil)
ok(sum, w, r) ok(sum, w, r)
} }
func (s HTTPServer) version(w http.ResponseWriter, r *http.Request) {
id := chi.URLParam(r, "id")
path := filepath.Clean(filepath.Join(s.documentRoot, "data", id))
sdir, err := os.Stat(path)
if err != nil {
notFound("id not found", w, r)
return
}
if !sdir.IsDir() {
notFound("id not found", w, r)
return
}
path = filepath.Join(path, "metadata.json")
f, err := os.OpenFile(path, os.O_RDONLY, 0)
if err != nil {
notFound("id not found", w, r)
return
}
defer f.Close()
var metadata game.Metadata
d := json.NewDecoder(f)
err = d.Decode(&metadata)
if err != nil {
fmt.Fprintln(os.Stderr, "error: an error occured while reading data:", err)
internalServerError(w, r)
return
}
ok(metadata.Version, w, r)
}

View File

@@ -1,34 +1,16 @@
package api package api
import ( import (
"cloudsave/pkg/remote/obj"
"encoding/json" "encoding/json"
"log" "log"
"net/http" "net/http"
"time" "time"
) )
type (
httpCore struct {
Status int `json:"status"`
Timestamp time.Time `json:"timestamp"`
Path string `json:"path"`
}
httpError struct {
httpCore
Error string `json:"error"`
Message string `json:"message"`
}
httpObject struct {
httpCore
Data any `json:"data"`
}
)
func internalServerError(w http.ResponseWriter, r *http.Request) { func internalServerError(w http.ResponseWriter, r *http.Request) {
e := httpError{ e := obj.HTTPError{
httpCore: httpCore{ HTTPCore: obj.HTTPCore{
Status: http.StatusInternalServerError, Status: http.StatusInternalServerError,
Path: r.RequestURI, Path: r.RequestURI,
Timestamp: time.Now(), Timestamp: time.Now(),
@@ -50,8 +32,8 @@ func internalServerError(w http.ResponseWriter, r *http.Request) {
} }
func notFound(message string, w http.ResponseWriter, r *http.Request) { func notFound(message string, w http.ResponseWriter, r *http.Request) {
e := httpError{ e := obj.HTTPError{
httpCore: httpCore{ HTTPCore: obj.HTTPCore{
Status: http.StatusNotFound, Status: http.StatusNotFound,
Path: r.RequestURI, Path: r.RequestURI,
Timestamp: time.Now(), Timestamp: time.Now(),
@@ -73,8 +55,8 @@ func notFound(message string, w http.ResponseWriter, r *http.Request) {
} }
func methodNotAllowed(w http.ResponseWriter, r *http.Request) { func methodNotAllowed(w http.ResponseWriter, r *http.Request) {
e := httpError{ e := obj.HTTPError{
httpCore: httpCore{ HTTPCore: obj.HTTPCore{
Status: http.StatusMethodNotAllowed, Status: http.StatusMethodNotAllowed,
Path: r.RequestURI, Path: r.RequestURI,
Timestamp: time.Now(), Timestamp: time.Now(),
@@ -96,8 +78,8 @@ func methodNotAllowed(w http.ResponseWriter, r *http.Request) {
} }
func unauthorized(w http.ResponseWriter, r *http.Request) { func unauthorized(w http.ResponseWriter, r *http.Request) {
e := httpError{ e := obj.HTTPError{
httpCore: httpCore{ HTTPCore: obj.HTTPCore{
Status: http.StatusUnauthorized, Status: http.StatusUnauthorized,
Path: r.RequestURI, Path: r.RequestURI,
Timestamp: time.Now(), Timestamp: time.Now(),
@@ -120,8 +102,8 @@ func unauthorized(w http.ResponseWriter, r *http.Request) {
} }
func forbidden(w http.ResponseWriter, r *http.Request) { func forbidden(w http.ResponseWriter, r *http.Request) {
e := httpError{ e := obj.HTTPError{
httpCore: httpCore{ HTTPCore: obj.HTTPCore{
Status: http.StatusForbidden, Status: http.StatusForbidden,
Path: r.RequestURI, Path: r.RequestURI,
Timestamp: time.Now(), Timestamp: time.Now(),
@@ -142,14 +124,14 @@ func forbidden(w http.ResponseWriter, r *http.Request) {
} }
} }
func ok(obj interface{}, w http.ResponseWriter, r *http.Request) { func ok(o interface{}, w http.ResponseWriter, r *http.Request) {
e := httpObject{ e := obj.HTTPObject{
httpCore: httpCore{ HTTPCore: obj.HTTPCore{
Status: http.StatusOK, Status: http.StatusOK,
Path: r.RequestURI, Path: r.RequestURI,
Timestamp: time.Now(), Timestamp: time.Now(),
}, },
Data: obj, Data: o,
} }
payload, err := json.Marshal(e) payload, err := json.Marshal(e)
@@ -164,8 +146,8 @@ func ok(obj interface{}, w http.ResponseWriter, r *http.Request) {
} }
func badRequest(message string, w http.ResponseWriter, r *http.Request) { func badRequest(message string, w http.ResponseWriter, r *http.Request) {
e := httpError{ e := obj.HTTPError{
httpCore: httpCore{ HTTPCore: obj.HTTPCore{
Status: http.StatusBadRequest, Status: http.StatusBadRequest,
Path: r.RequestURI, Path: r.RequestURI,
Timestamp: time.Now(), Timestamp: time.Now(),

3
go.mod
View File

@@ -6,4 +6,7 @@ require (
github.com/go-chi/chi/v5 v5.2.1 github.com/go-chi/chi/v5 v5.2.1
github.com/google/subcommands v1.2.0 github.com/google/subcommands v1.2.0
golang.org/x/crypto v0.38.0 golang.org/x/crypto v0.38.0
golang.org/x/term v0.32.0
) )
require golang.org/x/sys v0.33.0 // indirect

4
go.sum
View File

@@ -4,3 +4,7 @@ github.com/google/subcommands v1.2.0 h1:vWQspBTo2nEqTUFita5/KeEWlUL8kQObDFbub/EN
github.com/google/subcommands v1.2.0/go.mod h1:ZjhPrFU+Olkh9WazFPsl27BQ4UPiG37m3yTrtFlrHVk= github.com/google/subcommands v1.2.0/go.mod h1:ZjhPrFU+Olkh9WazFPsl27BQ4UPiG37m3yTrtFlrHVk=
golang.org/x/crypto v0.38.0 h1:jt+WWG8IZlBnVbomuhg2Mdq0+BBQaHbtqHEFEigjUV8= golang.org/x/crypto v0.38.0 h1:jt+WWG8IZlBnVbomuhg2Mdq0+BBQaHbtqHEFEigjUV8=
golang.org/x/crypto v0.38.0/go.mod h1:MvrbAqul58NNYPKnOra203SB9vpuZW0e+RRZV+Ggqjw= golang.org/x/crypto v0.38.0/go.mod h1:MvrbAqul58NNYPKnOra203SB9vpuZW0e+RRZV+Ggqjw=
golang.org/x/sys v0.33.0 h1:q3i8TbbEz+JRD9ywIRlyRAQbM0qF7hu24q3teo2hbuw=
golang.org/x/sys v0.33.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
golang.org/x/term v0.32.0 h1:DR4lr0TjUs3epypdhTOkMmuF5CDFJ/8pOnbzMZPQ7bg=
golang.org/x/term v0.32.0/go.mod h1:uZG1FhGx848Sqfsq4/DlJr3xGGsYMu/L5GW4abiaEPQ=

View File

@@ -2,8 +2,11 @@ package game
import ( import (
"cloudsave/pkg/tools/id" "cloudsave/pkg/tools/id"
"crypto/md5"
"encoding/hex"
"encoding/json" "encoding/json"
"fmt" "fmt"
"io"
"os" "os"
"path/filepath" "path/filepath"
) )
@@ -13,6 +16,7 @@ type (
ID string `json:"id"` ID string `json:"id"`
Name string `json:"name"` Name string `json:"name"`
Path string `json:"path"` Path string `json:"path"`
Version int `json:"version"`
} }
) )
@@ -99,8 +103,66 @@ func Remove(gameID string) error {
} }
func Hash(gameID string) (string, error) { func Hash(gameID string) (string, error) {
content, err := os.ReadFile(filepath.Join(datastorepath, d.Name(), "data.tar.gz")) path := filepath.Join(datastorepath, gameID, "data.tar.gz")
if err != nil {
f, err := os.OpenFile(path, os.O_RDONLY, 0)
if err != nil {
return "", err
} }
defer f.Close()
hasher := md5.New()
if _, err := io.Copy(hasher, f); err != nil {
return "", err
}
sum := hasher.Sum(nil)
return hex.EncodeToString(sum), nil
}
func Version(gameID string) (int, error) {
path := filepath.Join(datastorepath, gameID, "metadata.json")
f, err := os.OpenFile(path, os.O_RDONLY, 0)
if err != nil {
return 0, err
}
defer f.Close()
var metadata Metadata
d := json.NewDecoder(f)
err = d.Decode(&metadata)
if err != nil {
return 0, err
}
return metadata.Version, nil
}
func SetVersion(gameID string, version int) error {
path := filepath.Join(datastorepath, gameID, "metadata.json")
f, err := os.OpenFile(path, os.O_RDWR, 0740)
if err != nil {
return err
}
defer f.Close()
var metadata Metadata
d := json.NewDecoder(f)
err = d.Decode(&metadata)
if err != nil {
return err
}
f.Seek(0, io.SeekStart)
metadata.Version = version
e := json.NewEncoder(f)
err = e.Encode(metadata)
if err != nil {
return err
}
return nil
} }

View File

@@ -0,0 +1,96 @@
package client
import (
"cloudsave/pkg/remote/obj"
"encoding/json"
"errors"
"fmt"
"net/http"
"net/url"
)
type (
Client struct {
baseURL string
username string
password string
}
)
func New(baseURL, username, password string) *Client {
return &Client{
baseURL: baseURL,
username: username,
password: password,
}
}
func (c *Client) Hash(gameID string) (string, error) {
u, err := url.JoinPath(c.baseURL, "api", "v1", "game", gameID, "hash")
if err != nil {
return "", err
}
o, err := c.get(u)
if err != nil {
return "", err
}
if h, ok := (o).(string); ok {
return h, nil
}
return "", errors.New("invalid payload sent by the server")
}
func (c *Client) Version(gameID string) (int, error) {
u, err := url.JoinPath(c.baseURL, "api", "v1", "game", gameID, "version")
if err != nil {
return 0, err
}
o, err := c.get(u)
if err != nil {
return 0, err
}
if h, ok := (o).(int); ok {
return h, nil
}
return 0, errors.New("invalid payload sent by the server")
}
func (c *Client) get(url string) (any, error) {
cli := http.Client{}
req, err := http.NewRequest("GET", url, nil)
if err != nil {
return nil, err
}
req.SetBasicAuth(c.username, c.password)
res, err := cli.Do(req)
if err != nil {
return nil, err
}
defer res.Body.Close()
if res.StatusCode != 200 {
return nil, fmt.Errorf("server returns an unexpected status code: %d %s", res.StatusCode, res.Status)
}
var httpObject obj.HTTPObject
d := json.NewDecoder(res.Body)
err = d.Decode(&httpObject)
if err != nil {
return nil, err
}
return httpObject, nil
}
func (c *Client) post() {
}

22
pkg/remote/obj/obj.go Normal file
View File

@@ -0,0 +1,22 @@
package obj
import "time"
type (
HTTPCore struct {
Status int `json:"status"`
Timestamp time.Time `json:"timestamp"`
Path string `json:"path"`
}
HTTPError struct {
HTTPCore
Error string `json:"error"`
Message string `json:"message"`
}
HTTPObject struct {
HTTPCore
Data any `json:"data"`
}
)

View File

@@ -0,0 +1,26 @@
package credentials
import (
"bufio"
"fmt"
"os"
"strings"
"golang.org/x/term"
)
func Read() (string, string, error) {
fmt.Print("Enter username: ")
reader := bufio.NewReader(os.Stdin)
username, _ := reader.ReadString('\n')
username = strings.TrimSpace(username)
fmt.Printf("password for %s: ", username)
password, err := term.ReadPassword(int(os.Stdin.Fd()))
if err != nil {
return "", "", err
}
fmt.Println()
return username, string(password), nil
}