com.github.apetrelli.scafa.systray.ScafaSystrayMain Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of scafa-systray Show documentation
Management of profiles of Scafa inside SysTray.
There is a newer version: 1.0.0-alpha-3
/**
 * Scafa - A universal non-caching proxy for the road warrior
 * Copyright (C) 2015  Antonio Petrelli
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see .
 */
package com.github.apetrelli.scafa.systray;

import java.awt.AWTException;
import java.awt.SystemTray;
import java.awt.event.ItemEvent;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;

import javax.imageio.ImageIO;
import javax.swing.JCheckBoxMenuItem;
import javax.swing.JMenu;
import javax.swing.JMenuItem;
import javax.swing.JPopupMenu;

import com.github.apetrelli.scafa.ScafaLauncher;
import com.github.apetrelli.scafa.jxtrayicon.JXTrayIcon;

public class ScafaSystrayMain {

    private static final Logger LOG = Logger.getLogger(ScafaSystrayMain.class.getName());

    public static void main(String[] args) {
        prepareTrayIcon();

    }

    private static void prepareTrayIcon() {
        if (SystemTray.isSupported()) {
            BufferedImage image = null;
            try (InputStream is = ScafaSystrayMain.class.getResourceAsStream("/scafa.png")) {
                image = ImageIO.read(is);
            } catch (IOException e) {
                LOG.log(Level.SEVERE, "Cannot load systray image", e);
                System.exit(1);
            }
            JXTrayIcon trayIcon = createIcon(image);
            try {
                SystemTray.getSystemTray().add(trayIcon);
            } catch (AWTException e) {
                LOG.log(Level.SEVERE, "Cannot add systray image", e);
                System.exit(1);
            }
        }
    }

    private static JXTrayIcon createIcon(BufferedImage image) {
        JXTrayIcon trayIcon = new JXTrayIcon(image);
        trayIcon.setToolTip("Scafa");
        JPopupMenu menu = new JPopupMenu();
        JMenu profilesMenu = new JMenu("Profiles");
        ScafaLauncher launcher = new ScafaLauncher();
        launcher.initialize();
        String[] profiles = launcher.getProfiles();
        List profileItems = new ArrayList(profiles.length);
        String lastUsedProfile = launcher.getLastUsedProfile();
        for (int i = 0; i < profiles.length; i++) {
            String profile = profiles[i];
            JCheckBoxMenuItem item = new JCheckBoxMenuItem(profile);
            profileItems.add(item);
            profilesMenu.add(item);
            if (lastUsedProfile.equals(profile)) {
                item.setSelected(true);
                launcher.launch(profile);
            }
            item.addItemListener(t -> {
                if (t.getStateChange() == ItemEvent.SELECTED) {
                    launcher.stop();
                    profileItems.stream().filter(u -> {return u != item;}).forEach(u -> u.setSelected(false));
                    launcher.launch(profile);
                    launcher.saveLastUsedProfile(profile);
                }
            });
        }
        menu.add(profilesMenu);
        menu.addSeparator();
        JMenuItem exit = new JMenuItem("Exit");
        exit.addActionListener(t -> {
            launcher.stop();
            SystemTray.getSystemTray().remove(trayIcon);
            System.exit(0);
        });
        menu.add(exit);
        trayIcon.setJPopupMenu(menu);
        menu.setInvoker(null);
        return trayIcon;
    }

}