Merge pull request #6 from RaazKH/fix-issue-5

fix issue 5

Author: RaazKH

src/main/java/com/compassCameraControl/CompassCameraControlPlugin.java

@@ -8,10 +8,13 @@
 import net.runelite.api.MenuAction;
 import net.runelite.api.SoundEffectID;
 import net.runelite.api.events.MenuOptionClicked;
+import net.runelite.api.events.MenuEntryAdded;
 import net.runelite.client.config.ConfigManager;
 import net.runelite.client.eventbus.Subscribe;
 import net.runelite.client.plugins.Plugin;
 import net.runelite.client.plugins.PluginDescriptor;
+import net.runelite.api.MenuEntry;
+import org.apache.commons.lang3.ArrayUtils;
 
 @Slf4j
 @PluginDescriptor(
@@ -32,24 +35,56 @@ public class CompassCameraControlPlugin extends Plugin
 	private static final int EAST_YAW = 512;
 	private static final int WEST_YAW = 1536;
 
+	private static final String SNAP_CARDINAL = "Snap Cardinal";
+	private static final String CYCLE_CARDINAL = "Cycle Cardinal";
+
 	@Subscribe
-	public void onMenuOptionClicked(MenuOptionClicked event)
+	public void onMenuEntryAdded(MenuEntryAdded event)
 	{
-		if (event.getMenuAction() == MenuAction.CC_OP && event.getMenuOption().equals("Look North"))
+		if (event.getOption().equals("Look North"))
 		{
+			String newOption;
 			switch (config.controlMode())
 			{
 				case SNAP_TO_CLOSEST:
-					alignYaw();
+					newOption = SNAP_CARDINAL;
 					break;
 
 				case CYCLE:
 				default:
+					newOption = CYCLE_CARDINAL;
+					break;
+			}
+
+			MenuEntry[] menuEntries = client.getMenu().getMenuEntries();
+			MenuEntry newEntry = client.getMenu().createMenuEntry(menuEntries.length - 1)
+					.setOption(newOption)
+					.setType(MenuAction.CC_OP);
+
+			menuEntries = ArrayUtils.add(menuEntries, newEntry);
+			client.getMenu().setMenuEntries(menuEntries);
+		}
+	}
+
+	@Subscribe
+	public void onMenuOptionClicked(MenuOptionClicked event)
+	{
+		if (event.getMenuAction() == MenuAction.CC_OP)
+		{
+			switch (event.getMenuOption())
+			{
+				case SNAP_CARDINAL:
+					alignYaw();
+					event.consume();
+					client.playSoundEffect(SoundEffectID.UI_BOOP);
+					break;
+
+				case CYCLE_CARDINAL:
 					cycleYaw();
+					event.consume();
+					client.playSoundEffect(SoundEffectID.UI_BOOP);
 					break;
 			}
-			event.consume();
-			client.playSoundEffect(SoundEffectID.UI_BOOP);
 		}
 	}