package main

import (
	"config"
	"datax"
	"flag"
	"fmt"
	"os"
	"runtime"
	"syscon"
	"syslog"
	"time"
)

func main() {
	if runtime.GOOS != "linux" {
		fmt.Println("this application is supposed to run on linux, there is nothing to do in here... \nexiting with status 0")
		return
	}
	defer syslog.HandlePan()

	fixDaemon := flag.NewFlagSet("daemon", flag.ExitOnError)
	confPath := fixDaemon.String("config", "fixrate.conf", "gnu config file for fixrate service")
	fixDaemon.Usage = fixRateUsage
	fixUser := flag.NewFlagSet("users", flag.ExitOnError)
	fixUser.Usage = addUserUsage
	confPathforUser := fixUser.String("config", "fixrate.conf", "gnu config file for fixrate service")
	userName := fixUser.String("username", "sina@snix.ir", "a username to add/modify in database")
	userRate := fixUser.Int("userrate", 10, "how many e-mails user should be able to send")
	userReset := fixUser.Int("counter", 120, "time interval between user counter reset")
	flag.Usage = globUsage
	if len(os.Args) < 2 {
		fmt.Println("expected 'daemon' or 'users' commands")
		flag.Usage()
		os.Exit(1)
	}
	switch os.Args[1] {
	case "daemon":
		fixDaemon.Parse(os.Args[2:])
		conf := config.ReadConfigFromFile(*confPath)
		syscon.StartNewService(conf)
	case "users":
		fixUser.Parse(os.Args[2:])
		if len(os.Args[2:]) == 0 {
			fmt.Println("expect more options -- use --h for information")
			fixUser.Usage()
			os.Exit(1)
		}
		conf := config.ReadConfigFromFile(*confPathforUser)
		if err := datax.SQLInit(conf); err != nil {
			panic(syslog.BigError{Why: err, Cod: 1})
		}
		defer datax.DBClose()
		temp := &datax.UserAccount{
			Username:  *userName,
			UserType:  true,
			LastReset: time.Now(),
			Limit:     *userRate,
			Reset:     *userReset,
		}
		if err := datax.CreateNewUser(temp); err != nil {
			panic(syslog.BigError{Why: err, Cod: 1})
		}
		fmt.Printf("\033[32minfo:\033[0m username: %v with %v e-mails per %v seconds added to database.\n", *userName, *userRate, *userReset)
	default:
		fmt.Println("expected 'daemon' or 'users' commands")
		flag.Usage()
		os.Exit(1)
	}
}
func globUsage() {
	fmt.Printf(`usage of fixrate postfix module snix.ir LLC:
%v commands... [ OPTIONS ] ...

commands:
  daemon         starting fixrate daemon, should be used by systemd
    --config     pass a file to read configuration from. default: ./fixrate.conf

  users          add or modify users and attributes in database
    --config     pass a file to read configuration from. default: ./fixrate.conf
    --username   a username to add/modify in database. default is sina@snix.ir
    --counter    time interval (seconds) between user counter reset. default is 120
    --userrate   how many e-mails user should be able to send. default is 10

example: 
---- adding name@domain.com ---- 100 e-mail per 10 minutes:
%v users --username name@domain.com -- userrate 100 --counter 600

Copyright (c) 2021 git.snix.ir, All rights reserved.
Developed BY sina@snix.ir --> Sina Ghaderi  
This work is licensed under the terms of the MIT license.
`, os.Args[0], os.Args[0])
}

func fixRateUsage() {
	fmt.Printf(`usage of fixrate postfix module snix.ir LLC:
%v daemon [ OPTIONS ] ...

options:
    --config    pass a file to read configuration from. default: ./fixrate.conf
    --h         print this banner and exit.

Copyright (c) 2021 git.snix.ir, All rights reserved.
Developed BY sina@snix.ir --> Sina Ghaderi  
This work is licensed under the terms of the MIT license.
`, os.Args[0])
}

func addUserUsage() {
	fmt.Printf(`usage of fixrate postfix module snix.ir LLC:
%v users [ OPTIONS ] ...

options:
    --config      pass a file to read configuration from. default: ./fixrate.conf
    --username    a username to add/modify in database. default is sina@snix.ir
    --counter     time interval (seconds) between user counter reset. default is 120
    --userrate    how many e-mails user should be able to send. default is 10

example: 
---- adding name@domain.com ---- 100 e-mail per 10 minutes:
%v users --username name@domain.com -- userrate 100 --counter 600
	
Copyright (c) 2021 git.snix.ir, All rights reserved.
Developed BY sina@snix.ir --> Sina Ghaderi  
This work is licensed under the terms of the MIT license.
`, os.Args[0], os.Args[0])
}