package cli

import (
	"context"
	"fmt"
	"sync/atomic"

	"github.com/pkg/errors"

	"github.com/kopia/kopia/repo"
	"github.com/kopia/kopia/repo/content"
)

var (
	contentListCommand        = contentCommands.Command("list", "List contents").Alias("ls")
	contentListLong           = contentListCommand.Flag("long", "Long output").Short('l').Bool()
	contentListPrefix         = contentListCommand.Flag("prefix", "Prefix").String()
	contentListIncludeDeleted = contentListCommand.Flag("deleted", "Include deleted content").Bool()
	contentListDeletedOnly    = contentListCommand.Flag("deleted-only", "Only show deleted content").Bool()
	contentListSummary        = contentListCommand.Flag("summary", "Summarize the list").Short('s').Bool()
	contentListHuman          = contentListCommand.Flag("human", "Human-readable output").Short('h').Bool()
)

func runContentListCommand(ctx context.Context, rep *repo.Repository) error {
	var count int32

	var totalSize int64

	err := rep.Content.IterateContents(
		ctx,
		content.IterateOptions{
			Prefix:         content.ID(*contentListPrefix),
			IncludeDeleted: *contentListIncludeDeleted || *contentListDeletedOnly,
		},
		func(b content.Info) error {
			if *contentListDeletedOnly && !b.Deleted {
				return nil
			}
			atomic.AddInt64(&totalSize, int64(b.Length))
			atomic.AddInt32(&count, 1)
			if *contentListLong {
				optionalDeleted := ""
				if b.Deleted {
					optionalDeleted = " (deleted)"
				}
				fmt.Printf("%v %v %v %v+%v%v\n",
					b.ID,
					formatTimestamp(b.Timestamp()),
					b.PackBlobID,
					b.PackOffset,
					maybeHumanReadableBytes(*contentListHuman, int64(b.Length)),
					optionalDeleted)
			} else {
				fmt.Printf("%v\n", b.ID)
			}

			return nil
		})

	if err != nil {
		return errors.Wrap(err, "error iterating")
	}

	if *contentListSummary {
		fmt.Printf("Total: %v contents, %v total size\n",
			maybeHumanReadableCount(*contentListHuman, int64(count)),
			maybeHumanReadableBytes(*contentListHuman, totalSize))
	}

	return nil
}

func init() {
	contentListCommand.Action(repositoryAction(runContentListCommand))
}