Copy paste to banquet for fools
This commit is contained in:
181
Projects/BanquetForFools/BanquetForCyka/UISnapshotPanelBuyAll.cs
Normal file
181
Projects/BanquetForFools/BanquetForCyka/UISnapshotPanelBuyAll.cs
Normal file
@@ -0,0 +1,181 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using HarmonyLib;
|
||||
using Snapshots;
|
||||
using UnityEngine;
|
||||
using UnityEngine.EventSystems;
|
||||
using UnityEngine.UI;
|
||||
|
||||
namespace BanquetForCyka {
|
||||
[HarmonyPatch]
|
||||
public class UISnapshotPanelBuyAll {
|
||||
public static UISnapshotPanel panel;
|
||||
public static Traverse panelTraverse;
|
||||
public static SnapshotLiveData selectedData;
|
||||
|
||||
[HarmonyPostfix]
|
||||
[HarmonyPatch(typeof(UISnapshotPanel), "OnPool")]
|
||||
public static void PostfixCreate(UISnapshotPanel __instance) {
|
||||
if (Main.debugBuyAll.Value)
|
||||
Console.WriteLine("UISnapshotPanel.OnPool: {0}", __instance);
|
||||
panel = __instance;
|
||||
panelTraverse = Traverse.Create(__instance);
|
||||
var placeButton = panelTraverse.Field("m_PlaceButton").GetValue<Button>();
|
||||
placeButton.gameObject.AddComponent<RightClickHandler>();
|
||||
|
||||
var swapButton = panelTraverse.Field("m_SwapButton").GetValue<Button>();
|
||||
swapButton.gameObject.AddComponent<RightClickHandlerSwap>();
|
||||
}
|
||||
|
||||
[HarmonyPostfix]
|
||||
[HarmonyPatch(typeof(UISnapshotPanel), "OnSelectedChanged")]
|
||||
public static void PostfixCreate(UISnapshotPanel __instance, ref SnapshotLiveData selectedData) {
|
||||
if (Main.debugBuyAll.Value)
|
||||
Console.WriteLine("UISnapshotPanel.OnSelectedChanged: {0}", __instance);
|
||||
UISnapshotPanelBuyAll.selectedData = selectedData;
|
||||
}
|
||||
}
|
||||
|
||||
public abstract class BaseRightClickHandler : MonoBehaviour, IPointerClickHandler {
|
||||
protected static Traverse m_TechAvailLookupTraverse;
|
||||
protected static Dictionary<Snapshot, TechDataAvailValidation> m_TechAvailLookup;
|
||||
|
||||
protected virtual void Awake() {
|
||||
var trav = Traverse.Create(Singleton.Manager<ManSnapshots>.inst);
|
||||
m_TechAvailLookupTraverse = trav.Field("m_TechAvailLookup");
|
||||
m_TechAvailLookup = m_TechAvailLookupTraverse.GetValue<Dictionary<Snapshot, TechDataAvailValidation>>();
|
||||
}
|
||||
|
||||
protected Dictionary<BlockTypes, int> CalculateMissingBlocks(
|
||||
Dictionary<BlockTypes, TechDataAvailValidation.BlockTypeAvailability> blockAvailability, bool isSpawning) {
|
||||
Dictionary<BlockTypes, int> missing = new Dictionary<BlockTypes, int>();
|
||||
|
||||
foreach (var kvp in blockAvailability) {
|
||||
int numMissing;
|
||||
if (isSpawning)
|
||||
numMissing = kvp.Value.numRequired - kvp.Value.numInInventory;
|
||||
else
|
||||
numMissing = kvp.Value.numRequired - kvp.Value.numInInventory - kvp.Value.numOnPlayerTech;
|
||||
}
|
||||
|
||||
if (missing.Count == 0)
|
||||
foreach (var kvp in blockAvailability) missing.Add(kvp.Key, kvp.Value.numRequired);
|
||||
return missing;
|
||||
}
|
||||
|
||||
protected int CalculateTotalCost(Dictionary<BlockTypes, int> missingBlocks) {
|
||||
int totalCost = 0;
|
||||
RecipeManager recipeManager = Singleton.Manager<RecipeManager>.inst;
|
||||
|
||||
foreach (var kvp in missingBlocks) {
|
||||
int cost = recipeManager.GetBlockBuyPrice(kvp.Key) * kvp.Value;
|
||||
totalCost += cost;
|
||||
|
||||
if (Main.debugBuyAll.Value)
|
||||
Console.WriteLine("{0} of {1} would cost {2}, total now {3}", kvp.Value, kvp.Key, cost, totalCost);
|
||||
}
|
||||
return totalCost;
|
||||
}
|
||||
|
||||
protected bool TryPurchaseBlocks(Dictionary<BlockTypes, int> missingBlocks, int totalCost) {
|
||||
ManPlayer player = Singleton.Manager<ManPlayer>.inst;
|
||||
if (player.GetCurrentMoney() < totalCost) {
|
||||
if (Main.debugBuyAll.Value)
|
||||
Console.WriteLine("Not enough money, have {0} but need {1}, nothing to do",
|
||||
player.GetCurrentMoney(), totalCost);
|
||||
return false;
|
||||
}
|
||||
|
||||
player.PayMoney(totalCost);
|
||||
foreach (var kvp in missingBlocks) {
|
||||
if (Main.debugBuyAll.Value)
|
||||
Console.WriteLine("Buying {0} of {1}", kvp.Value, kvp.Key);
|
||||
player.PlayerInventory.HostAddItem(kvp.Key, kvp.Value);
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
protected bool ProcessPurchase(
|
||||
Dictionary<BlockTypes, TechDataAvailValidation.BlockTypeAvailability> blockAvailability, bool isSpawning) {
|
||||
try {
|
||||
if (blockAvailability == null) {
|
||||
if (Main.debugBuyAll.Value)
|
||||
Console.WriteLine("Block availability is null (wtf?), nothing to do");
|
||||
return false;
|
||||
}
|
||||
|
||||
var missingBlocks = CalculateMissingBlocks(blockAvailability, isSpawning);
|
||||
int totalCost = CalculateTotalCost(missingBlocks);
|
||||
|
||||
if (totalCost > 0) {
|
||||
if (Main.debugBuyAll.Value)
|
||||
Console.WriteLine("Total cost: {0}", totalCost);
|
||||
return TryPurchaseBlocks(missingBlocks, totalCost);
|
||||
}
|
||||
|
||||
if (Main.debugBuyAll.Value)
|
||||
Console.WriteLine("No blocks missing or no cost calculated");
|
||||
return false;
|
||||
} catch (Exception e) {
|
||||
if (Main.debugBuyAll.Value)
|
||||
Console.WriteLine("Error during purchase processing: {0}", e);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
protected Dictionary<BlockTypes, TechDataAvailValidation.BlockTypeAvailability> GetCurrentBlockAvailability() {
|
||||
SnapshotLiveData selectedSnapshotData = UISnapshotPanelBuyAll.selectedData;
|
||||
Snapshot selectedSnapshot = selectedSnapshotData.m_Snapshot;
|
||||
if (selectedSnapshot == null) {
|
||||
if (Main.debugBuyAll.Value)
|
||||
Console.WriteLine("Selected snapshot is null wtf??: {0}", gameObject.name);
|
||||
return null;
|
||||
}
|
||||
|
||||
TechDataAvailValidation techDataAvail;
|
||||
if (!m_TechAvailLookup.TryGetValue(selectedSnapshot, out techDataAvail)) {
|
||||
if (Main.debugBuyAll.Value)
|
||||
Console.WriteLine("Failed to find TechDataAvailValidation for snapshot: {0}", selectedSnapshot);
|
||||
return null;
|
||||
}
|
||||
|
||||
var trav = Traverse.Create(techDataAvail);
|
||||
var mBlockAvailabilityField = trav.Field("m_BlockAvailability");
|
||||
var mBlockAvailability =
|
||||
mBlockAvailabilityField
|
||||
.GetValue<Dictionary<BlockTypes, TechDataAvailValidation.BlockTypeAvailability>>();
|
||||
|
||||
return mBlockAvailability;
|
||||
}
|
||||
|
||||
public abstract void OnPointerClick(PointerEventData eventData);
|
||||
}
|
||||
|
||||
public class RightClickHandler : BaseRightClickHandler {
|
||||
public override void OnPointerClick(PointerEventData eventData) {
|
||||
if (Main.debugBuyAll.Value)
|
||||
Console.WriteLine("UISnapshotPanel.OnPointerClick: {0} {1}", gameObject.name, eventData.button);
|
||||
try {
|
||||
if (eventData.button == PointerEventData.InputButton.Right)
|
||||
ProcessPurchase(GetCurrentBlockAvailability(), true);
|
||||
} catch (Exception e) {
|
||||
if (Main.debugBuyAll.Value)
|
||||
Console.WriteLine("Shit exploded fml: {0}", e);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public class RightClickHandlerSwap : BaseRightClickHandler {
|
||||
public override void OnPointerClick(PointerEventData eventData) {
|
||||
if (Main.debugBuyAll.Value)
|
||||
Console.WriteLine("UISnapshotPanel.OnPointerClick: {0} {1}", gameObject.name, eventData.button);
|
||||
try {
|
||||
if (eventData.button == PointerEventData.InputButton.Right)
|
||||
ProcessPurchase(GetCurrentBlockAvailability(), false);
|
||||
} catch (Exception e) {
|
||||
if (Main.debugBuyAll.Value)
|
||||
Console.WriteLine("Shit exploded fml: {0}", e);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user