surrealpatch/sql/exprs.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 sql

import (
	"regexp"
	"strconv"
	"time"
)

// --------------------------------------------------
//
// --------------------------------------------------

func (p *Parser) parseTable() (Table, error) {

	_, lit, err := p.shouldBe(IDENT, NUMBER, DATE)
	if err != nil {
		return Table(""), &ParseError{Found: lit, Expected: []string{"table name"}}
	}

	return Table(lit), err

}

func (p *Parser) parseTables() (mul []Table, err error) {

	for {

		one, err := p.parseTable()
		if err != nil {
			return nil, err
		}

		mul = append(mul, one)

		// If the next token is not a comma then break the loop.
		if _, _, exi := p.mightBe(COMMA); !exi {
			break
		}

	}

	return

}

// --------------------------------------------------
//
// --------------------------------------------------

func (p *Parser) parseThing() (one *Thing, err error) {

	var tok Token
	var lit string
	var val interface{}

	one = &Thing{}

	_, _, err = p.shouldBe(EAT)
	if err != nil {
		return nil, err
	}

	_, one.TB, err = p.shouldBe(IDENT, NUMBER, DATE)
	if err != nil {
		return nil, &ParseError{Found: one.TB, Expected: []string{"table name"}}
	}

	_, _, err = p.shouldBe(COLON)
	if err != nil {
		return nil, err
	}

	tok, lit, err = p.shouldBe(IDENT, NUMBER, DOUBLE, DATE, TIME)
	if err != nil {
		return nil, &ParseError{Found: lit, Expected: []string{"table id"}}
	}

	switch tok {
	case IDENT:
		val = lit
	case NUMBER:
		val, err = strconv.ParseInt(lit, 10, 64)
	case DOUBLE:
		val, err = strconv.ParseFloat(lit, 64)
	case DATE:
		val, err = time.Parse("2006-01-02", lit)
	case TIME:
		val, err = time.Parse(time.RFC3339, lit)
	}

	if err != nil {
		return nil, err
	}

	one.ID = val

	return

}

func (p *Parser) parseThings() (mul []Expr, err error) {

	for {

		one, err := p.parseThing()
		if err != nil {
			return nil, err
		}

		mul = append(mul, one)

		// If the next token is not a comma then break the loop.
		if _, _, exi := p.mightBe(COMMA); !exi {
			break
		}

	}

	return

}

// --------------------------------------------------
//
// --------------------------------------------------

func (p *Parser) parseIdent() (Ident, error) {

	_, lit, err := p.shouldBe(IDENT)
	if err != nil {
		return Ident(""), &ParseError{Found: lit, Expected: []string{"name"}}
	}

	val, err := declare(IDENT, lit)

	return val.(Ident), err

}

func (p *Parser) parseArray() ([]interface{}, error) {

	_, lit, err := p.shouldBe(ARRAY)
	if err != nil {
		return nil, &ParseError{Found: lit, Expected: []string{"array"}}
	}

	val, err := declare(ARRAY, lit)

	return val.([]interface{}), err

}

func (p *Parser) parseNumber() (int64, error) {

	_, lit, err := p.shouldBe(NUMBER)
	if err != nil {
		return int64(0), &ParseError{Found: lit, Expected: []string{"number"}}
	}

	val, err := declare(NUMBER, lit)

	return val.(int64), err

}

func (p *Parser) parseDouble() (float64, error) {

	_, lit, err := p.shouldBe(NUMBER, DOUBLE)
	if err != nil {
		return float64(0), &ParseError{Found: lit, Expected: []string{"number"}}
	}

	val, err := declare(DOUBLE, lit)

	return val.(float64), err

}

func (p *Parser) parseString() (string, error) {

	_, lit, err := p.shouldBe(STRING)
	if err != nil {
		return string(""), &ParseError{Found: lit, Expected: []string{"string"}}
	}

	val, err := declare(STRING, lit)

	return val.(string), err

}

func (p *Parser) parseRegion() (string, error) {

	tok, lit, err := p.shouldBe(IDENT, STRING, REGION)
	if err != nil {
		return string(""), &ParseError{Found: lit, Expected: []string{"string"}}
	}

	val, err := declare(tok, lit)

	return val.(string), err

}

func (p *Parser) parseScript() (string, error) {

	tok, lit, err := p.shouldBe(STRING, REGION)
	if err != nil {
		return string(""), &ParseError{Found: lit, Expected: []string{"LUA script"}}
	}

	val, err := declare(tok, lit)

	return val.(string), err

}

func (p *Parser) parseRegexp() (string, error) {

	tok, lit, err := p.shouldBe(REGEX)
	if err != nil {
		return string(""), &ParseError{Found: lit, Expected: []string{"regular expression"}}
	}

	val, err := declare(tok, lit)

	return val.(*regexp.Regexp).String(), err

}

func (p *Parser) parseBoolean() (bool, error) {

	tok, lit, err := p.shouldBe(TRUE, FALSE)
	if err != nil {
		return bool(false), &ParseError{Found: lit, Expected: []string{"boolean"}}
	}

	val, err := declare(tok, lit)

	return val.(bool), err

}

func (p *Parser) parseDefault() (interface{}, error) {

	tok, lit, err := p.shouldBe(NULL, NOW, DATE, TIME, TRUE, FALSE, NUMBER, STRING, REGION, ARRAY, JSON)
	if err != nil {
		return nil, err
	}

	return declare(tok, lit)

}

func (p *Parser) parseExpr() (mul []*Field, err error) {

	var tok Token
	var lit string
	var exi bool

	for {

		one := &Field{}

		tok, lit, err = p.shouldBe(IDENT, ID, OEDGE, IEDGE, BEDGE, NULL, ALL, TIME, TRUE, FALSE, STRING, REGION, NUMBER, DOUBLE, JSON)
		if err != nil {
			return nil, &ParseError{Found: lit, Expected: []string{"field name"}}
		}

		one.Expr, err = declare(tok, lit)
		if err != nil {
			return
		}

		// Next token might be AS
		if _, _, exi = p.mightBe(AS); exi {

			tok, lit, err = p.shouldBe(IDENT)
			if err != nil {
				return nil, &ParseError{Found: lit, Expected: []string{"field alias"}}
			}

			one.Alias, err = declare(tok, lit)
			if err != nil {
				return
			}

		}

		mul = append(mul, one)

		// If the next token is not a comma then break the loop.
		if _, _, exi = p.mightBe(COMMA); !exi {
			break
		}

	}

	return

}