swirl/biz/registry.go

package biz

import (
	"context"
	"encoding/base64"
	"encoding/json"

	"github.com/cuigh/auxo/net/web"
	"github.com/cuigh/swirl/dao"
	"github.com/cuigh/swirl/model"
	"github.com/docker/docker/api/types"
)

type RegistryBiz interface {
	Search() ([]*model.Registry, error)
	Find(id string) (*model.Registry, error)
	GetAuth(url string) (auth string, err error)
	Delete(id, name string, user web.User) (err error)
	Create(registry *model.Registry, user web.User) (err error)
	Update(registry *model.Registry, user web.User) (err error)
}

func NewRegistry(d dao.Interface, eb EventBiz) RegistryBiz {
	return &registryBiz{d: d, eb: eb}
}

type registryBiz struct {
	d  dao.Interface
	eb EventBiz
}

func (b *registryBiz) Create(r *model.Registry, user web.User) (err error) {
	r.ID = createId()
	r.CreatedAt = now()
	r.UpdatedAt = r.CreatedAt
	r.CreatedBy = newOperator(user)
	r.UpdatedBy = r.CreatedBy

	err = b.d.RegistryCreate(context.TODO(), r)
	if err == nil {
		b.eb.CreateRegistry(EventActionCreate, r.ID, r.Name, user)
	}
	return
}

func (b *registryBiz) Update(r *model.Registry, user web.User) (err error) {
	r.UpdatedAt = now()
	r.UpdatedBy = newOperator(user)
	err = b.d.RegistryUpdate(context.TODO(), r)
	if err == nil {
		b.eb.CreateRegistry(EventActionUpdate, r.ID, r.Name, user)
	}
	return
}

func (b *registryBiz) Search() (registries []*model.Registry, err error) {
	registries, err = b.d.RegistryGetAll(context.TODO())
	if err == nil {
		for _, r := range registries {
			r.Password = ""
		}
	}
	return
}

func (b *registryBiz) Find(id string) (registry *model.Registry, err error) {
	registry, err = b.d.RegistryGet(context.TODO(), id)
	if err == nil {
		registry.Password = ""
	}
	return
}

func (b *registryBiz) GetAuth(url string) (auth string, err error) {
	var (
		r   *model.Registry
		buf []byte
	)
	if r, err = b.d.RegistryGetByURL(context.TODO(), url); err == nil && r != nil {
		cfg := &types.AuthConfig{
			ServerAddress: r.URL,
			Username:      r.Username,
			Password:      r.Password,
		}
		if buf, err = json.Marshal(cfg); err == nil {
			auth = base64.URLEncoding.EncodeToString(buf)
		}
	}
	return
}

func (b *registryBiz) Delete(id, name string, user web.User) (err error) {
	err = b.d.RegistryDelete(context.TODO(), id)
	if err == nil {
		b.eb.CreateRegistry(EventActionDelete, id, name, user)
	}
	return
}