larryh/corso
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update to /connector/sharepoint/restore.go

Browse Source 
Issue #2169: Adds logic to append destination name to file name. Title is not altered.
This commit is contained in:
Danny Adams 2023-01-21 23:36:08 -05:00
parent d16b13bd64
commit 9ea6e2921b
1 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/internal/connector/sharepoint/restore.go
View File

@ -7,7 +7,6 @@ import (
"runtime/trace" "runtime/trace"
"github.com/microsoftgraph/msgraph-beta-sdk-go/models" "github.com/microsoftgraph/msgraph-beta-sdk-go/models"
"github.com/microsoftgraph/msgraph-beta-sdk-go/sites"
"github.com/pkg/errors" "github.com/pkg/errors"
"github.com/alcionai/corso/src/internal/connector/graph" "github.com/alcionai/corso/src/internal/connector/graph"
@ -284,7 +283,6 @@ func restoreSitePage(
var ( var (
dii = details.ItemInfo{} dii = details.ItemInfo{}
pageName = itemData.UUID() pageName = itemData.UUID()
pageURL string
) )
byteArray, err := io.ReadAll(itemData.ToReader()) byteArray, err := io.ReadAll(itemData.ToReader())
@ -298,8 +296,13 @@ func restoreSitePage(
return dii, errors.Wrapf(err, "failed to create Page object %s", pageName) return dii, errors.Wrapf(err, "failed to create Page object %s", pageName)
} }
// POST the Publish newName := fmt.Sprintf("%s_%s", destName, *page.GetName())
// https://learn.microsoft.com/en-us/graph/api/sitepage-create?view=graph-rest-beta page.SetName(&newName)
fmt.Printf("Page Name: %s\n", *page.GetName())
// Restore is a 2-Step Process in Graph API
// 1. Create the Page on the site
// 2. Publish the site
// See: https://learn.microsoft.com/en-us/graph/api/sitepage-create?view=graph-rest-beta
restoredPage, err := service.Client().SitesById(siteID).Pages().Post(ctx, page, nil) restoredPage, err := service.Client().SitesById(siteID).Pages().Post(ctx, page, nil)
if err != nil { if err != nil {
sendErr := support.ConnectorStackErrorTraceWrap( sendErr := support.ConnectorStackErrorTraceWrap(
@ -316,18 +319,19 @@ func restoreSitePage(
return dii, fmt.Errorf("created page %s did not return webURL from API", *restoredPage.GetId()) return dii, fmt.Errorf("created page %s did not return webURL from API", *restoredPage.GetId())
} }
pageURL = *restoredPage.GetWebUrl() pageID := *restoredPage.GetId()
publishRequest := sites.NewItemPagesItemPublishRequestBuilder(pageURL, service.Adapter())
err = publishRequest.Post(ctx, nil) err = service.Client().SitesById(siteID).PagesById(pageID).Publish().Post(ctx, nil)
if err != nil { if err != nil {
return dii, support.ConnectorStackErrorTraceWrap( return dii, support.ConnectorStackErrorTraceWrap(
err, err,
"unable to publish page ID: "+*restoredPage.GetId()+" API Error Details", "unable to publish page ID: "+pageID+" API Error Details",
) )
} }
dii.SharePoint = sharePointPageInfo(restoredPage, int64(len(byteArray))) dii.SharePoint = sharePointPageInfo(restoredPage, int64(len(byteArray)))
// Storing new pageID in unused field.
dii.SharePoint.ParentPath = pageID
return dii, nil return dii, nil
} }