sql.go

/*
 * Copyright (c) 2018. Abstrium SAS <team (at) pydio.com>
 * This file is part of Pydio Cells.
 *
 * Pydio Cells is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Pydio Cells is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with Pydio Cells.  If not, see <http://www.gnu.org/licenses/>.
 *
 * The latest code can be found at <https://pydio.com>.
 */

package namespace

import (
	"github.com/gobuffalo/packr"
	"github.com/pydio/cells/common/config"
	"github.com/pydio/cells/common/proto/idm"
	"github.com/pydio/cells/common/service/proto"
	"github.com/pydio/cells/common/sql"
	"github.com/pydio/cells/common/sql/resources"
	migrate "github.com/rubenv/sql-migrate"
)

var (
	queries = map[string]string{
		"Add":    `insert into idm_usr_meta_ns (namespace, label, ns_order, indexable, definition) values (?,?,?,?,?)`,
		"Delete": `delete from idm_usr_meta_ns where namespace=?`,
		"List":   `select * from idm_usr_meta_ns order by ns_order asc`,
	}
)

// Impl of the Mysql interface
type sqlimpl struct {
	*sql.Handler

	*resources.ResourcesSQL
}

// Init handler for the SQL DAO
func (s *sqlimpl) Init(options config.Map) error {

	// super
	s.DAO.Init(options)

	// Preparing the resources
	s.ResourcesSQL = resources.NewDAO(s.Handler, "idm_usr_meta_ns.namespace").(*resources.ResourcesSQL)
	if err := s.ResourcesSQL.Init(options); err != nil {
		return err
	}

	// Doing the database migrations
	migrations := &sql.PackrMigrationSource{
		Box:         packr.NewBox("../../../idm/meta/namespace/migrations"),
		Dir:         s.Driver(),
		TablePrefix: s.Prefix() + "_ns",
	}

	_, err := migrate.Exec(s.DB(), s.Driver(), migrations, migrate.Up)
	if err != nil {
		return err
	}

	// Preparing the db statements
	if options.Bool("prepare", true) {
		for key, query := range queries {
			if err := s.Prepare(key, query); err != nil {
				return err
			}
		}
	}

	s.Add(&idm.UserMetaNamespace{
		Namespace: ReservedNamespaceBookmark,
		Label:     "Bookmarks",
		Policies: []*service.ResourcePolicy{
			{Action: service.ResourcePolicyAction_READ, Subject: "*", Effect: service.ResourcePolicy_allow},
			{Action: service.ResourcePolicyAction_WRITE, Subject: "*", Effect: service.ResourcePolicy_allow},
		},
	})

	return nil
}

func (dao *sqlimpl) Add(ns *idm.UserMetaNamespace) error {
	indexableValue := 0
	if ns.Indexable {
		indexableValue = 1
	}
	_, err := dao.GetStmt("Add").Exec(
		ns.Namespace,
		ns.Label,
		ns.Order,
		indexableValue,
		ns.JsonDefinition,
	)
	if err != nil {
		return err
	}
	if len(ns.Policies) > 0 {
		if err := dao.AddPolicies(false, ns.Namespace, ns.Policies); err != nil {
			return err
		}
	}
	return nil
}

func (dao *sqlimpl) Del(ns *idm.UserMetaNamespace) (e error) {

	if _, err := dao.GetStmt("Delete").Exec(ns.Namespace); err != nil {
		return err
	}

	if err := dao.DeletePoliciesForResource(ns.Namespace); err != nil {
		return err
	}

	return nil
}

func (dao *sqlimpl) List() (result map[string]*idm.UserMetaNamespace, err error) {

	res, err := dao.GetStmt("List").Query()
	if err != nil {
		return nil, err
	}
	defer res.Close()
	result = make(map[string]*idm.UserMetaNamespace)
	for res.Next() {
		ns := new(idm.UserMetaNamespace)
		var indexableValue int
		if err := res.Scan(&ns.Namespace, &ns.Label, &ns.Order, &indexableValue, &ns.JsonDefinition); err != nil {
			return nil, err
		}
		if indexableValue == 1 {
			ns.Indexable = true
		}
		// Add policies
		pol, err := dao.GetPoliciesForResource(ns.Namespace)
		if err != nil {
			return nil, err
		}
		ns.Policies = pol
		result[ns.Namespace] = ns
	}
	return
}