remove fields from other layers

This commit is contained in:
neha-Gupta1 2023-08-31 16:55:26 +05:30
parent 6ae0baff43
commit e0465585ce
5 changed files with 16 additions and 54 deletions

View File

@ -63,7 +63,7 @@ func CreateCollections(
// chanPager := handler.NewChannelsPager(qp.ProtectedResource.ID()) // chanPager := handler.NewChannelsPager(qp.ProtectedResource.ID())
// enumerating channels // enumerating channels
pager := handler.NewChannelsPager(qp.ProtectedResource.ID(), []string{}) pager := handler.NewChannelsPager(qp.ProtectedResource.ID())
// Loop through all pages returned by Graph API. // Loop through all pages returned by Graph API.
for { for {
@ -179,13 +179,11 @@ func populateCollections(
// } // }
// ictx = clues.Add(ictx, "previous_path", prevPath) // ictx = clues.Add(ictx, "previous_path", prevPath)
// TODO: Neha check this
var fields []string
// TODO: the handler should provide this implementation. // TODO: the handler should provide this implementation.
// TODO: if we doing this messages are items for us. // TODO: if we doing this messages are items for us.
items, err := collectItems( items, err := collectItems(
ctx, ctx,
bh.NewMessagePager(qp.ProtectedResource.ID(), ptr.Val(c.GetId()), fields)) bh.NewMessagePager(qp.ProtectedResource.ID(), ptr.Val(c.GetId())))
if err != nil { if err != nil {
el.AddRecoverable(ctx, clues.Stack(err)) el.AddRecoverable(ctx, clues.Stack(err))
continue continue

View File

@ -3,10 +3,11 @@ package groups
import ( import (
"context" "context"
"github.com/alcionai/corso/src/pkg/selectors"
"github.com/alcionai/corso/src/pkg/services/m365/api"
"github.com/microsoft/kiota-abstractions-go/serialization" "github.com/microsoft/kiota-abstractions-go/serialization"
"github.com/microsoftgraph/msgraph-sdk-go/models" "github.com/microsoftgraph/msgraph-sdk-go/models"
"github.com/alcionai/corso/src/pkg/selectors"
"github.com/alcionai/corso/src/pkg/services/m365/api"
) )
var _ BackupHandler = &groupBackupHandler{} var _ BackupHandler = &groupBackupHandler{}
@ -14,40 +15,40 @@ var _ BackupHandler = &groupBackupHandler{}
type groupBackupHandler struct { type groupBackupHandler struct {
ac api.Channels ac api.Channels
groupID string groupID string
scope selectors.GroupsScope
} }
func NewGroupBackupHandler(groupID string, ac api.Channels, scope selectors.GroupsScope) groupBackupHandler { func NewGroupBackupHandler(groupID string, ac api.Channels, scope selectors.GroupsScope) groupBackupHandler {
return groupBackupHandler{ return groupBackupHandler{
ac: ac, ac: ac,
groupID: groupID, groupID: groupID,
scope: scope,
} }
} }
func (gHandler groupBackupHandler) GetChannelByID(ctx context.Context, teamID, channelID string) (models.Channelable, error) { func (gHandler groupBackupHandler) GetChannelByID(
ctx context.Context,
teamID, channelID string,
) (models.Channelable, error) {
return gHandler.ac.Client.Channels().GetChannel(ctx, teamID, channelID) return gHandler.ac.Client.Channels().GetChannel(ctx, teamID, channelID)
} }
func (gHandler groupBackupHandler) NewChannelsPager( func (gHandler groupBackupHandler) NewChannelsPager(
teamID string, teamID string,
fields []string,
) api.ChannelDeltaEnumerator { ) api.ChannelDeltaEnumerator {
return gHandler.ac.NewChannelPager(teamID, fields) return gHandler.ac.NewChannelPager(teamID)
} }
func (gHandler groupBackupHandler) GetMessageByID( func (gHandler groupBackupHandler) GetMessageByID(
ctx context.Context, ctx context.Context,
teamID, channelID, itemID string, teamID, channelID, itemID string,
) (models.ChatMessageable, error) { ) (models.ChatMessageable, error) {
return gHandler.ac.GetMessageByID(ctx, teamID, channelID, itemID) chatMessage, _, err := gHandler.ac.GetMessage(ctx, teamID, channelID, itemID)
return chatMessage, err
} }
func (gHandler groupBackupHandler) NewMessagePager( func (gHandler groupBackupHandler) NewMessagePager(
teamID, channelID string, teamID, channelID string,
fields []string,
) api.ChannelMessageDeltaEnumerator { ) api.ChannelMessageDeltaEnumerator {
return gHandler.ac.NewMessagePager(teamID, channelID, fields) return gHandler.ac.NewMessagePager(teamID, channelID)
} }
func (gHandler groupBackupHandler) GetMessageReplies( func (gHandler groupBackupHandler) GetMessageReplies(

View File

@ -16,7 +16,6 @@ type BackupHandler interface {
) (models.Channelable, error) ) (models.Channelable, error)
NewChannelsPager( NewChannelsPager(
teamID string, teamID string,
fields []string,
) api.ChannelDeltaEnumerator ) api.ChannelDeltaEnumerator
GetMessageByID( GetMessageByID(
@ -25,7 +24,6 @@ type BackupHandler interface {
) (models.ChatMessageable, error) ) (models.ChatMessageable, error)
NewMessagePager( NewMessagePager(
teamID, channelID string, teamID, channelID string,
fields []string,
) api.ChannelMessageDeltaEnumerator ) api.ChannelMessageDeltaEnumerator
GetMessageReplies( GetMessageReplies(

View File

@ -107,7 +107,7 @@ func (c Channels) GetChannelByName(
func (c Channels) GetMessage( func (c Channels) GetMessage(
ctx context.Context, ctx context.Context,
teamID, channelID, itemID string, teamID, channelID, itemID string,
) (serialization.Parsable, *details.GroupsInfo, error) { ) (models.ChatMessageable, *details.GroupsInfo, error) {
var size int64 var size int64
message, err := c.Stable. message, err := c.Stable.
@ -126,27 +126,6 @@ func (c Channels) GetMessage(
return message, ChannelMessageInfo(message, size), nil return message, ChannelMessageInfo(message, size), nil
} }
// GetMessage retrieves a ChannelMessage item.
func (c Channels) GetMessageByID(
ctx context.Context,
teamID, channelID, itemID string,
) (models.ChatMessageable, error) {
message, err := c.Stable.
Client().
Teams().
ByTeamId(teamID).
Channels().
ByChannelId(channelID).
Messages().
ByChatMessageId(itemID).
Get(ctx, nil)
if err != nil {
return nil, graph.Stack(ctx, err)
}
return message, nil
}
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
// replies // replies
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------

View File

@ -32,17 +32,10 @@ type MessagePageCtrl struct {
func (c Channels) NewMessagePager( func (c Channels) NewMessagePager(
teamID, teamID,
channelID string, channelID string,
fields []string,
) *MessagePageCtrl { ) *MessagePageCtrl {
requestConfig := &teams.ItemChannelsItemMessagesDeltaRequestBuilderGetRequestConfiguration{
QueryParameters: &teams.ItemChannelsItemMessagesDeltaRequestBuilderGetQueryParameters{
Select: fields,
},
}
res := &MessagePageCtrl{ res := &MessagePageCtrl{
gs: c.Stable, gs: c.Stable,
options: requestConfig, options: nil,
builder: c.Stable. builder: c.Stable.
Client(). Client().
Teams(). Teams().
@ -100,17 +93,10 @@ type channelPageCtrl struct {
func (c Channels) NewChannelPager( func (c Channels) NewChannelPager(
teamID string, teamID string,
fields []string,
) *channelPageCtrl { ) *channelPageCtrl {
requestConfig := &teams.ItemChannelsRequestBuilderGetRequestConfiguration{
QueryParameters: &teams.ItemChannelsRequestBuilderGetQueryParameters{
Select: fields,
},
}
res := &channelPageCtrl{ res := &channelPageCtrl{
gs: c.Stable, gs: c.Stable,
options: requestConfig, options: nil,
builder: c.Stable. builder: c.Stable.
Client(). Client().
Teams(). Teams().