56 lines
2.3 KiB
C#
56 lines
2.3 KiB
C#
// Copyright (c) 2026 Duplicati Inc. All rights reserved.
|
|
|
|
using Duplicati.Library.Common.IO;
|
|
using Duplicati.Library.Interface;
|
|
using Google.Apis.Drive.v3;
|
|
using System.Runtime.CompilerServices;
|
|
|
|
namespace Duplicati.Proprietary.GoogleWorkspace.SourceItems;
|
|
|
|
internal class SharedDrivesSourceEntry(SourceProvider provider, string parentPath, string? userId, DriveService driveService)
|
|
: MetaEntryBase(Util.AppendDirSeparator(SystemIO.IO_OS.PathCombine(parentPath, "Shared Drives")), null, null)
|
|
{
|
|
public override IAsyncEnumerable<ISourceProviderEntry> Enumerate(CancellationToken cancellationToken)
|
|
=> EnumerateSharedDrives(provider, this.Path, userId, driveService, cancellationToken);
|
|
|
|
public static async IAsyncEnumerable<ISourceProviderEntry> EnumerateSharedDrives(SourceProvider provider, string path, string? userId, DriveService driveService, [EnumeratorCancellation] CancellationToken cancellationToken)
|
|
{
|
|
var request = driveService.Drives.List();
|
|
|
|
string? nextPageToken = null;
|
|
do
|
|
{
|
|
if (cancellationToken.IsCancellationRequested) yield break;
|
|
request.PageToken = nextPageToken;
|
|
var drives = await request.ExecuteAsync(cancellationToken);
|
|
|
|
if (drives.Drives != null)
|
|
{
|
|
foreach (var drive in drives.Drives)
|
|
{
|
|
if (cancellationToken.IsCancellationRequested) yield break;
|
|
|
|
if (!provider.LicenseApprovedForEntry(path, GoogleRootType.SharedDrives, drive.Id))
|
|
yield break;
|
|
|
|
yield return new SharedDriveSourceEntry(path, userId!, drive, driveService);
|
|
}
|
|
}
|
|
nextPageToken = drives.NextPageToken;
|
|
} while (!string.IsNullOrEmpty(nextPageToken));
|
|
}
|
|
|
|
public override Task<Dictionary<string, string?>> GetMinorMetadata(CancellationToken cancellationToken)
|
|
{
|
|
return Task.FromResult(new Dictionary<string, string?>
|
|
{
|
|
{ "gsuite:v", "1" },
|
|
{ "gsuite:Type", SourceItemType.SharedDrives.ToString() },
|
|
{ "gsuite:Name", "Shared Drives" },
|
|
{ "gsuite:Id", "Shared Drives" }
|
|
}
|
|
.Where(kv => !string.IsNullOrEmpty(kv.Value))
|
|
.ToDictionary(kv => kv.Key, kv => kv.Value));
|
|
}
|
|
}
|