transform_create.go

// Copyright (C) 2019 Storj Labs, Inc.
// Copyright (C) 2017 Space Monkey, Inc.
// See LICENSE for copying information.

package xform

import (
	"storj.io/dbx/ast"
	"storj.io/dbx/errutil"
	"storj.io/dbx/ir"
)

func transformCreate(lookup *lookup, ast_cre *ast.Create) (
	cre *ir.Create, err error) {

	model, err := lookup.FindModel(ast_cre.Model)
	if err != nil {
		return nil, err
	}

	cre = &ir.Create{
		Model:    model,
		Raw:      ast_cre.Raw.Get(),
		NoReturn: ast_cre.NoReturn.Get(),
		Replace:  ast_cre.Replace.Get(),
		Suffix:   transformSuffix(ast_cre.Suffix),
	}
	if cre.Suffix == nil {
		cre.Suffix = DefaultCreateSuffix(cre)
	}

	if cre.Replace &&
		!cre.Raw &&
		cre.Model.BasicPrimaryKey() != nil &&
		len(cre.Model.Unique) == 0 {

		return nil, errutil.New(ast_cre.Replace.Pos,
			`non-raw replace with basic primary key and no unique `+
				`constraints guarantees no replace can happen`)
	}

	return cre, nil
}