kpsync/app/webdav.go

package app

import (
	"errors"
	"io"
	"net/http"
	"os"
	"strings"
	"time"

	"git.blackforestbytes.com/BlackForestBytes/goext/cryptext"
	"git.blackforestbytes.com/BlackForestBytes/goext/exerr"
	"git.blackforestbytes.com/BlackForestBytes/goext/timeext"
)

var ETagConflictError = errors.New("ETag conflict")

func (app *Application) downloadDatabase() (string, time.Time, string, int64, error) {

	client := http.Client{Timeout: 90 * time.Second}

	req, err := http.NewRequest("GET", app.config.WebDAVURL, nil)
	if err != nil {
		return "", time.Time{}, "", 0, exerr.Wrap(err, "").Build()
	}

	req.SetBasicAuth(app.config.WebDAVUser, app.config.WebDAVPass)

	resp, err := client.Do(req)
	if err != nil {
		return "", time.Time{}, "", 0, exerr.Wrap(err, "Failed to download remote database").Build()
	}

	if resp.StatusCode != http.StatusOK {
		return "", time.Time{}, "", 0, exerr.New(exerr.TypeInternal, "Failed to download remote database").Int("sc", resp.StatusCode).Build()
	}

	bin, err := io.ReadAll(resp.Body)
	if err != nil {
		return "", time.Time{}, "", 0, exerr.Wrap(err, "Failed to read response body").Build()
	}

	etag := resp.Header.Get("ETag")
	if etag == "" {
		return "", time.Time{}, "", 0, exerr.New(exerr.TypeInternal, "ETag header is missing").Build()
	}
	etag = strings.Trim(etag, "\"\r\n ")

	lmStr := resp.Header.Get("Last-Modified")
	lm, err := time.Parse("Mon, 02 Jan 2006 15:04:05 MST", lmStr)
	if err != nil {
		return "", time.Time{}, "", 0, exerr.Wrap(err, "Failed to parse Last-Modified header").Build()
	}
	lm = lm.In(timeext.TimezoneBerlin)

	sha := cryptext.BytesSha256(bin)

	sz := int64(len(bin))

	err = os.WriteFile(app.dbFile, bin, 0644)
	if err != nil {
		return "", time.Time{}, "", 0, exerr.Wrap(err, "Failed to write database file").Build()
	}

	return etag, lm, sha, sz, nil
}

func (app *Application) getRemoteETag() (string, time.Time, error) {
	client := http.Client{Timeout: 90 * time.Second}

	req, err := http.NewRequest("HEAD", app.config.WebDAVURL, nil)
	if err != nil {
		return "", time.Time{}, exerr.Wrap(err, "").Build()
	}

	req.SetBasicAuth(app.config.WebDAVUser, app.config.WebDAVPass)

	resp, err := client.Do(req)
	if err != nil {
		return "", time.Time{}, exerr.Wrap(err, "Failed to download remote database").Build()
	}

	if resp.StatusCode != http.StatusOK {
		return "", time.Time{}, exerr.New(exerr.TypeInternal, "Failed to download remote database").Int("sc", resp.StatusCode).Build()
	}

	etag := resp.Header.Get("ETag")
	if etag == "" {
		return "", time.Time{}, exerr.New(exerr.TypeInternal, "ETag header is missing").Build()
	}
	etag = strings.Trim(etag, "\"\r\n ")

	lmStr := resp.Header.Get("Last-Modified")

	lm, err := time.Parse("Mon, 02 Jan 2006 15:04:05 MST", lmStr)
	if err != nil {
		return "", time.Time{}, exerr.Wrap(err, "Failed to parse Last-Modified header").Build()
	}

	lm = lm.In(timeext.TimezoneBerlin)

	return etag, lm, nil
}

func (app *Application) uploadDatabase(etagIfMatch *string) (string, time.Time, string, int64, error) {

	client := http.Client{Timeout: 90 * time.Second}

	req, err := http.NewRequest("PUT", app.config.WebDAVURL, nil)
	if err != nil {
		return "", time.Time{}, "", 0, exerr.Wrap(err, "").Build()
	}

	if etagIfMatch != nil {
		req.Header.Set("If-Match", "\""+*etagIfMatch+"\"")
	}

	bin, err := os.ReadFile(app.dbFile)
	if err != nil {
		return "", time.Time{}, "", 0, exerr.Wrap(err, "Failed to read database file").Build()
	}

	sha := cryptext.BytesSha256(bin)

	sz := int64(len(bin))

	req.ContentLength = sz

	resp, err := client.Do(req)
	if err != nil {
		return "", time.Time{}, "", 0, exerr.Wrap(err, "Failed to upload remote database").Build()
	}
	defer func() { _ = resp.Body.Close() }()

	if resp.StatusCode == http.StatusOK || resp.StatusCode == http.StatusCreated || resp.StatusCode == http.StatusNoContent {

		etag := resp.Header.Get("ETag")
		if etag == "" {
			return "", time.Time{}, "", 0, exerr.New(exerr.TypeInternal, "ETag header is missing").Build()
		}
		etag = strings.Trim(etag, "\"\r\n ")

		lmStr := resp.Header.Get("Last-Modified")
		lm, err := time.Parse("Mon, 02 Jan 2006 15:04:05 MST", lmStr)
		if err != nil {
			return "", time.Time{}, "", 0, exerr.Wrap(err, "Failed to parse Last-Modified header").Build()
		}
		lm = lm.In(timeext.TimezoneBerlin)

		return etag, lm, sha, sz, nil
	}

	if resp.StatusCode == http.StatusPreconditionFailed {
		return "", time.Time{}, "", 0, ETagConflictError
	}

	return "", time.Time{}, "", 0, exerr.New(exerr.TypeInternal, "Failed to upload remote database").Int("sc", resp.StatusCode).Build()
}