v/compiler/query.v

// Copyright (c) 2019 Alexander Medvednikov. All rights reserved.
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.

module main 

import strings 

// `db.select from User where id == 1 && nr_bookings > 0` 
fn (p mut Parser) select_query(fn_ph int) string {
	mut q := 'select ' 
	p.check(.key_select) 
	n := p.check_name() 
	if n == 'count' {
		q += 'count(*) from ' 
		p.check_name() 
	} 
	table_name := p.check_name() 
	// Register this type's fields as variables so they can be used in where expressions 
	typ := p.table.find_type(table_name) 
	if typ.name == '' {
		p.error('unknown type `$table_name`') 
	} 
	//fields := typ.fields.filter(typ == 'string' || typ == 'int') 
	// get only string and int fields 
	mut fields := []Var 
	for i, field in typ.fields {
		if field.typ != 'string' && field.typ != 'int' { 
			continue
		} 
		fields << field 
	} 
	if fields.len == 0 {
		p.error('V orm: select: empty fields in `$table_name`') 
	} 
	if fields[0].name != 'id' {
		p.error('V orm: `id int` must be the first field in `$table_name`') 
	} 
	// 'select id, name, age from...' 
	if n == 'from' {
		for i, field in fields {
			q += field.name  
			if i < fields.len - 1 {
				q += ', ' 
			} 
		} 
		q += ' from ' 
	} 
	for field in fields {
		//println('registering sql field var $field.name') 
		if field.typ != 'string' && field.typ != 'int' { 
			continue
		} 
		p.cur_fn.register_var({ field | is_used:true })  
	} 
	q += table_name 
	// `where` statement 
	if p.tok == .name && p.lit == 'where' { 
		p.next() 
		p.cgen.start_tmp() 
		p.is_sql = true 
		p.bool_expression() 
		p.is_sql = false 
		q += ' where ' + p.cgen.end_tmp() 
	} 
	// limit? 
	mut query_one := false 
	if p.tok == .name && p.lit == 'limit' {  
		p.next() 
		p.cgen.start_tmp() 
		p.is_sql = true 
		p.bool_expression() 
		p.is_sql = false 
		limit := p.cgen.end_tmp() 
		q += ' limit ' + limit 
		// `limit 1` means we are getting `User`, not `[]User` 
		if limit.trim_space() == '1' { 
			query_one = true 
		} 
	} 
	println('sql query="$q"') 
	if n == 'count' {
		p.cgen.set_placeholder(fn_ph, 'pg__DB_q_int(')
		p.gen(', tos2("$q"))') 
	} else { 
		// Build an object, assign each field. 
		tmp := p.get_tmp() 
		mut obj_gen := strings.new_builder(100) 
		for i, field in fields {
			mut cast := '' 
			if field.typ == 'int' {
				cast = 'string_int' 
			} 
			obj_gen.writeln('$tmp . $field.name = $cast( *(string*)array__get(row.vals, $i) );')    
		} 
		// One object 
		if query_one { 
			p.cgen.insert_before('

pg__Row row = pg__DB_exec_one(db, tos2("$q"));
$table_name $tmp; 
${obj_gen.str()} 

')
		p.cgen.resetln(tmp) 
} 
		// Array 
		else {
			p.cgen.insert_before('

array_pg__Row rows = pg__DB_exec(db, tos2("$q"));
//printf("ROWS LEN=%d\\n", rows.len); 
// TODO preallocate 
array arr_$tmp = new_array(0, 0, sizeof($table_name));  
for (int i = 0; i < rows.len; i++) { 
	pg__Row row = *(pg__Row*)array__get(rows, i); 
	$table_name $tmp; 
	${obj_gen.str()} 
	_PUSH(&arr_$tmp, $tmp, ${tmp}2, $table_name);  
} 
')
		p.cgen.resetln('arr_$tmp') 
} 
		 
	} 
	if n == 'count' {
		return 'int' 
	}	else if query_one {
		return table_name 
	}  else {
		p.register_array('array_$table_name') 
		return 'array_$table_name' 
	} 
} 

// `db.insert(user)` 
fn (p mut Parser) insert_query(fn_ph int) { 
	p.check_name() 
	p.check(.lpar) 
	var_name := p.check_name() 
	p.check(.rpar) 
	var := p.cur_fn.find_var(var_name) 
	typ := p.table.find_type(var.typ) 
	mut fields := []Var 
	for i, field in typ.fields {
		if field.typ != 'string' && field.typ != 'int' { 
			continue
		} 
		fields << field 
	} 
	if fields.len == 0 {
		p.error('V orm: insert: empty fields in `$var.typ`') 
	} 
	if fields[0].name != 'id' {
		p.error('V orm: `id int` must be the first field in `$var.typ`') 
	} 
	table_name := var.typ 
	mut sfields := ''  // 'name, city, country' 
	mut params := '' // params[0] = 'bob'; params[1] = 'Vienna'; 
	mut vals := ''  // $1, $2, $3... 
	mut nr_vals := 0 
	for i, field in fields {
		if field.name == 'id' {
			continue 
		} 
		sfields += field.name 
		vals += '$' + i.str() 
		nr_vals++ 
		params += 'params[${i-1}] = '
		if field.typ == 'string' { 
			params += '$var_name . $field.name .str;\n'    
		}  else if field.typ == 'int' {
			params += 'int_str($var_name . $field.name).str;\n'    
		} else {
			p.error('V ORM: unsupported type `$field.typ`') 
		} 
		if i < fields.len - 1 { 
			sfields += ', ' 
			vals += ', ' 
		} 
	} 
	p.cgen.insert_before('char* params[$nr_vals];' + params) 
	p.cgen.set_placeholder(fn_ph, 'PQexecParams( ')
	p.genln('.conn, "insert into $table_name ($sfields) values ($vals)", $nr_vals,
0, params, 0, 0, 0)') 
}