package main

import (
	"database/sql"
	_ "github.com/go-sql-driver/mysql"
	"log"
	"time"
)

func main() {
	log.Println("Reading config...")
	cfg, err := readConfig()
	if err != nil {
		log.Println("Could not read config")
		return
	}
	log.Println("Done!")

	switch cfg.DataSource {
	case "stdin":
		processFromStdin()
	case "mysq":
		pricessFromDB()
	default:
		log.Println("Invalid dataSource in config. Needs to be either 'stdin' or 'mysql'.")
	}

	log.Println("Finished processing, now exiting")
}

func processFromStdin() {
	log.Println("Starting to process from stdin...")
	input := readFromStdin()
	rDatChan := parseRawData(input, cfg)
	cleanFromStdin(rDatChan, cfg)
}

func processFromDB() {
	log.Print("Cleaning data...")
	starttime := time.Now()
	numOfRowsNotCleaned, err := cleanFromDB(cfg)
	if err != nil {
		log.Println(err)
		log.Println("Exiting...")
		return
	}
	log.Println("Done!")

	// If  either all rows are processed or if there is no limit for the processing
	// we can safely add noise to the cleaned data
	if (numOfRowsNotCleaned == 0 || cfg.Limit == 0) && cfg.Epsilon >= 0 {
		log.Println("Adding differential privacy noise to processed data...")
		db, err := sql.Open("mysql", cfg.DBUser+":"+cfg.DBPass+"@"+cfg.DBConn+"/"+cfg.DBName)
		if err != nil {
			log.Println("Failed to connect to db:", err)
			return
		}
		defer db.Close()

		ival, err := cfg.getInterval()
		if err != nil {
			log.Println("erronous interval in conf prevents the privatization of data:", err)
			return
		}

		err = privatizeCleaned(db, starttime.Add(-2*ival), cfg)
		if err != nil {
			log.Println("Failed to privatize data:", err)
		}
		log.Println("Done!")
	}
}