feat: add checksum header for asset uploads

Author: simulot

immich/call.go

@@ -12,6 +12,7 @@ import (
 	"sync/atomic"
 	"time"
 
+	"github.com/simulot/immich-go/internal/assets"
 	"github.com/simulot/immich-go/internal/fshelper"
 )
 
@@ -155,8 +156,10 @@ func (sc *serverCall) request(
 	}
 	opts = append(opts, setAPIKey())
 	for _, opt := range opts {
-		if sc.joinError(opt(sc, req)) != nil {
-			return nil
+		if opt != nil {
+			if sc.joinError(opt(sc, req)) != nil {
+				return nil
+			}
 		}
 	}
 	return req
@@ -276,7 +279,9 @@ func (sc *serverCall) do(fnRequest requestFunction, opts ...serverResponseOption
 
 	// We have a success
 	for _, opt := range opts {
-		_ = sc.joinError(opt(sc, resp))
+		if opt != nil {
+			_ = sc.joinError(opt(sc, resp))
+		}
 	}
 	if sc.err != nil {
 		return sc.Err(req, resp, nil)
@@ -293,6 +298,16 @@ func setBody(body io.ReadCloser) serverRequestOption {
 	}
 }
 
+func setImmichChecksum(a *assets.Asset) serverRequestOption {
+	if a.Checksum == "" {
+		return nil
+	}
+	return func(sc *serverCall, req *http.Request) error {
+		req.Header.Set("x-immich-checksum", a.Checksum)
+		return nil
+	}
+}
+
 func setAcceptJSON() serverRequestOption {
 	return func(sc *serverCall, req *http.Request) error {
 		req.Header.Add("Accept", "application/json")

immich/upload.go

@@ -98,17 +98,18 @@ func (ic *ImmichClient) uploadAsset(ctx context.Context, la *assets.Asset, endPo
 	switch endPoint {
 	case EndPointAssetUpload:
 		errCall = ic.newServerCall(ctx, EndPointAssetUpload).
-			do(postRequest("/assets", m.FormDataContentType(), setContextValue(callValues), setAcceptJSON(), setBody(body)), responseJSON(&ar))
+			do(postRequest("/assets", m.FormDataContentType(), setContextValue(callValues), setAcceptJSON(), setImmichChecksum(la), setBody(body)), responseJSON(&ar))
 	case EndPointAssetReplace:
 		errCall = ic.newServerCall(ctx, EndPointAssetReplace).
-			do(putRequest("/assets/"+replaceID+"/original", setContextValue(callValues), setAcceptJSON(), setContentType(m.FormDataContentType()), setBody(body)), responseJSON(&ar))
+			do(putRequest("/assets/"+replaceID+"/original", setContextValue(callValues), setAcceptJSON(), setImmichChecksum(la), setContentType(m.FormDataContentType()), setBody(body)), responseJSON(&ar))
 	}
 	err = errors.Join(err, errCall)
 	return ar, err
 }
 
 func (ic *ImmichClient) prepareCallValues(la *assets.Asset, s fs.FileInfo, ext, mtype string) map[string]string {
 	callValues := map[string]string{}
+
 	callValues["deviceAssetId"] = fmt.Sprintf("%s-%d", path.Base(la.OriginalFileName), s.Size())
 	callValues["deviceId"] = ic.DeviceUUID
 	callValues["assetType"] = mtype