pro_meta.go

// Copyright Project Harbor Authors
//
// 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 dao

import (
	"fmt"
	"strings"
	"time"

	"github.com/goharbor/harbor/src/common/models"
)

// AddProjectMetadata adds metadata for a project
func AddProjectMetadata(meta *models.ProjectMetadata) error {
	now := time.Now()
	sql := `insert into project_metadata
				(project_id, name, value, creation_time, update_time, deleted)
				 values (?, ?, ?, ?, ?, false)`
	_, err := GetOrmer().Raw(sql, meta.ProjectID, meta.Name, meta.Value,
		now, now).Exec()
	return err
}

// DeleteProjectMetadata deleted metadata of a project. If name is absent
// all metadatas will be deleted, otherwise only the metadatas specified
// by name will be deleted
func DeleteProjectMetadata(projectID int64, name ...string) error {
	params := make([]interface{}, 1)
	sql := `update project_metadata 
			set deleted = true 
			where project_id = ?`
	params = append(params, projectID)

	if len(name) > 0 {
		sql += fmt.Sprintf(` and name in ( %s )`, ParamPlaceholderForIn(len(name)))
		params = append(params, name)
	}

	_, err := GetOrmer().Raw(sql, params).Exec()
	return err
}

// UpdateProjectMetadata updates metadata of a project
func UpdateProjectMetadata(meta *models.ProjectMetadata) error {
	sql := `update project_metadata 
				set value = ?, update_time = ? 
				where project_id = ? and name = ? and deleted = false`
	_, err := GetOrmer().Raw(sql, meta.Value, time.Now(), meta.ProjectID,
		meta.Name).Exec()
	return err
}

// GetProjectMetadata returns the metadata of a project. If name is absent
// all metadatas will be returned, otherwise only the metadatas specified
// by name will be returned
func GetProjectMetadata(projectID int64, name ...string) ([]*models.ProjectMetadata, error) {
	proMetas := []*models.ProjectMetadata{}
	params := make([]interface{}, 1)

	sql := `select * from project_metadata 
				where project_id = ? and deleted = false`
	params = append(params, projectID)

	if len(name) > 0 {
		sql += fmt.Sprintf(` and name in ( %s )`, ParamPlaceholderForIn(len(name)))
		params = append(params, name)
	}

	_, err := GetOrmer().Raw(sql, params).QueryRows(&proMetas)
	return proMetas, err
}

// ParamPlaceholderForIn returns a string that contains placeholders for sql keyword "in"
// e.g. n=3, returns "?,?,?"
func ParamPlaceholderForIn(n int) string {
	placeholders := []string{}
	for i := 0; i < n; i++ {
		placeholders = append(placeholders, "?")
	}
	return strings.Join(placeholders, ",")
}

// ListProjectMetadata ...
func ListProjectMetadata(name, value string) ([]*models.ProjectMetadata, error) {
	sql := `select * from project_metadata 
				where name = ? and value = ? and deleted = false`
	metadatas := []*models.ProjectMetadata{}
	_, err := GetOrmer().Raw(sql, name, value).QueryRows(&metadatas)
	return metadatas, err
}