track item size in onedrive restores (#1159)

## Description

Hacks in the bytes written as the iteminfo size during onedrive item restores.  Would use the drive itemable, but the size property isn't instantiated locally during creation, and cannot be mutated afterwards.

## Type of change

- [x] 🐛 Bugfix

## Issue(s)

* #1113

## Test Plan

- [x] 💪 Manual

src/internal/connector/onedrive/drive.go

@@ -125,7 +125,10 @@ func getFolder(ctx context.Context, service graph.Service, driveID string, paren
 }
 
 // Create a new item in the specified folder
-func createItem(ctx context.Context, service graph.Service, driveID string, parentFolderID string,
+func createItem(
+	ctx context.Context,
+	service graph.Service,
+	driveID, parentFolderID string,
 	newItem models.DriveItemable,
 ) (models.DriveItemable, error) {
 	// Graph SDK doesn't yet provide a POST method for `/children` so we set the `rawUrl` ourselves as recommended

src/internal/connector/onedrive/item.go

@@ -60,24 +60,31 @@ func driveItemReader(
 		return nil, nil, errors.Wrapf(err, "failed to download file from %s", *downloadURL)
 	}
 
-	return driveItemInfo(item), resp.Body, nil
+	return driveItemInfo(item, *item.GetSize()), resp.Body, nil
 }
 
 // driveItemInfo will populate a details.OneDriveInfo struct
-// with properties from the drive item.
-func driveItemInfo(di models.DriveItemable) *details.OneDriveInfo {
+// with properties from the drive item.  ItemSize is specified
+// separately for restore processes because the local itemable
+// doesn't have its size value updated as a side effect of creation,
+// and kiota drops any SetSize update.
+func driveItemInfo(di models.DriveItemable, itemSize int64) *details.OneDriveInfo {
 	return &details.OneDriveInfo{
 		ItemType: details.OneDriveItem,
 		ItemName: *di.GetName(),
 		Created:  *di.GetCreatedDateTime(),
 		Modified: *di.GetLastModifiedDateTime(),
-		Size:     *di.GetSize(),
+		Size:     itemSize,
 	}
 }
 
 // driveItemWriter is used to initialize and return an io.Writer to upload data for the specified item
 // It does so by creating an upload session and using that URL to initialize an `itemWriter`
-func driveItemWriter(ctx context.Context, service graph.Service, driveID, itemID string, itemSize int64,
+func driveItemWriter(
+	ctx context.Context,
+	service graph.Service,
+	driveID, itemID string,
+	itemSize int64,
 ) (io.Writer, error) {
 	// TODO: @vkamra verify if var session is the desired input
 	session := msup.NewCreateUploadSessionPostRequestBody()

src/internal/connector/onedrive/restore.go

@@ -251,10 +251,10 @@ func restoreItem(
 	}
 
 	// Upload the stream data
-	_, err = io.CopyBuffer(w, itemData.ToReader(), copyBuffer)
+	written, err := io.CopyBuffer(w, itemData.ToReader(), copyBuffer)
 	if err != nil {
 		return nil, errors.Wrapf(err, "failed to upload data: item %s", itemName)
 	}
 
-	return driveItemInfo(newItem), nil
+	return driveItemInfo(newItem, written), nil
 }