// Copyright (c) 2020 Gitpod GmbH. All rights reserved.
// Licensed under the GNU Affero General Public License (AGPL).
// See License-AGPL.txt in the project root for license information.

package api

import (
	"encoding/base64"
	"fmt"

	proto "github.com/golang/protobuf/proto"
)

// ToBase64 marshals the image spec using protobuf and encodes it in base64
func (spec *ImageSpec) ToBase64() (res string, err error) {
	if spec == nil {
		return
	}

	rspec, err := proto.Marshal(spec)
	if err != nil {
		return "", fmt.Errorf("cannot marshal image spec: %w", err)
	}

	return base64.StdEncoding.EncodeToString(rspec), nil
}

// ImageSpecFromBase64 decodes an image specification from a base64 encoded protobuf message
func ImageSpecFromBase64(input string) (res *ImageSpec, err error) {
	if len(input) == 0 {
		return
	}

	specPB, err := base64.StdEncoding.DecodeString(input)
	if err != nil {
		return nil, fmt.Errorf("cannot decode image spec: %w", err)
	}

	var spec ImageSpec
	err = proto.Unmarshal(specPB, &spec)
	if err != nil {
		return nil, fmt.Errorf("cannot unmarshal image spec: %w", err)
	}

	return &spec, nil
}