package test

import (
	ispec "github.com/opencontainers/image-spec/specs-go/v1"

	mTypes "zotregistry.io/zot/pkg/meta/types"
)

type RepoImage struct {
	Image
	Tag        string
	Statistics mTypes.DescriptorStatistics
}

type RepoMultiArchImage struct {
	MultiarchImage
	ImageStatistics map[string]mTypes.DescriptorStatistics
	Tag             string
}

type Repo struct {
	Name            string
	Images          []RepoImage
	MultiArchImages []RepoMultiArchImage
	IsBookmarked    bool
	IsStarred       bool
}

func GetMetadataForRepos(repos ...Repo) ([]mTypes.RepoMetadata, map[string]mTypes.ManifestMetadata,
	map[string]mTypes.IndexData,
) {
	var (
		reposMetadata       = []mTypes.RepoMetadata{}
		manifestMetadataMap = map[string]mTypes.ManifestMetadata{}
		indexDataMap        = map[string]mTypes.IndexData{}
	)

	for _, repo := range repos {
		repoMeta := mTypes.RepoMetadata{
			Name:         repo.Name,
			Tags:         map[string]mTypes.Descriptor{},
			Signatures:   map[string]mTypes.ManifestSignatures{},
			Statistics:   map[string]mTypes.DescriptorStatistics{},
			IsStarred:    repo.IsStarred,
			IsBookmarked: repo.IsBookmarked,
		}

		for _, image := range repo.Images {
			addImageMetaToMetaDB(image, repoMeta, manifestMetadataMap)
		}

		for _, multiArch := range repo.MultiArchImages {
			if multiArch.ImageStatistics == nil {
				multiArch.ImageStatistics = map[string]mTypes.DescriptorStatistics{}
			}

			repoMeta.Tags[multiArch.Tag] = mTypes.Descriptor{
				MediaType: ispec.MediaTypeImageIndex,
				Digest:    multiArch.DigestStr(),
			}

			repoMeta.Statistics[multiArch.DigestStr()] = multiArch.ImageStatistics[multiArch.DigestStr()]

			for _, image := range multiArch.Images {
				addImageMetaToMetaDB(RepoImage{
					Image:      image,
					Statistics: multiArch.ImageStatistics[image.DigestStr()],
				}, repoMeta, manifestMetadataMap)
			}

			indexDataMap[multiArch.indexDescriptor.Digest.String()] = mTypes.IndexData{
				IndexBlob: multiArch.indexDescriptor.Data,
			}
		}

		reposMetadata = append(reposMetadata, repoMeta)
	}

	return reposMetadata, manifestMetadataMap, indexDataMap
}

func addImageMetaToMetaDB(image RepoImage, repoMeta mTypes.RepoMetadata,
	manifestMetadataMap map[string]mTypes.ManifestMetadata,
) {
	if image.Tag != "" {
		repoMeta.Tags[image.Tag] = mTypes.Descriptor{
			MediaType: ispec.MediaTypeImageManifest,
			Digest:    image.DigestStr(),
		}
	}
	// here we can do many more checks about the images like check for referrers, signatures but it's not needed yet
	// I need just the tags for now and the fake signature.

	// This is done just to mark a manifest as signed in the resulted RepoMeta
	if image.Manifest.ArtifactType == TestFakeSignatureArtType && image.Manifest.Subject != nil {
		signedManifestDig := image.Manifest.Subject.Digest.String()
		repoMeta.Signatures[signedManifestDig] = mTypes.ManifestSignatures{
			"fakeSignature": []mTypes.SignatureInfo{{SignatureManifestDigest: image.ManifestDescriptor.Digest.String()}},
		}
	}

	repoMeta.Statistics[image.DigestStr()] = image.Statistics

	manifestMetadataMap[image.DigestStr()] = mTypes.ManifestMetadata{
		ManifestBlob:  image.ManifestDescriptor.Data,
		ConfigBlob:    image.ConfigDescriptor.Data,
		DownloadCount: image.Statistics.DownloadCount,
	}
}