Bug Fixes

Author: CallOfCreator

NewMod/Buttons/Visionary/CaptureButton.cs

@@ -57,6 +57,7 @@ protected override void OnClick()
         {
             var timestamp = System.DateTime.UtcNow.ToString("yyyy-MM-dd_HH-mm-ss");
             string path = Path.Combine(VisionaryUtilities.ScreenshotDirectory, $"screenshot_{timestamp}.png");
+            NewMod.Instance.Log.LogMessage($"Screenshot will be saved to: {path}");
             Coroutines.Start(Utils.CaptureScreenshot(path));
         }
 

NewMod/Components/ScreenEffects/ShadowFluxEffect.cs

@@ -7,20 +7,26 @@ namespace NewMod.Components.ScreenEffects
     [RegisterInIl2Cpp]
     public class ShadowFluxEffect(IntPtr ptr) : MonoBehaviour(ptr)
     {
-        public Texture2D texture = NewModAsset.NoiseTex.LoadAsset();
         public float noiseScale = 2f;
         public float speed = 0.3f;
         public float edgeWidth = 0.25f;
         public float threshold = 0.55f;
         public float opacity = 0.75f;
         public float darkness = 0.8f;
         public Color tint = Color.white;
-        public static Shader _shader = NewModAsset.ShadowFluxShader.LoadAsset();
         public Material _mat;
 
         public void OnEnable()
         {
-            _mat = new Material(_shader) { hideFlags = HideFlags.DontSave };
+            var shader = NewModAsset.ShadowFluxShader.LoadAsset();
+            var texture = NewModAsset.NoiseTex.LoadAsset();
+
+            if (shader == null)
+            {
+                NewMod.Instance.Log.LogError("ShadowFluxEffect - Shader null");
+            }
+
+            _mat = new Material(shader) { hideFlags = HideFlags.DontSave };
             _mat.SetTexture("_NoiseTex", texture);
         }
         public void OnDisable()

NewMod/Components/ShadowZone.cs

@@ -54,6 +54,9 @@ public void Update()
                 if (active && lp.PlayerId == shadeId)
                 {
                     lp.cosmetics.SetPhantomRoleAlpha(1);
+                    lp.cosmetics.ToggleHat(true); 
+                    lp.cosmetics.ToggleVisor(true); 
+                    lp.cosmetics.TogglePet(true); 
                     lp.cosmetics.nameText.gameObject.SetActive(true);
 
                     if (killButton.currentTarget)
@@ -83,6 +86,9 @@ public void Update()
                     if (mode is ShadeOptions.ShadowMode.Invisible or ShadeOptions.ShadowMode.Both)
                     {
                         lp.cosmetics.SetPhantomRoleAlpha(0);
+                        lp.cosmetics.ToggleHat(false); 
+                        lp.cosmetics.ToggleVisor(false);
+                        lp.cosmetics.TogglePet(false); 
                         lp.cosmetics.nameText.gameObject.SetActive(false);
                     }
 
@@ -122,6 +128,9 @@ public void Update()
                 if (lp.PlayerId == shadeId)
                 {
                     lp.cosmetics.SetPhantomRoleAlpha(1);
+                    lp.cosmetics.ToggleHat(true); 
+                    lp.cosmetics.TogglePet(false); 
+                    lp.cosmetics.ToggleVisor(false); 
                     lp.cosmetics.nameText.gameObject.SetActive(true);
 
                     if (killButton.currentTarget)

NewMod/Patches/HudPatch.cs

@@ -25,6 +25,7 @@
 using NewMod.Roles;
 using NewMod.Components;
 using NewMod.Buttons.WraithCaller;
+using System.IO;
 
 namespace NewMod.Utilities
 {
@@ -799,8 +800,11 @@ public static IEnumerator CaptureScreenshot(string filePath)
 
             HudManager.Instance.SetHudActive(PlayerControl.LocalPlayer, PlayerControl.LocalPlayer.Data.Role, false);
             SoundManager.Instance.PlaySound(clip, false, 1f, null);
-            ScreenCapture.CaptureScreenshot(filePath, 4);
-            NewMod.Instance.Log.LogInfo($"Capturing screenshot at {System.IO.Path.GetFileName(filePath)}.");
+            yield return new WaitForEndOfFrame();
+            var tex = ScreenCapture.CaptureScreenshotAsTexture(4);
+            File.WriteAllBytes(filePath, tex.EncodeToPNG());
+            Object.Destroy(tex);
+            NewMod.Instance.Log.LogInfo($"Capturing screenshot at {Path.GetFileName(filePath)}.");
 
             yield return new WaitForSeconds(0.2f);
 
@@ -829,10 +833,6 @@ public static IEnumerator StartFeignDeath(PlayerControl player)
 
             SoundManager.Instance.PlaySound(clip, false, 1f, null);
 
-            if (player.AmOwner)
-            {
-                HudManager.Instance.SetHudActive(player, player.Data.Role, false);
-            }
             yield return new WaitForSeconds(0.5f);
 
             var body = player.GetNearestDeadBody(15f);
@@ -847,6 +847,11 @@ public static IEnumerator StartFeignDeath(PlayerControl player)
 
             Coroutines.Start(CoroutinesHelper.CoNotify("<color=green>You are now feigning death.\nYou will be revived in 10 seconds if unreported.</color>"));
 
+            if (player.AmOwner)
+            {
+                HudManager.Instance.SetHudActive(player, player.Data.Role, false);
+            }
+
             float timer = 10f;
             while (timer > 0)
             {

NewMod/Utilities/Utils.cs

@@ -1,5 +1,4 @@
 using System;
-using System.Collections.Generic;
 using System.IO;
 using System.Collections;
 using UnityEngine;
@@ -33,11 +32,21 @@ public static string ScreenshotDirectory
         {
             get
             {
-                string directory = OperatingSystem.IsAndroid() ? Environment.GetEnvironmentVariable("STAR_DATA_PATH") : Path.Combine(Application.persistentDataPath, "NewMod", "Screenshots");
-                if (!Directory.Exists(directory))
+                string basePath;
+
+                if (OperatingSystem.IsAndroid())
+                {
+                    basePath = Environment.GetEnvironmentVariable("STAR_DATA_PATH")!;
+                }
+                else
                 {
-                    Directory.CreateDirectory(directory);
+                    basePath = Application.persistentDataPath;
                 }
+
+                string directory = Path.Combine(basePath, "NewMod", "Screenshots");
+
+                Directory.CreateDirectory(directory);
+
                 return directory;
             }
         }
@@ -151,25 +160,6 @@ public static IEnumerator ShowScreenshot(Sprite sprite, DateTime timestamp, floa
             _showing = false;
         }
 
-        // <summary>
-        /// Loads and displays a screenshot from a given file path.  
-        /// If the file does not exist, no action is taken.
-        /// </summary>
-        /// <param name="filePath">The full path of the screenshot file to display.</param>
-        /// <param name="displayDuration">The duration, in seconds, to display the screenshot.</param>
-        /// <returns>An IEnumerator coroutine for handling display.</returns>
-        public static IEnumerator ShowScreenshotByPath(string filePath, float displayDuration)
-        {
-            if (!File.Exists(filePath)) yield break;
-
-            byte[] data = File.ReadAllBytes(filePath);
-            Texture2D tex = new(2, 2);
-            tex.LoadImage(data);
-            Sprite screenshotSprite = Sprite.Create(tex, new Rect(0, 0, tex.width, tex.height), new Vector2(0.5f, 0.5f));
-
-            yield return ShowScreenshot(screenshotSprite, File.GetCreationTime(filePath), displayDuration);
-        }
-
         /// <summary>
         /// Deletes all screenshots from the Visionary screenshot directory.
         /// </summary>