chai = require('chai') chai.use(require('chai-things')) _ = require 'lodash' chai.use((chai, utils) -> expect = chai.expect assertionPrototype = chai.Assertion.prototype utils.addProperty(assertionPrototype, 'query', -> obj = utils.flag(this, 'object') expect(obj).to.be.an.instanceof Array ) queryType = (type) -> -> obj = utils.flag(this, 'object') expect(obj).to.contain.something.that.equals type bodyClause = (bodyType) -> (bodyClauses...) -> obj = utils.flag(this, 'object') for bodyClause, i in bodyClauses expect(obj).to.contain.something.that.deep.equals([bodyType, bodyClause], bodyType + ' - ' + i) return this multiBodyClause = (bodyType) -> (bodyClauses...) -> obj = utils.flag(this, 'object') expect(obj).to.contain.something.that.deep.equals([bodyType, bodyClauses], bodyType) return this select = do -> bodySelect = bodyClause('Select') typeSelect = queryType('SelectQuery') -> typeSelect.call(this) bodySelect.apply(this, arguments) utils.addMethod(assertionPrototype, 'select', select) utils.addChainableMethod(assertionPrototype, 'selects', select) utils.addProperty(assertionPrototype, 'inserts', queryType('InsertQuery')) utils.addProperty(assertionPrototype, 'updates', queryType('UpdateQuery')) utils.addProperty(assertionPrototype, 'upserts', queryType('UpsertQuery')) utils.addProperty(assertionPrototype, 'deletes', queryType('DeleteQuery')) utils.addMethod(assertionPrototype, 'fields', multiBodyClause('Fields')) utils.addMethod(assertionPrototype, 'values', multiBodyClause('Values')) fromClause = bodyClause('From') utils.addMethod(assertionPrototype, 'from', (bodyClauses...) -> bodyClauses = bodyClauses.map((v) -> if typeof v == 'string' return ['Table', v] return ['Alias', ['Table', v[0]], v[1]] ) fromClause.apply(this, bodyClauses) ) utils.addMethod(assertionPrototype, 'where', bodyClause('Where')) utils.addMethod(assertionPrototype, 'orderby', (bodyClauses...) -> bodyType = 'OrderBy' obj = utils.flag(this, 'object') expect(obj).to.contain.something.that.deep.equals([bodyType].concat(bodyClauses), bodyType) return this ) utils.addMethod(assertionPrototype, 'groupby', multiBodyClause('GroupBy')) utils.addMethod(assertionPrototype, 'where', bodyClause('Where')) utils.addMethod(assertionPrototype, 'limit', bodyClause('Limit')) utils.addMethod(assertionPrototype, 'offset', bodyClause('Offset')) ) fs = require('fs') generateClientModel = (input) -> sbvrTypes = require '@balena/sbvr-types' typeVocab = fs.readFileSync(require.resolve('@balena/sbvr-types/Type.sbvr'), 'utf8') SBVRParser = require('@balena/sbvr-parser').SBVRParser.createInstance() SBVRParser.enableReusingMemoizations(SBVRParser._sideEffectingRules) SBVRParser.AddCustomAttribute('Database ID Field:') SBVRParser.AddCustomAttribute('Database Table Name:') SBVRParser.AddBuiltInVocab(typeVocab) LF2AbstractSQL = require '@balena/lf-to-abstract-sql' LF2AbstractSQLTranslator = LF2AbstractSQL.createTranslator(sbvrTypes) lf = SBVRParser.matchAll(input, 'Process') return LF2AbstractSQLTranslator(lf, 'Process') sbvrModel = fs.readFileSync(require.resolve('./model.sbvr'), 'utf8') exports.clientModel = clientModel = generateClientModel(sbvrModel) odataNameToSqlName = (odataName) -> odataName.replace(/__/g, '-').replace(/_/g, ' ') exports.sqlNameToOdataName = (sqlName) -> sqlName.replace(/-/g, '__').replace(/ /g, '_') exports.operandToAbstractSQLFactory = (binds = [], defaultResource = 'pilot', defaultParentAlias = defaultResource) -> return operandToAbstractSQL = (operand, resource = defaultResource, parentAlias = defaultParentAlias) -> if operand.abstractsql? return operand.abstractsql if _.isBoolean(operand) binds.push(['Boolean', operand]) return ['Bind', binds.length - 1] if typeof operand == 'number' binds.push(['Number', operand]) return ['Bind', binds.length - 1] if _.isDate(operand) binds.push(['Date', operand]) return ['Bind', binds.length - 1] if typeof operand == 'string' if operand is 'null' return ['Null'] if operand.charAt(0) is '(' return operand.slice(1, -1).split(',').map (op) -> n = _.parseInt(op) if _.isFinite(n) return operandToAbstractSQL(n) operandToAbstractSQL(op) if operand.charAt(0) is "'" binds.push(['Text', decodeURIComponent(operand[1...(operand.length - 1)])]) return ['Bind', binds.length - 1] fieldParts = operand.split('/') if fieldParts.length > 1 alias = parentAlias previousResource = _(parentAlias).split('.').last() for resourceName in fieldParts[...-1] sqlName = odataNameToSqlName(resourceName) sqlNameParts = sqlName.split('-') mapping = _.get(clientModel.relationships[previousResource], sqlNameParts.join('.')).$ refTable = mapping[1][0] if sqlNameParts.length > 1 and not refTable.includes('-') # Add the verb to tables that don't include the verb already alias = "#{alias}.#{sqlNameParts[0]}-#{refTable}" else alias = "#{alias}.#{refTable}" previousResource = refTable mapping = [alias, _.last(fieldParts)] else mapping = [resource, odataNameToSqlName(operand)] return ['ReferencedField'].concat(mapping) if Array.isArray(operand) return operandToAbstractSQL(operand...) if _.isObject(operand) return [ 'Duration', operand ] throw new Error('Unknown operand type: ' + operand) exports.operandToOData = operandToOData = (operand) -> if operand.odata? return operand.odata if _.isDate(operand) return "datetime'" + encodeURIComponent(operand.toISOString()) + "'" if Array.isArray(operand) return operandToOData(operand[0]) if _.isObject(operand) duration = [] t = false if operand.negative duration.push('-') duration.push('P') if operand.day? duration.push(operand.day, 'D') if operand.hour? t = true duration.push('T', operand.hour, 'H') if operand.minute? if not t t = true duration.push('T') duration.push(operand.minute, 'M') if operand.second? if not t t = true duration.push('T') duration.push(operand.second, 'S') if duration.length < 3 throw new Error('Duration must contain at least 1 component') return "duration'#{duration.join('')}'" return operand exports.shortenAlias = shortenAlias = (alias) -> while alias.length >= 64 alias = alias.replace(/(^|[^-])pilot/, '$1pi').replace(/trained\-pilot/, 'tr-pi') return alias exports.aliasFields = do -> aliasField = (resourceAlias, verb, field) -> if field[0] is 'ReferencedField' return [field[0], shortenAlias("#{resourceAlias}.#{verb}#{field[1]}"), field[2]] if field[0] is 'Alias' return [ 'Alias' aliasField(resourceAlias, verb, field[1]) field[2] ] else return field return (resourceAlias, fields, verb) -> if verb? verb = verb + '-' else verb = '' fields.map(_.partial(aliasField, resourceAlias, verb)) exports.pilotFields = [ ['Alias', ['ReferencedField', 'pilot', 'created at'], 'created_at'] ['Alias', ['ReferencedField', 'pilot', 'modified at'], 'modified_at'] ['ReferencedField', 'pilot', 'id'] ['ReferencedField', 'pilot', 'person'] ['Alias', ['ReferencedField', 'pilot', 'is experienced'], 'is_experienced'] ['ReferencedField', 'pilot', 'name'] ['ReferencedField', 'pilot', 'age'] ['Alias', ['ReferencedField', 'pilot', 'favourite colour'], 'favourite_colour'] ['Alias', ['ReferencedField', 'pilot', 'is on-team'], 'is_on__team'], ['ReferencedField', 'pilot', 'licence'] ['Alias', ['ReferencedField', 'pilot', 'hire date'], 'hire_date'] ['Alias', ['ReferencedField', 'pilot', 'was trained by-pilot'], 'was_trained_by__pilot'] ] exports.licenceFields = [ ['Alias', ['ReferencedField', 'licence', 'created at'], 'created_at'] ['Alias', ['ReferencedField', 'licence', 'modified at'], 'modified_at'] ['ReferencedField', 'licence', 'id'] ['ReferencedField', 'licence', 'name'] ] exports.planeFields = [ ['Alias', ['ReferencedField', 'plane', 'created at'], 'created_at'] ['Alias', ['ReferencedField', 'plane', 'modified at'], 'modified_at'] ['ReferencedField', 'plane', 'id'] ['ReferencedField', 'plane', 'name'] ] exports.pilotCanFlyPlaneFields = [ ['Alias', ['ReferencedField', 'pilot-can fly-plane', 'created at'], 'created_at'] ['Alias', ['ReferencedField', 'pilot-can fly-plane', 'modified at'], 'modified_at'] ['ReferencedField', 'pilot-can fly-plane', 'pilot'] ['Alias', ['ReferencedField', 'pilot-can fly-plane', 'can fly-plane'], 'can_fly__plane'] ['ReferencedField', 'pilot-can fly-plane', 'id'] ] exports.teamFields = [ ['Alias', ['ReferencedField', 'team', 'created at'], 'created_at'] ['Alias', ['ReferencedField', 'team', 'modified at'], 'modified_at'] ['Alias', ['ReferencedField', 'team', 'favourite colour'], 'favourite_colour'] ] exports.$count = [[ 'Alias', ['Count', '*'], '$count' ]]