datasync

type Agent struct {
	CreationTime time.Time
	Tags         map[string]string
	AgentArn     string
	Name         string
	Status       string
	EndpointType string
}

type AgentListEntry struct {
	AgentArn string
	Name     string
	Status   string
}

type Ec2Config struct {
	SubnetArn         string
	SecurityGroupArns []string
}

type FsxNfsProtocol struct {
	MountOptions *MountOptions
}

type FsxProtocol struct {
	NFS *FsxNfsProtocol
	SMB *FsxSmbProtocol
}

type FsxSmbProtocol struct {
	MountOptions *MountOptions
	Domain       string
	Password     string
	User         string
}

type Handler struct {
	Backend StorageBackend
	// contains filtered or unexported fields
}

type HdfsNameNode struct {
	Hostname string
	Port     int32
}

type InMemoryBackend struct {
	// contains filtered or unexported fields
}

type Location struct {
	CreationTime time.Time
	LocationArn  string
	LocationURI  string
}

type LocationAzureBlob struct {
	CreationTime     time.Time
	SasConfiguration *SasConfiguration
	LocationArn      string
	LocationURI      string
	ContainerURL     string
	BlobType         string
	AccessTier       string
	Subdirectory     string
	AgentArns        []string
}

type LocationEfs struct {
	CreationTime            time.Time
	Ec2Config               *Ec2Config
	LocationArn             string
	LocationURI             string
	EfsFilesystemArn        string
	AccessPointArn          string
	FileSystemAccessRoleArn string
	InTransitEncryption     string
	Subdirectory            string
}

type LocationFsxLustre struct {
	CreationTime      time.Time
	LocationArn       string
	LocationURI       string
	FsxFilesystemArn  string
	Subdirectory      string
	SecurityGroupArns []string
}

type LocationFsxOntap struct {
	CreationTime             time.Time
	Protocol                 *FsxProtocol
	LocationArn              string
	LocationURI              string
	StorageVirtualMachineArn string
	Subdirectory             string
	SecurityGroupArns        []string
}

type LocationFsxOpenZfs struct {
	CreationTime      time.Time
	Protocol          *FsxProtocol
	LocationArn       string
	LocationURI       string
	FsxFilesystemArn  string
	Subdirectory      string
	SecurityGroupArns []string
}

type LocationFsxWindows struct {
	CreationTime      time.Time
	LocationArn       string
	LocationURI       string
	FsxFilesystemArn  string
	Domain            string
	User              string
	Subdirectory      string
	SecurityGroupArns []string
}

type LocationHdfs struct {
	CreationTime       time.Time
	QopConfiguration   *QopConfiguration
	KmsKeyProviderURI  string
	LocationArn        string
	LocationURI        string
	AuthenticationType string
	SimpleUser         string
	KerberosPrincipal  string
	Subdirectory       string
	AgentArns          []string
	NameNodes          []HdfsNameNode
	BlockSize          int64
	ReplicationFactor  int32
}

type LocationListEntry struct {
	CreationTime time.Time
	LocationArn  string
	LocationURI  string
}

type LocationNfs struct {
	CreationTime   time.Time
	MountOptions   *MountOptions
	LocationArn    string
	LocationURI    string
	ServerHostname string
	Subdirectory   string
	AgentArns      []string
}

type LocationObjectStorage struct {
	CreationTime   time.Time
	LocationArn    string
	LocationURI    string
	ServerHostname string
	ServerProtocol string
	BucketName     string
	AccessKey      string
	Subdirectory   string
	AgentArns      []string
	ServerPort     int32
}

type LocationS3 struct {
	CreationTime   time.Time
	S3Config       S3Config
	LocationArn    string
	LocationURI    string
	S3BucketArn    string
	Subdirectory   string
	S3StorageClass string
}

type LocationSmb struct {
	CreationTime   time.Time
	MountOptions   *MountOptions
	LocationArn    string
	LocationURI    string
	ServerHostname string
	Domain         string
	User           string
	Subdirectory   string
	AgentArns      []string
}

type MountOptions struct {
	Version string
}

type Provider struct{}

type QopConfiguration struct {
	DataTransferProtection string
	RPCProtection          string
}

type S3Config struct {
	BucketAccessRoleArn string
}

type SasConfiguration struct {
	Token string
}

type StorageBackend interface {
	// Agent operations
	CreateAgent(name string, activationKey string, tags map[string]string) (*Agent, error)
	DescribeAgent(agentArn string) (*Agent, error)
	UpdateAgent(agentArn, name string) error
	DeleteAgent(agentArn string) error
	ListAgents(maxResults int32, nextToken string) ([]*AgentListEntry, string, error)

	// Location operations (S3)
	CreateLocationS3(
		subdirectory, s3BucketArn, s3StorageClass string,
		s3Config S3Config,
		tags map[string]string,
	) (*Location, error)
	DescribeLocationS3(locationArn string) (*LocationS3, error)
	DeleteLocation(locationArn string) error
	ListLocations(maxResults int32, nextToken string) ([]*LocationListEntry, string, error)

	// Task operations
	CreateTask(
		sourceLocationArn, destinationLocationArn, name, cloudWatchLogGroupArn string,
		tags map[string]string,
	) (*Task, error)
	DescribeTask(taskArn string) (*Task, error)
	UpdateTask(taskArn, name, cloudWatchLogGroupArn string) error
	DeleteTask(taskArn string) error
	ListTasks(maxResults int32, nextToken string) ([]*TaskListEntry, string, error)

	// Task execution operations
	StartTaskExecution(taskArn string) (*TaskExecution, error)
	CancelTaskExecution(taskExecutionArn string) error
	DescribeTaskExecution(taskExecutionArn string) (*TaskExecution, error)
	ListTaskExecutions(taskArn string, maxResults int32, nextToken string) ([]*TaskExecutionListEntry, string, error)

	// Location operations (S3 update)
	UpdateLocationS3(locationArn, subdirectory, s3StorageClass string, s3Config S3Config) error

	// Task execution update
	UpdateTaskExecution(taskExecutionArn string, options map[string]any) error

	// Location operations (Azure Blob)
	CreateLocationAzureBlob(
		containerURL, subdirectory, blobType, accessTier string,
		sasConfig *SasConfiguration,
		agentArns []string,
		tags map[string]string,
	) (*Location, error)
	DescribeLocationAzureBlob(locationArn string) (*LocationAzureBlob, error)
	UpdateLocationAzureBlob(
		locationArn, containerURL, subdirectory, blobType, accessTier string,
		sasConfig *SasConfiguration,
		agentArns []string,
	) error

	// Location operations (EFS)
	CreateLocationEfs(
		efsFilesystemArn, subdirectory, accessPointArn, fileSystemAccessRoleArn, inTransitEncryption string,
		ec2Config *Ec2Config,
		tags map[string]string,
	) (*Location, error)
	DescribeLocationEfs(locationArn string) (*LocationEfs, error)
	UpdateLocationEfs(
		locationArn, subdirectory, accessPointArn, fileSystemAccessRoleArn, inTransitEncryption string,
		ec2Config *Ec2Config,
	) error

	// Location operations (FSx Lustre)
	CreateLocationFsxLustre(
		fsxFilesystemArn, subdirectory string,
		securityGroupArns []string,
		tags map[string]string,
	) (*Location, error)
	DescribeLocationFsxLustre(locationArn string) (*LocationFsxLustre, error)
	UpdateLocationFsxLustre(locationArn, subdirectory string) error

	// Location operations (FSx ONTAP)
	CreateLocationFsxOntap(
		storageVirtualMachineArn, subdirectory string,
		protocol *FsxProtocol,
		securityGroupArns []string,
		tags map[string]string,
	) (*Location, error)
	DescribeLocationFsxOntap(locationArn string) (*LocationFsxOntap, error)
	UpdateLocationFsxOntap(locationArn, subdirectory string, protocol *FsxProtocol) error

	// Location operations (FSx OpenZFS)
	CreateLocationFsxOpenZfs(
		fsxFilesystemArn, subdirectory string,
		protocol *FsxProtocol,
		securityGroupArns []string,
		tags map[string]string,
	) (*Location, error)
	DescribeLocationFsxOpenZfs(locationArn string) (*LocationFsxOpenZfs, error)
	UpdateLocationFsxOpenZfs(locationArn, subdirectory string, protocol *FsxProtocol) error

	// Location operations (FSx Windows)
	CreateLocationFsxWindows(
		fsxFilesystemArn, subdirectory, domain, user, password string,
		securityGroupArns []string,
		tags map[string]string,
	) (*Location, error)
	DescribeLocationFsxWindows(locationArn string) (*LocationFsxWindows, error)
	UpdateLocationFsxWindows(locationArn, subdirectory, domain, user, password string) error

	// Location operations (HDFS)
	CreateLocationHdfs(
		subdirectory, authenticationType, simpleUser string,
		kerberosPrincipal, kerberosKeytab, kerberosKrb5Conf, kmsKeyProviderURI string,
		nameNodes []HdfsNameNode,
		blockSize int64,
		replicationFactor int32,
		qopConfig *QopConfiguration,
		agentArns []string,
		tags map[string]string,
	) (*Location, error)
	DescribeLocationHdfs(locationArn string) (*LocationHdfs, error)
	UpdateLocationHdfs(
		locationArn, subdirectory, authenticationType, simpleUser string,
		kerberosPrincipal, kerberosKeytab, kerberosKrb5Conf, kmsKeyProviderURI string,
		nameNodes []HdfsNameNode,
		blockSize int64,
		replicationFactor int32,
		qopConfig *QopConfiguration,
		agentArns []string,
	) error

	// Location operations (NFS)
	CreateLocationNfs(
		serverHostname, subdirectory string,
		mountOptions *MountOptions,
		agentArns []string,
		tags map[string]string,
	) (*Location, error)
	DescribeLocationNfs(locationArn string) (*LocationNfs, error)
	UpdateLocationNfs(locationArn, subdirectory string, mountOptions *MountOptions, agentArns []string) error

	// Location operations (Object Storage)
	CreateLocationObjectStorage(
		serverHostname, serverProtocol, bucketName, subdirectory, accessKey, secretKey string,
		serverPort int32,
		agentArns []string,
		tags map[string]string,
	) (*Location, error)
	DescribeLocationObjectStorage(locationArn string) (*LocationObjectStorage, error)
	UpdateLocationObjectStorage(
		locationArn, serverProtocol, subdirectory, accessKey, secretKey string,
		serverPort int32,
		agentArns []string,
	) error

	// Location operations (SMB)
	CreateLocationSmb(
		serverHostname, subdirectory, domain, user, password string,
		mountOptions *MountOptions,
		agentArns []string,
		tags map[string]string,
	) (*Location, error)
	DescribeLocationSmb(locationArn string) (*LocationSmb, error)
	UpdateLocationSmb(
		locationArn, subdirectory, domain, user, password string,
		mountOptions *MountOptions,
		agentArns []string,
	) error

	// Tag operations
	TagResource(resourceArn string, tags map[string]string) error
	UntagResource(resourceArn string, keys []string) error
	ListTagsForResource(resourceArn string, maxResults int32, nextToken string) (map[string]string, string, error)

	AccountID() string
	Region() string
	Reset()
	Snapshot(ctx context.Context) []byte
	Restore(ctx context.Context, data []byte) error
}

type Task struct {
	CreationTime            time.Time
	Tags                    map[string]string
	TaskArn                 string
	Name                    string
	Status                  string
	SourceLocationArn       string
	DestinationLocationArn  string
	CloudWatchLogGroupArn   string
	CurrentTaskExecutionArn string
}

type TaskExecution struct {
	StartTime                time.Time
	Options                  map[string]any
	TaskExecutionArn         string
	Status                   string
	EstimatedFilesToTransfer int64
	EstimatedBytesToTransfer int64
	FilesTransferred         int64
	BytesTransferred         int64
}

type TaskExecutionListEntry struct {
	TaskExecutionArn string
	Status           string
}

type TaskListEntry struct {
	TaskArn string
	Name    string
	Status  string
}