//
//  rhimportd
//
//  The Radio Helsinki Rivendell Import Daemon
//
//
//  Copyright (C) 2015 Christian Pointner <equinox@helsinki.at>
//
//  This file is part of rhimportd.
//
//  rhimportd is free software: you can redistribute it and/or modify
//  it under the terms of the GNU General Public License as published by
//  the Free Software Foundation, either version 3 of the License, or
//  any later version.
//
//  rhimportd is distributed in the hope that it will be useful,
//  but WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//  GNU General Public License for more details.
//
//  You should have received a copy of the GNU General Public License
//  along with rhimportd. If not, see <http://www.gnu.org/licenses/>.
//

package rhimport

import (
	"database/sql"
	"fmt"
	_ "github.com/go-sql-driver/mysql"
	"github.com/vaughan0/go-ini"
)

type getPasswordResult struct {
	password string
	err      error
}

type getPasswordRequest struct {
	username string
	cached   bool
	response chan getPasswordResult
}

type Config struct {
	configfile        string
	RDXportEndpoint   string
	db_host           string
	db_user           string
	db_passwd         string
	db_db             string
	dbh               *sql.DB
	password_cache    map[string]string
	getPasswordChan   chan getPasswordRequest
	dbGetPasswordStmt *sql.Stmt
	quit              chan bool
	done              chan bool
}

func get_ini_value(file ini.File, section string, key string, dflt string) string {
	value, ok := file.Get(section, key)
	if ok {
		return value
	}
	return dflt
}

func (self *Config) read_config_file() error {
	file, err := ini.LoadFile(self.configfile)
	if err != nil {
		return err
	}

	self.db_host = get_ini_value(file, "mySQL", "Hostname", "localhost")
	self.db_user = get_ini_value(file, "mySQL", "Loginname", "rivendell")
	self.db_passwd = get_ini_value(file, "mySQL", "Password", "letmein")
	self.db_db = get_ini_value(file, "mySQL", "Database", "rivendell")
	return nil
}

func (self *Config) init_database() (err error) {
	dsn := fmt.Sprintf("%s:%s@tcp(%s:3306)/%s", self.db_user, self.db_passwd, self.db_host, self.db_db)
	if self.dbh, err = sql.Open("mysql", dsn); err != nil {
		return
	}
	if self.dbGetPasswordStmt, err = self.dbh.Prepare("select PASSWORD from USERS where LOGIN_NAME = ?"); err != nil {
		return
	}

	return
}

func (self *Config) getPassword(username string, cached bool) (pwd string, err error) {

	if cached {
		pwd = self.password_cache[username]
	}

	if pwd == "" {
		err = self.dbGetPasswordStmt.QueryRow(username).Scan(&pwd)
		if err != nil {
			if err == sql.ErrNoRows {
				err = fmt.Errorf("user '%s' not known by rivendell", username)
			}
			return
		}
		self.password_cache[username] = pwd
	}

	return
}

func (self *Config) dispatchRequests() {
	defer func() { self.done <- true }()
	for {
		select {
		case <-self.quit:
			return
		case req := <-self.getPasswordChan:
			if req.cached {
				rhdl.Println("Config: got getPassword request for", req.username, "(cached)")
			} else {
				rhdl.Println("Config: got getPassword request for", req.username, "(not cached)")
			}
			pwd, err := self.getPassword(req.username, req.cached)
			req.response <- getPasswordResult{pwd, err}
		}
	}
}

func (self *Config) Cleanup() {
	self.quit <- true
	<-self.done
	close(self.quit)
	close(self.done)
	close(self.getPasswordChan)
	if self.dbh != nil {
		self.dbh.Close()
	}
	if self.dbGetPasswordStmt != nil {
		self.dbGetPasswordStmt.Close()
	}
}

func NewConfig(configfile, rdxport_endpoint *string) (conf *Config, err error) {
	conf = new(Config)
	conf.configfile = *configfile
	if err = conf.read_config_file(); err != nil {
		return
	}
	conf.quit = make(chan bool)
	conf.done = make(chan bool)
	conf.RDXportEndpoint = *rdxport_endpoint
	conf.password_cache = make(map[string]string)
	conf.getPasswordChan = make(chan getPasswordRequest)

	if err = conf.init_database(); err != nil {
		return
	}

	go conf.dispatchRequests()
	return
}