module main

import config
import db.sqlite
import os

type GitlyDb = sqlite.DB

fn connect_db(conf config.Config) !GitlyDb {
	path := first_env(['GITLY_SQLITE_PATH', 'GITLY_DB_PATH'], conf.sqlite.path)
	mut db := sqlite.connect(path)!
	if db.busy_timeout(10000) != 0 {
		return error('failed to configure sqlite busy timeout')
	}
	db.exec('pragma journal_mode = WAL;') or { eprintln('cannot enable sqlite WAL mode: ${err}') }
	return GitlyDb(db)
}

fn db_backend_name() string {
	return 'sqlite'
}

fn db_exec_values(mut db GitlyDb, query string) ![][]string {
	rows := db.exec(query)!
	mut values := [][]string{cap: rows.len}
	for row in rows {
		values << row.vals.clone()
	}
	return values
}

fn db_last_insert_id(mut db GitlyDb) int {
	rows := db.exec('select last_insert_rowid()') or { return 0 }
	if rows.len > 0 && rows[0].vals.len > 0 {
		return rows[0].vals[0].int()
	}
	return 0
}

fn db_column_exists(mut db GitlyDb, table_name string, column_name string) !bool {
	rows := db_exec_values(mut db, 'pragma table_info(${sql_table(table_name)})')!
	for row in rows {
		if row.len > 1 && row[1] == column_name {
			return true
		}
	}
	return false
}

fn db_bool_column_type() string {
	return 'INTEGER NOT NULL DEFAULT 0'
}

fn first_env(keys []string, fallback string) string {
	for key in keys {
		value := os.getenv(key)
		if value != '' {
			return value
		}
	}
	return fallback
}