main.go

package main

import (
	"fmt"
	"io/ioutil"
	"os"
	"path/filepath"
	"strings"
	"runtime"
)

func main() {
	// Read environment variables
	envVars := map[string]string{
		"Difficulty":                          os.Getenv("DIFFICULTY"),
		"DayTimeSpeedRate":                    os.Getenv("DAY_TIME_SPEED_RATE"),
		"NightTimeSpeedRate":                  os.Getenv("NIGHT_TIME_SPEED_RATE"),
		"ExpRate":                             os.Getenv("EXP_RATE"),
		"PalCaptureRate":                      os.Getenv("PAL_CAPTURE_RATE"),
		"PalSpawnNumRate":                     os.Getenv("PAL_SPAWN_NUM_RATE"),
		"PalDamageRateAttack":                 os.Getenv("PAL_DAMAGE_RATE_ATTACK"),
		"PalDamageRateDefense":                os.Getenv("PAL_DAMAGE_RATE_DEFENSE"),
		"PlayerDamageRateAttack":              os.Getenv("PLAYER_DAMAGE_RATE_ATTACK"),
		"PlayerDamageRateDefense":             os.Getenv("PLAYER_DAMAGE_RATE_DEFENSE"),
		"PlayerStomachDecreaceRate":           os.Getenv("PLAYER_STOMACH_DECREACE_RATE"),
		"PlayerStaminaDecreaceRate":           os.Getenv("PLAYER_STAMINA_DECREACE_RATE"),
		"PlayerAutoHPRegeneRate":              os.Getenv("PLAYER_AUTO_HP_REGENE_RATE"),
		"PlayerAutoHpRegeneRateInSleep":       os.Getenv("PLAYER_AUTO_HP_REGENE_RATE_IN_SLEEP"),
		"PalStomachDecreaceRate":              os.Getenv("PAL_STOMACH_DECREACE_RATE"),
		"PalStaminaDecreaceRate":              os.Getenv("PAL_STAMINA_DECREACE_RATE"),
		"PalAutoHPRegeneRate":                 os.Getenv("PAL_AUTO_HP_REGENE_RATE"),
		"PalAutoHpRegeneRateInSleep":          os.Getenv("PAL_AUTO_HP_REGENE_RATE_IN_SLEEP"),
		"BuildObjectDamageRate":               os.Getenv("BUILD_OBJECT_DAMAGE_RATE"),
		"BuildObjectDeteriorationDamageRate":  os.Getenv("BUILD_OBJECT_DETERIORATION_DAMAGE_RATE"),
		"CollectionDropRate":                  os.Getenv("COLLECTION_DROP_RATE"),
		"CollectionObjectHpRate":              os.Getenv("COLLECTION_OBJECT_HP_RATE"),
		"CollectionObjectRespawnSpeedRate":    os.Getenv("COLLECTION_OBJECT_RESPAWN_SPEED_RATE"),
		"EnemyDropItemRate":                   os.Getenv("ENEMY_DROP_ITEM_RATE"),
		"DeathPenalty":                        os.Getenv("DEATH_PENALTY"),
		"bEnablePlayerToPlayerDamage":         os.Getenv("ENABLE_PLAYER_TO_PLAYER_DAMAGE"),
		"bEnableFriendlyFire":                 os.Getenv("ENABLE_FRIENDLY_FIRE"),
		"bEnableInvaderEnemy":                 os.Getenv("ENABLE_ENEMY"),
		"bActiveUNKO":                         os.Getenv("ACTIVE_UNKO"),
		"bEnableAimAssistPad":                 os.Getenv("ENABLE_AIM_ASSIST_PAD"),
		"bEnableAimAssistKeyboard":            os.Getenv("ENABLE_AIM_ASSIST_KEYBOARD"),
		"DropItemMaxNum":                      os.Getenv("DROP_ITEM_MAX_NUM"),
		"DropItemMaxNum_UNKO":                 os.Getenv("DROP_ITEM_MAX_NUM_UNKO"),
		"BaseCampMaxNum":                      os.Getenv("BASE_CAMP_MAX_NUM"),
		"BaseCampWorkerMaxNum":                os.Getenv("BASE_CAMP_WORKER_MAX_NUM"),
		"DropItemAliveMaxHours":               os.Getenv("DROP_ITEM_ALIVE_MAX_HOURS"),
		"bAutoResetGuildNoOnlinePlayers":      os.Getenv("AUTO_RESET_GUILD_NO_ONLINE_PLAYERS"),
		"AutoResetGuildTimeNoOnlinePlayers":   os.Getenv("AUTO_RESET_GUILD_TIME_NO_ONLINE_PLAYERS"),
		"GuildPlayerMaxNum":                   os.Getenv("GUILD_PLAYER_MAX_NUM"),
		"PalEggDefaultHatchingTime":           os.Getenv("PAL_EGG_DEFAULT_HATCHING_TIME"),
		"WorkSpeedRate":                       os.Getenv("WORK_SPEED_RATE"),
		"bIsMultiplay":                        os.Getenv("IS_MULTIPLAY"),
		"bIsPvP":                              os.Getenv("IS_PVP"),
		"bCanPickupOtherGuildDeathPenaltyDrop":os.Getenv("CAN_PICKUP_OTHER_GUILD_DEATH_PENALTY_DROP"),
		"bEnableNonLoginPenalty":              os.Getenv("ENABLE_NON_LOGIN_PENALTY"),
		"bEnableFastTravel":                   os.Getenv("ENABLE_FAST_TRAVEL"),
		"bIsStartLocationSelectByMap":         os.Getenv("IS_START_LOCATION_SELECT_BY_MAP"),
		"bExistPlayerAfterLogout":             os.Getenv("EXIST_PLAYER_AFTER_LOGOUT"),
		"bEnableDefenseOtherGuildPlayer":      os.Getenv("ENABLE_DEFENSE_OTHER_GUILD_PLAYER"),
		"CoopPlayerMaxNum":                    os.Getenv("COOP_PLAYER_MAX_NUM"),
		"ServerPlayerMaxNum":                  os.Getenv("MAX_PLAYERS"),
		"ServerName":                          os.Getenv("SERVER_NAME"),
		"ServerDescription":                   os.Getenv("SERVER_DESCRIPTION"),
		"ServerPassword":                      os.Getenv("SERVER_PASSWORD"),
		"AdminPassword":                       os.Getenv("ADMIN_PASSWORD"),
		"PublicIP":                            os.Getenv("PUBLIC_IP"),
		"PublicPort":                          os.Getenv("SERVER_PORT"),
		"RCONPort":                            os.Getenv("RCON_PORT"),
		"RCONEnabled":                         os.Getenv("RCON_ENABLE"),
		"bUseAuth":                            os.Getenv("USE_AUTH"),
		"BanListURL":                          os.Getenv("BAN_LIST_URL"),
		// Add other environment variables and corresponding INI keys here
	}

	// Specify keys for which quotes should be added
	envVarsQuotes := map[string]bool{
		"ServerName":     true,
		"ServerPassword": true,
		"AdminPassword":  true,
		"ServerDescription": true,
		"BanListURL": true,
		// Add other keys as needed
	}

	// Determine the operating system
	var osFolder string
	switch runtime.GOOS {
	case "windows":
		osFolder = "WindowsServer"
	case "linux":
		osFolder = "LinuxServer"
	default:
		fmt.Println("Unsupported operating system")
		return
	}

	// Get the absolute path to the INI file
	iniFilePath, err := filepath.Abs(fmt.Sprintf("Pal/Saved/Config/%s/PalWorldSettings.ini", osFolder))
		if err != nil {
		fmt.Printf("Error getting absolute path: %v\n", err)
		return
	}

	// Check if the INI file exists
	if _, err := os.Stat(iniFilePath); os.IsNotExist(err) {
		fmt.Printf("INI file not found, Exiting")
		return
	}

	// Read the contents of the original INI file
	iniContent, err := ioutil.ReadFile(iniFilePath)
	if err != nil {
		fmt.Printf("Error reading INI file: %v\n", err)
		return
	}

	// Update values based on environment variables
	for key, value := range envVars {
		if value != "" {
			fmt.Printf("Updating key: %s with value: %s\n", key, value)
			setINIValue(&iniContent, key, value, envVarsQuotes[key])
		}
	}

	// Write the updated contents back to the INI file
	err = ioutil.WriteFile(iniFilePath, iniContent, 0644)
	if err != nil {
		fmt.Printf("Error writing updated INI file: %v\n", err)
		return
	}

	fmt.Println("INI file updated successfully.")
}

// setINIValue updates the value for the specified key in the INI content.
func setINIValue(content *[]byte, key, value string, addQuotes bool) {
	// Convert content to string for easy manipulation
	contentStr := string(*content)

	// Create the search string for the key
	searchStr := fmt.Sprintf("%s=", key)

	// Find the position of the key in the content
	pos := strings.Index(contentStr, searchStr)
	if pos == -1 {
		// Key not found
		fmt.Printf("Key not found: %s\n", key)
		return
	}

	// Find the end position of the value (comma or end of line)
	endPos := strings.Index(contentStr[pos:], ",")
	if endPos == -1 {
		// If there is no comma, check if it's at the end of the content
		endPos = len(contentStr) - pos
	}

	// If addQuotes is true and the key requires quotes, add quotes around the value
	if addQuotes {
		value = fmt.Sprintf(`"%s"`, value)
	}

	// Calculate the positions in the byte slice
	start := pos + len(searchStr)
	end := pos + endPos

	// Ensure the end position is within the slice bounds
	if end > len(*content) {
		end = len(*content)
	}

	// Update the content slice in place
	*content = append((*content)[:start], append([]byte(value), (*content)[end:]...)...)
}