main.go

package main

import (
	"flag"
	"log"

	"github.com/evris99/dex-limit-order/bot"
	"github.com/evris99/dex-limit-order/config"
	"github.com/evris99/dex-limit-order/database"
	"github.com/evris99/dex-limit-order/manager"
	"github.com/evris99/dex-limit-order/wallet"
)

func main() {
	walletPath := flag.String("wallet", "", "if set creates a wallet at the given path")
	confPath := flag.String("config", "./configuration.toml", "the path to the configuration flag")
	userID := flag.Int64("new_user_id", -1, "a user telegram ID to add to the database")
	flag.Parse()

	// Generate wallet and exit
	if *walletPath != "" {
		if _, err := wallet.GenerateFromTerm(*walletPath); err != nil {
			log.Fatalln(err)
		}
		log.Fatalln("Successfully created wallet")
	}

	// Load the configuration from the given file
	config, err := config.Load(*confPath)
	if err != nil {
		log.Fatalln(err)
	}

	// Creates a database connection
	DB, err := database.New(config.DBPath)
	if err != nil {
		log.Fatalln(err)
	}

	// Create new user and exit
	if *userID != -1 {
		if err := DB.AddUser(*userID); err != nil {
			log.Fatalln(err)
		}
		log.Fatalf("Successfully added user with telegram ID: %d", *userID)
	}

	// Initializes a wallet instance
	wallet, err := wallet.LoadWithPassword(config.WalletPath, config.Password)
	if err != nil {
		log.Fatalln(err)
	}

	// Migrates the database
	if err := DB.Migrate(); err != nil {
		log.Fatalln(err)
	}

	// Initialize telegram bot
	telegramBot, err := bot.New(DB, config.TelegramAPIKey)
	if err != nil {
		log.Fatalln(err)
	}

	// Create order manager
	manager, err := manager.New(telegramBot, wallet, DB, config.ChainID, config.EndpointURL, config.PancakeRouterHex)
	if err != nil {
		log.Fatalln(err)
	}

	// Start watching existing orders from the database
	if err := manager.StartDBOrders(); err != nil {
		log.Fatalln(err)
	}

	log.Println("Bot started")
	bot.Listen(telegramBot, DB, manager)
}