wow-item-generator/main.go

package main

import (
	"flag"
	"fmt"
	"io"
	"log"
	"os"
	"strings"

	"github.com/araxiaonline/endgame-item-generator/internal/db/mysql"
	"github.com/araxiaonline/endgame-item-generator/internal/db/sqlite"
	"github.com/araxiaonline/endgame-item-generator/internal/items"

	_ "github.com/go-sql-driver/mysql"
	"github.com/joho/godotenv"
)

func main() {

	log.SetFlags(log.LstdFlags | log.Lshortfile)
	godotenv.Load()
	// database.models.Connect()

	debug := flag.Bool("debug", false, "Enable verbose logging inside generator")
	itemLevel := flag.Int("ilvl", 300, "Specify the item level to start scaling from, expansion and difficulty modifiers scale up.")
	difficulty := flag.Int("difficulty", 3, "set the difficulty of the dungeon, defaults to 3 (mythic) 4 (legendary) 5 (ascendant)")
	// levelUp := flag.Bool("levelUp", false, "Boss items require higher +1 level to equip, defaults to false")
	baselevel := flag.Int("baselevel", 80, "set the base level for items to be used, defaults to 80 this is required for levelUp flag")
	flag.Parse()

	if difficulty == nil || *difficulty < 3 || *difficulty > 5 {
		log.Fatal("difficulty must be between 3-5")
		os.Exit(1)
	}

	if itemLevel == nil || *itemLevel < 280 {
		log.Fatal("item level must be greater than 280")
		os.Exit(1)
	}

	if baselevel == nil || *baselevel < 0 {
		log.Fatal("base level must be greater than 80")
		os.Exit(1)
	}

	if *debug {
		log.SetOutput(os.Stdout)
	} else {
		log.SetOutput(io.Discard)
	}

	// Connect to Mysql
	mysqlDb, err := mysql.Connect(&mysql.MySqlConfig{
		Host:     os.Getenv("DB_HOST"),
		User:     os.Getenv("DB_USER"),
		Password: os.Getenv("DB_PASSWORD"),
		Database: os.Getenv("DB_NAME"),
	})

	if err != nil {
		log.Fatal(err)
	}

	// Connect to SqlList for EndGame Mapping
	sqliteDb, err := sqlite.Connect("./data/items.db")
	if err != nil {
		log.Fatal(err)
	}

	rareItems, err := mysqlDb.GetRarePlusItems(0, 0)
	if err != nil {
		log.Fatal(err)
	}

	for itr, dbItem := range rareItems {
		item := items.ItemFromDbItem(dbItem)

		// if it is a rare item then we need to scale it up to epic
		if *item.Quality < 5 {
			*item.Quality = 4
		}

		statsList, err := item.GetStatList()
		if err != nil {
			log.Fatal(err)
			continue
		}

		log.Printf("Item: %v Entry: %v StatsList: %v\n", item.Name, item.Entry, statsList)
		rndItem, err := sqliteDb.GetRandItem(*item.Class, *item.Subclass, statsList, false)
		if err != nil {
			log.Print(err)
			continue
		}

		if rndItem == (sqlite.HighLevelItem{}) {
			log.Fatalf("Failed to get random item for %v Entry: %v\n", item.Name, item.Entry)
		}

		log.Printf("Random Item: %v Entry: %v\n", rndItem.Name, rndItem.Entry)

		highLevelItem, err := mysqlDb.GetItem(rndItem.Entry)
		if err != nil {
			log.Fatal(err)
			continue
		}

		// Take the high level item that has been selected for stats and remap to current item

		// Now apply logic build out the different difficulties and item levels based on source of item
		lookupItem, err := sqliteDb.GetItemFromDungeon(item.Entry)
		if err != nil {
			if !strings.Contains(err.Error(), "no rows in result set") {
				log.Fatalf("failed to lookup item %v from dungeon: %v", item.Entry, err)
			}
		}

		// Start at the item level and scale up based details about the source of the item

		// if the item is from a dungeon and not a boss and is a craftable / world item so set to base level
		if lookupItem.Entry == 0 {

			Scale(highLevelItem, &item, *itemLevel, 3)
			fmt.Print(items.ItemToSql(item, *baselevel, *difficulty))
			continue
		}

		// if the item is from a dungeon and not a boss item
		if lookupItem.CreatureId == 0 {

			if lookupItem.DungeonLevel < 60 {
				Scale(highLevelItem, &item, *itemLevel+5, 3)
				fmt.Print(items.ItemToSql(item, *baselevel, *difficulty))
			}

			if lookupItem.DungeonLevel == 60 && lookupItem.Expansion == 0 {
				Scale(highLevelItem, &item, *itemLevel+10, 3)
				fmt.Print(items.ItemToSql(item, *baselevel+2, *difficulty))
			}

			if lookupItem.DungeonLevel < 70 && lookupItem.Expansion == 1 {
				Scale(highLevelItem, &item, *itemLevel+7, 3)
				fmt.Print(items.ItemToSql(item, *baselevel, *difficulty))
			}

			if lookupItem.DungeonLevel == 70 && lookupItem.Expansion == 1 {
				Scale(highLevelItem, &item, *itemLevel+10, 3)
				fmt.Print(items.ItemToSql(item, *baselevel+2, *difficulty))
			}

			if lookupItem.DungeonLevel < 80 && lookupItem.Expansion == 2 {
				Scale(highLevelItem, &item, *itemLevel+7, 3)
			}

			if lookupItem.DungeonLevel == 80 && lookupItem.Expansion == 2 {
				Scale(highLevelItem, &item, *itemLevel+10, 3)
				fmt.Print(items.ItemToSql(item, *baselevel+2, *difficulty))
			}
		} else {

			var finalBonus int = 0
			var quality int = 4
			// check if it is the final boss
			if mysql.IsFinalBoss(lookupItem.CreatureId) {
				finalBonus = 5
				quality = 5
			}
			// if the item is from a boss fight
			if lookupItem.DungeonLevel < 60 {
				Scale(highLevelItem, &item, *itemLevel+9+finalBonus, quality)
				fmt.Print(items.ItemToSql(item, *baselevel, *difficulty))
			}

			if lookupItem.DungeonLevel == 60 && lookupItem.Expansion == 0 {
				Scale(highLevelItem, &item, *itemLevel+17+finalBonus, quality)
				fmt.Print(items.ItemToSql(item, *baselevel+2, *difficulty))
			}

			if lookupItem.DungeonLevel < 70 && lookupItem.Expansion == 1 {
				Scale(highLevelItem, &item, *itemLevel+10+finalBonus, quality)
				fmt.Print(items.ItemToSql(item, *baselevel, *difficulty))
			}

			if lookupItem.DungeonLevel == 70 && lookupItem.Expansion == 1 {
				Scale(highLevelItem, &item, *itemLevel+19+finalBonus, quality)
				fmt.Print(items.ItemToSql(item, *baselevel+2, *difficulty))
			}

			if lookupItem.DungeonLevel < 80 && lookupItem.Expansion == 2 {
				Scale(highLevelItem, &item, *itemLevel+12+finalBonus, quality)
			}

			if lookupItem.DungeonLevel == 80 && lookupItem.Expansion == 2 {
				Scale(highLevelItem, &item, *itemLevel+22+finalBonus, quality)
				fmt.Print(items.ItemToSql(item, *baselevel+3, *difficulty))
			}
		}

		fmt.Printf("\n -- Item Updated: %v Entry: %v\n", item.Name, item.Entry)
		if itr >= 300 {
			// os.exit(0)
		}
	}
}

func Scale(highLevelItem mysql.DbItem, item *items.Item, itemLevel, quality int) {
	item.ApplyStats(items.ItemFromDbItem(highLevelItem))
	item.ScaleItem(itemLevel, quality)
	log.Printf("Item Name: %v Stat1: %v Stat2: %v Stat3: %v Stat4: %v Stat5: %v Stat6: %v Stat7: %v Stat8: %v \n",
		item.Name, *item.StatValue1, *item.StatValue2, *item.StatValue3, *item.StatValue4, *item.StatValue5, *item.StatValue6, *item.StatValue7, *item.StatValue8)
}