// Copyright 2019 Yunion
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package volume_mount

import (
	"context"

	"yunion.io/x/jsonutils"
	"yunion.io/x/pkg/errors"

	"yunion.io/x/onecloud/pkg/apis"
	imageapi "yunion.io/x/onecloud/pkg/apis/image"
	"yunion.io/x/onecloud/pkg/compute/options"
	"yunion.io/x/onecloud/pkg/httperrors"
	"yunion.io/x/onecloud/pkg/mcclient"
	"yunion.io/x/onecloud/pkg/mcclient/auth"
	imagemod "yunion.io/x/onecloud/pkg/mcclient/modules/image"
)

type povImage struct {
}

func newDiskPostOverlayImage() iDiskPostOverlay {
	return &povImage{}
}

func (p povImage) validateData(ctx context.Context, userCred mcclient.TokenCredential, pov *apis.ContainerVolumeMountDiskPostOverlay) error {
	img := pov.Image
	if img.Id == "" {
		return httperrors.NewMissingParameterError("image id")
	}
	s := auth.GetAdminSession(ctx, options.Options.Region)
	obj, err := imagemod.Images.Get(s, img.Id, nil)
	if err != nil {
		return errors.Wrapf(err, "Get image by id %s", img.Id)
	}
	imgObj := new(imageapi.ImageDetails)
	if err := obj.Unmarshal(imgObj); err != nil {
		return errors.Wrap(err, "unmarshal image details")
	}
	pov.Image.Id = imgObj.Id
	props := imgObj.Properties
	usedByStr, ok := props[imageapi.IMAGE_USED_BY_POST_OVERLAY]
	if !ok {
		return errors.Wrapf(err, "Get %s", imageapi.IMAGE_USED_BY_POST_OVERLAY)
	}
	if usedByStr != "true" {
		return errors.Errorf("image isn't used by post overlay")
	}
	pathMapStr := props[imageapi.IMAGE_INTERNAL_PATH_MAP]
	pathMapObj, err := jsonutils.ParseString(pathMapStr)
	if err != nil {
		return errors.Wrapf(err, "json parse path_map: %s", pathMapStr)
	}
	pathMap := make(map[string]string)
	if err := pathMapObj.Unmarshal(pathMap); err != nil {
		return errors.Wrapf(err, "unmarshal pathMapObj")
	}
	if len(pov.Image.PathMap) == 0 {
		pov.Image.PathMap = pathMap
	}
	if len(pov.Image.HostLowerMap) != 0 {
		for hostPath, _ := range pov.Image.HostLowerMap {
			_, ok := pov.Image.PathMap[hostPath]
			if !ok {
				return httperrors.NewNotFoundError("host_path %s of host_lower_map doesn't found in path_map", hostPath)
			}
		}
	}
	return nil
}

func (p povImage) getContainerTargetDirs(ov *apis.ContainerVolumeMountDiskPostOverlay) []string {
	pathMap := ov.Image.PathMap
	ctrPaths := []string{}
	for _, ctrPath := range pathMap {
		ctrPaths = append(ctrPaths, ctrPath)
	}
	return ctrPaths
}