package main

import (
	"fmt"
	"io"
	"os"
	"strings"

	"github.com/bettercap/bettercap/core"
	"github.com/bettercap/bettercap/log"
	"github.com/bettercap/bettercap/modules"
	"github.com/bettercap/bettercap/session"

	"github.com/evilsocket/islazy/str"
	"github.com/evilsocket/islazy/tui"
)

func main() {
	sess, err := session.New()
	if err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
	defer sess.Close()

	if !tui.Effects() {
		if *sess.Options.NoColors {
			fmt.Printf("\n\nWARNING: Terminal colors have been disabled, view will be very limited.\n\n")
		} else {
			fmt.Printf("\n\nWARNING: This terminal does not support colors, view will be very limited.\n\n")
		}
	}

	appName := fmt.Sprintf("%s v%s", core.Name, core.Version)

	fmt.Printf("%s (type '%s' for a list of commands)\n\n", tui.Bold(appName), tui.Bold("help"))

	sess.Register(modules.NewEventsStream(sess))
	sess.Register(modules.NewTicker(sess))
	sess.Register(modules.NewUpdateModule(sess))
	sess.Register(modules.NewCapletsModule(sess))
	sess.Register(modules.NewMacChanger(sess))
	sess.Register(modules.NewProber(sess))
	sess.Register(modules.NewDiscovery(sess))
	sess.Register(modules.NewArpSpoofer(sess))
	sess.Register(modules.NewDHCP6Spoofer(sess))
	sess.Register(modules.NewDNSSpoofer(sess))
	sess.Register(modules.NewSniffer(sess))
	sess.Register(modules.NewPacketProxy(sess))
	sess.Register(modules.NewAnyProxy(sess))
	sess.Register(modules.NewTcpProxy(sess))
	sess.Register(modules.NewHttpProxy(sess))
	sess.Register(modules.NewHttpsProxy(sess))
	sess.Register(modules.NewHttpServer(sess))
	sess.Register(modules.NewRestAPI(sess))
	sess.Register(modules.NewWOL(sess))
	sess.Register(modules.NewWiFiModule(sess))
	sess.Register(modules.NewBLERecon(sess))
	sess.Register(modules.NewSynScanner(sess))
	sess.Register(modules.NewGPS(sess))
	sess.Register(modules.NewMySQLServer(sess))

	if err = sess.Start(); err != nil {
		log.Fatal("%s", err)
	}

	// Commands sent with -eval are used to set specific
	// caplet parameters (i.e. arp.spoof.targets) via command
	// line, therefore they need to be executed first otherwise
	// modules might already be started.
	for _, cmd := range session.ParseCommands(*sess.Options.Commands) {
		if err = sess.Run(cmd); err != nil {
			log.Error("Error while running '%s': %s", tui.Bold(cmd), tui.Red(err.Error()))
		}
	}

	// Some modules are enabled by default in order
	// to make the interactive session useful.
	for _, modName := range str.Comma(*sess.Options.AutoStart) {
		if err = sess.Run(modName + " on"); err != nil {
			log.Fatal("Error while starting module %s: %s", modName, err)
		}
	}

	// Then run the caplet if specified.
	if *sess.Options.Caplet != "" {
		if err = sess.RunCaplet(*sess.Options.Caplet); err != nil {
			log.Error("Error while running caplet %s: %s", tui.Bold(*sess.Options.Caplet), tui.Red(err.Error()))
		}
	}

	// Eventually start the interactive session.
	for sess.Active {
		line, err := sess.ReadLine()
		if err != nil {
			if err == io.EOF {
				continue
			} else if err.Error() == "Interrupt" {
				var ans string
				fmt.Printf("Are you sure you want to quit this session? y/n ")
				fmt.Scan(&ans)

				if strings.ToLower(ans) == "y" {
					sess.Run("exit")
					os.Exit(0)
				}
				continue
			} else {
				log.Fatal("%s", err)
			}
		}

		for _, cmd := range session.ParseCommands(line) {
			if err = sess.Run(cmd); err != nil {
				log.Error("%s", err)
			}
		}
	}
}