surrealpatch/util/item/each.go
2016-09-14 21:46:23 +01:00

423 lines
9.3 KiB
Go

// Copyright © 2016 Abcum Ltd
//
// 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 item
import (
"fmt"
"regexp"
"github.com/robertkrimen/otto"
"github.com/yuin/gopher-lua"
"github.com/abcum/surreal/cnf"
"github.com/abcum/surreal/sql"
"github.com/abcum/surreal/util/conv"
"github.com/abcum/surreal/util/data"
)
func each(fld *sql.DefineFieldStatement, initial *data.Doc, current *data.Doc) (err error) {
var e bool
var i interface{}
var c interface{}
i = initial.Get(fld.Name).Data()
if fld.Readonly && i != nil {
current.Set(i, fld.Name)
return
}
if fld.Code != "" {
if cnf.Settings.DB.Lang == "js" {
vm := otto.New()
vm.Set("doc", current.Copy())
ret, err := vm.Run("(function() { " + fld.Code + " })()")
if err != nil {
return fmt.Errorf("Problem executing code: %v %v", fld.Code, err.Error())
}
if ret.IsUndefined() {
current.Del(fld.Name)
} else {
val, _ := ret.Export()
current.Set(val, fld.Name)
}
}
if cnf.Settings.DB.Lang == "lua" {
vm := lua.NewState()
defer vm.Close()
vm.SetGlobal("doc", toLUA(vm, current.Copy()))
if err := vm.DoString(fld.Code); err != nil {
return fmt.Errorf("Problem executing code: %v %v", fld.Code, err.Error())
}
ret := vm.Get(-1)
if ret == lua.LNil {
current.Del(fld.Name)
} else {
current.Set(frLUA(ret), fld.Name)
}
}
}
c = current.Get(fld.Name).Data()
e = current.Exists(fld.Name)
if fld.Default != nil && e == false {
switch val := fld.Default.(type) {
case sql.Null, *sql.Null:
current.Set(nil, fld.Name)
default:
current.Set(fld.Default, fld.Name)
case sql.Ident:
current.Set(current.Get(val.ID).Data(), fld.Name)
case *sql.Ident:
current.Set(current.Get(val.ID).Data(), fld.Name)
}
}
c = current.Get(fld.Name).Data()
e = current.Exists(fld.Name)
if fld.Notnull && e == true && c == nil {
return fmt.Errorf("Field '%v' can't be null", fld.Name)
}
if fld.Mandatory && e == false {
return fmt.Errorf("Need to set field '%v'", fld.Name)
}
if c != nil && fld.Type != "" {
switch fld.Type {
case "url":
if val, err := conv.ConvertToUrl(c); err == nil {
current.Set(val, fld.Name)
} else {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to be a URL", fld.Name)
} else {
current.Iff(i, fld.Name)
}
}
case "uuid":
if val, err := conv.ConvertToUuid(c); err == nil {
current.Set(val, fld.Name)
} else {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to be a UUID", fld.Name)
} else {
current.Iff(i, fld.Name)
}
}
case "color":
if val, err := conv.ConvertToColor(c); err == nil {
current.Set(val, fld.Name)
} else {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to be a HEX or RGB color", fld.Name)
} else {
current.Iff(i, fld.Name)
}
}
case "email":
if val, err := conv.ConvertToEmail(c); err == nil {
current.Set(val, fld.Name)
} else {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to be an email address", fld.Name)
} else {
current.Iff(i, fld.Name)
}
}
case "phone":
if val, err := conv.ConvertToPhone(c); err == nil {
current.Set(val, fld.Name)
} else {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to be a phone number", fld.Name)
} else {
current.Iff(i, fld.Name)
}
}
case "array":
if val, err := conv.ConvertToArray(c); err == nil {
current.Set(val, fld.Name)
} else {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to be an array", fld.Name)
} else {
current.Iff(i, fld.Name)
}
}
case "object":
if val, err := conv.ConvertToObject(c); err == nil {
current.Set(val, fld.Name)
} else {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to be an object", fld.Name)
} else {
current.Iff(i, fld.Name)
}
}
case "domain":
if val, err := conv.ConvertToDomain(c); err == nil {
current.Set(val, fld.Name)
} else {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to be a domain name", fld.Name)
} else {
current.Iff(i, fld.Name)
}
}
case "base64":
if val, err := conv.ConvertToBase64(c); err == nil {
current.Set(val, fld.Name)
} else {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to be base64 data", fld.Name)
} else {
current.Iff(i, fld.Name)
}
}
case "string":
if val, err := conv.ConvertToString(c); err == nil {
current.Set(val, fld.Name)
} else {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to be a string", fld.Name)
} else {
current.Iff(i, fld.Name)
}
}
case "number":
if val, err := conv.ConvertToNumber(c); err == nil {
current.Set(val, fld.Name)
} else {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to be a number", fld.Name)
} else {
current.Iff(i, fld.Name)
}
}
case "boolean":
if val, err := conv.ConvertToBoolean(c); err == nil {
current.Set(val, fld.Name)
} else {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to be a boolean", fld.Name)
} else {
current.Iff(i, fld.Name)
}
}
case "datetime":
if val, err := conv.ConvertToDatetime(c); err == nil {
current.Set(val, fld.Name)
} else {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to be a datetime", fld.Name)
} else {
current.Iff(i, fld.Name)
}
}
case "latitude":
if val, err := conv.ConvertToLatitude(c); err == nil {
current.Set(val, fld.Name)
} else {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to be a latitude value", fld.Name)
} else {
current.Iff(i, fld.Name)
}
}
case "longitude":
if val, err := conv.ConvertToLongitude(c); err == nil {
current.Set(val, fld.Name)
} else {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to be a longitude value", fld.Name)
} else {
current.Iff(i, fld.Name)
}
}
case "custom":
if val, err := conv.ConvertToOneOf(c, fld.Enum...); err == nil {
current.Set(val, fld.Name)
} else {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to be one of %v", fld.Name, fld.Enum)
} else {
current.Iff(i, fld.Name)
}
}
}
}
if fld.Match != "" {
if reg, err := regexp.Compile(fld.Match); err != nil {
return fmt.Errorf("Regular expression /%v/ is invalid", fld.Match)
} else {
if !reg.MatchString(fmt.Sprintf("%v", c)) {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to match the regular expression /%v/", fld.Name, fld.Match)
} else {
current.Iff(i, fld.Name)
}
}
}
}
if fld.Min != 0 {
if c = current.Get(fld.Name).Data(); c != nil {
switch now := c.(type) {
case []interface{}:
if len(now) < int(fld.Min) {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to have at least %v items", fld.Name, fld.Min)
} else {
current.Iff(i, fld.Name)
}
}
case string:
if len(now) < int(fld.Min) {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to have at least %v characters", fld.Name, fld.Min)
} else {
current.Iff(i, fld.Name)
}
}
case float64:
if now < fld.Min {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to be >= %v", fld.Name, fld.Min)
} else {
current.Iff(i, fld.Name)
}
}
}
}
}
if fld.Max != 0 {
if c = current.Get(fld.Name).Data(); c != nil {
switch now := c.(type) {
case []interface{}:
if len(now) > int(fld.Max) {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to have %v or fewer items", fld.Name, fld.Max)
} else {
current.Iff(i, fld.Name)
}
}
case string:
if len(now) > int(fld.Max) {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to have %v or fewer characters", fld.Name, fld.Max)
} else {
current.Iff(i, fld.Name)
}
}
case float64:
if now > fld.Max {
if fld.Validate {
return fmt.Errorf("Field '%v' needs to be <= %v", fld.Name, fld.Max)
} else {
current.Iff(i, fld.Name)
}
}
}
}
}
c = current.Get(fld.Name).Data()
e = current.Exists(fld.Name)
if fld.Default != nil && e == false {
switch val := fld.Default.(type) {
case sql.Null, *sql.Null:
current.Set(nil, fld.Name)
default:
current.Set(fld.Default, fld.Name)
case sql.Ident:
current.Set(current.Get(val.ID).Data(), fld.Name)
case *sql.Ident:
current.Set(current.Get(val.ID).Data(), fld.Name)
}
}
c = current.Get(fld.Name).Data()
e = current.Exists(fld.Name)
if fld.Notnull && e == true && c == nil {
return fmt.Errorf("Field '%v' can't be null", fld.Name)
}
if fld.Mandatory && e == false {
return fmt.Errorf("Need to set field '%v'", fld.Name)
}
return
}