/*
 * Copyright (c) 2017, Adam <Adam@sigterm.info>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice, this
 *    list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright notice,
 *    this list of conditions and the following disclaimer in the documentation
 *    and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
package net.runelite.client.plugins.account;

import com.google.common.eventbus.Subscribe;
import java.awt.event.ActionEvent;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.util.concurrent.ScheduledExecutorService;
import javax.imageio.ImageIO;
import javax.inject.Inject;
import javax.swing.JButton;
import lombok.extern.slf4j.Slf4j;
import net.runelite.api.events.SessionClose;
import net.runelite.client.account.AccountSession;
import net.runelite.client.account.SessionManager;
import net.runelite.api.events.SessionOpen;
import net.runelite.client.plugins.Plugin;
import net.runelite.client.plugins.PluginDescriptor;
import net.runelite.client.ui.ClientUI;
import net.runelite.client.ui.TitleToolbar;
import net.runelite.client.util.RunnableExceptionLogger;

@PluginDescriptor(
	name = "Account",
	loadWhenOutdated = true
)
@Slf4j
public class AccountPlugin extends Plugin
{
	@Inject
	private SessionManager sessionManager;

	@Inject
	private ClientUI ui;

	@Inject
	private ScheduledExecutorService executor;

	private JButton loginButton;
	private JButton logoutButton;

	private static final BufferedImage LOGIN_IMAGE, LOGOUT_IMAGE;

	static
	{
		try
		{
			synchronized (ImageIO.class)
			{
				LOGIN_IMAGE = ImageIO.read(AccountPlugin.class.getResourceAsStream("login_icon.png"));
				LOGOUT_IMAGE = ImageIO.read(AccountPlugin.class.getResourceAsStream("logout_icon.png"));
			}
		}
		catch (IOException e)
		{
			throw new RuntimeException(e);
		}
	}

	@Override
	protected void startUp() throws Exception
	{
		loginButton = new JButton();
		loginButton.setToolTipText("Login");
		loginButton.addActionListener(this::loginClick);

		logoutButton = new JButton();
		logoutButton.setToolTipText("Logout");
		logoutButton.addActionListener(this::logoutClick);

		addAndRemoveButtons();
	}

	private void addAndRemoveButtons()
	{
		TitleToolbar tb = ui.getTitleToolbar();
		tb.remove(loginButton);
		tb.remove(logoutButton);
		if (sessionManager.getAccountSession() == null)
		{
			tb.addButton(loginButton, LOGIN_IMAGE, LOGIN_IMAGE);
		}
		else
		{
			tb.addButton(logoutButton, LOGOUT_IMAGE, LOGOUT_IMAGE);
		}
	}

	@Override
	protected void shutDown() throws Exception
	{
		ui.getTitleToolbar().remove(loginButton);
		ui.getTitleToolbar().remove(logoutButton);
	}

	private void loginClick(ActionEvent ae)
	{
		executor.execute(RunnableExceptionLogger.wrap(sessionManager::login));
	}

	private void logoutClick(ActionEvent ae)
	{
		sessionManager.logout();
	}

	@Subscribe
	public void onSessionClose(SessionClose e)
	{
		addAndRemoveButtons();
	}

	@Subscribe
	public void onSessionOpen(SessionOpen sessionOpen)
	{
		AccountSession session = sessionManager.getAccountSession();

		if (session.getUsername() == null)
		{
			return; // No username yet
		}

		log.debug("Session opened as {}", session.getUsername());

		addAndRemoveButtons();
	}

}

@@ -1700,24 +1700,56 @@
 ng.JButton;%0A
+import javax.swing.JOptionPane;%0A
 import lombo
@@ -3876,16 +3876,182 @@
  ae)%0A%09%7B%0A
+%09%09if (JOptionPane.YES_OPTION == JOptionPane.showConfirmDialog(ui,%0A%09%09%09%09%22Are you sure you want to logout?%22, %22Logout Confirmation%22,%0A%09%09%09%09JOptionPane.YES_NO_OPTION))%0A%09%09%7B%0A%09
 %09%09sessio
@@ -4065,24 +4065,28 @@
 r.logout();%0A
+%09%09%7D%0A
 %09%7D%0A%0A%09@Subscr

Add confirmation dialog to runelite logout button