From 9cb021dc743f6289b25f4f49fcfcc25f9ffbcfe1 Mon Sep 17 00:00:00 2001
From: Noah Pilarski <megapiggy@outerwildsmods.com>
Date: Fri, 16 Aug 2024 14:33:22 -0400
Subject: [PATCH] is modded/vanillla extensions

---
 Winch/Util/WinchExtensions.cs | 39 +++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/Winch/Util/WinchExtensions.cs b/Winch/Util/WinchExtensions.cs
index 88ad3a29..f95ac15f 100644
--- a/Winch/Util/WinchExtensions.cs
+++ b/Winch/Util/WinchExtensions.cs
@@ -17,15 +17,54 @@
 using UnityEngine.Localization.Tables;
 using UnityEngine.ResourceManagement.AsyncOperations;
 using UnityEngine.UI;
+using Winch.Components;
 using Winch.Core;
+using Winch.Data.Abilities;
 using Winch.Data.Character;
 using Winch.Data.Item;
+using Winch.Data.WorldEvent;
 using Winch.Util;
 using static ActiveAbilityInfoPanel;
 
 public static class WinchExtensions
 {
     #region DREDGE
+    /// <summary>
+    /// Check if the object is modded
+    /// </summary>
+    /// <returns>Whether this object is modded or not</returns>
+    public static bool IsModded(this ItemData item) => ItemUtil.ModdedItemDataDict.ContainsKey(item.id);
+    /// <inheritdoc cref="IsModded(ItemData)"/>
+    public static bool IsModded(this GridConfiguration gridConfiguration) => GridConfigUtil.ModdedGridConfigDict.ContainsKey(gridConfiguration.name);
+    /// <inheritdoc cref="IsModded(ItemData)"/>
+    public static bool IsModded(this SpeakerData speakerData) => speakerData is AdvancedSpeakerData;
+    /// <inheritdoc cref="IsModded(ItemData)"/>
+    public static bool IsModded(this Ability ability) => ability is ModdedAbility || ability.AbilityData.IsModded();
+    /// <inheritdoc cref="IsModded(ItemData)"/>
+    public static bool IsModded(this AbilityData ability) => ability is ModdedAbilityData;
+    /// <inheritdoc cref="IsModded(ItemData)"/>
+    public static bool IsModded(this WorldEvent worldEvent) => worldEvent is ModdedWorldEvent || worldEvent.worldEventData.IsModded();
+    /// <inheritdoc cref="IsModded(ItemData)"/>
+    public static bool IsModded(this WorldEventData worldEvent) => worldEvent is ModdedWorldEventData;
+
+    /// <summary>
+    /// Check if the object is vanilla
+    /// </summary>
+    /// <returns>Whether this object is vanilla or not</returns>
+    public static bool IsVanilla(this ItemData item) => !item.IsModded();
+    /// <inheritdoc cref="IsVanilla(ItemData)"/>
+    public static bool IsVanilla(this GridConfiguration gridConfiguration) => !gridConfiguration.IsModded();
+    /// <inheritdoc cref="IsVanilla(ItemData)"/>
+    public static bool IsVanilla(this SpeakerData speakerData) => !speakerData.IsModded();
+    /// <inheritdoc cref="IsVanilla(ItemData)"/>
+    public static bool IsVanilla(this Ability ability) => !ability.IsModded();
+    /// <inheritdoc cref="IsVanilla(ItemData)"/>
+    public static bool IsVanilla(this AbilityData ability) => !ability.IsModded();
+    /// <inheritdoc cref="IsVanilla(ItemData)"/>
+    public static bool IsVanilla(this WorldEvent worldEvent) => !worldEvent.IsModded();
+    /// <inheritdoc cref="IsVanilla(ItemData)"/>
+    public static bool IsVanilla(this WorldEventData worldEvent) => !worldEvent.IsModded();
+
     /// <summary>
     /// Converts a spatial item instances to a spatial item data.
     /// </summary>