// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
// snippet-start:[s3.go.restore_object]
package main

// snippet-start:[s3.go.restore_object.imports]
import (
	"flag"
	"fmt"

	"github.com/aws/aws-sdk-go/aws/session"
	"github.com/aws/aws-sdk-go/service/s3"
	"github.com/aws/aws-sdk-go/service/s3/s3iface"
)

// snippet-end:[s3.go.restore_object.imports]

// RestoreItem restores an item to a bucket for a number of days
// Inputs:
//
//	sess is the current session, which provides configuration for the SDK's service clients
//	bucket is the name of the bucket
//	item is the name of the bucket object to delete
//
// Output:
//
//	If success, nil
//	Otherwise, an error from the call to DeleteObject or WaitUntilObjectNotExists
func RestoreItem(svc s3iface.S3API, bucket *string, item *string, days *int64) error {
	// snippet-start:[s3.go.restore_object.call]
	_, err := svc.RestoreObject(&s3.RestoreObjectInput{
		Bucket: bucket,
		Key:    item,
		RestoreRequest: &s3.RestoreRequest{
			Days: days,
		},
	})
	// snippet-end:[s3.go.restore_object.call]
	if err != nil {
		return err
	}

	return nil
}

func main() {
	// snippet-start:[s3.go.restore_object.args]
	bucket := flag.String("b", "", "The bucket to which the object is restored")
	item := flag.String("i", "", "The object to restore")
	days := flag.Int64("d", 30, "The number of days to restore it from Glacier")
	flag.Parse()

	if *bucket == "" || *item == "" {
		fmt.Println("You must supply the bucket (-b BUCKET), and item to restore (-i ITEM")
		return
	}
	// snippet-end:[s3.go.restore_object.args]

	// snippet-start:[s3.go.restore_object.session]
	sess := session.Must(session.NewSessionWithOptions(session.Options{
		SharedConfigState: session.SharedConfigEnable,
	}))

	svc := s3.New(sess)
	// snippet-end:[s3.go.restore_object.session]

	err := RestoreItem(svc, bucket, item, days)
	if err != nil {
		fmt.Println("Got an error restoring the item:")
		fmt.Println(err)
		return
	}

	// snippet-start:[s3.go.restore_object.print]
	fmt.Println("Restored " + *item + " to " + *bucket)
	// snippet-end:[s3.go.restore_object.print]
}

// snippet-end:[s3.go.restore_object]
