/*
 * Copyright (c) 2004-2008, KNOPFLERFISH project
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following
 * conditions are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 *
 * - 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.
 *
 * - Neither the name of the KNOPFLERFISH project nor the names of its
 *   contributors may be used to endorse or promote products derived
 *   from this software without specific prior written permission.
 *
 * 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 org.knopflerfish.bundle.trayicon;

import org.osgi.framework.*;
import com.jeans.trayicon.*;
// import java.awt.*;   --- Can not use this because it TrayIcon collision in J2SE 6.0
import java.awt.Component;
import java.awt.Frame;
import java.awt.Image;
import java.awt.Point;
import java.awt.Toolkit;
import java.awt.event.*;
import java.awt.image.ImageObserver;
import java.io.*;
import java.net.URL;
import javax.swing.JPopupMenu;
import javax.swing.MenuElement;
import javax.swing.JMenuItem;
import javax.swing.JMenu;
import javax.swing.JCheckBoxMenuItem;
import javax.swing.event.*;

import org.knopflerfish.service.trayicon.*;

public class TrayIconWrapper {

  WindowsTrayIcon windowsTrayIcon;
  TrayIconManager manager;
  TrayIcon        trayIcon;

  public TrayIconWrapper(TrayIcon trayIcon) {
    this.manager  = Activator.manager;
    this.trayIcon = trayIcon;
    if(trayIcon == null) {
      throw new IllegalArgumentException("TrayIcon argument cannot be null");
    }
  }

  SwingTrayPopup swingPopup = null;
  JPopupMenu     jPopup     = null;

  Frame frame;

  public void open() {
    if(windowsTrayIcon != null) {
      Activator.log.info("id=" + trayIcon.getId() + " already open");
      return;
    }

    if(WindowsTrayIcon.isRunning(trayIcon.getId())) {
      Activator.log.warn("id=" + trayIcon.getId() + " already running");
    }

    try {

      Activator.log.info("open trayicon id=" + trayIcon.getId() +
                         ", name=" + trayIcon.getName());

      frame = new Frame("") {
          public Point getLocationOnScreen() {
            return new Point(0,0);
          }
        };

      WindowsTrayIcon.initTrayIcon(trayIcon.getId());

      Image img = loadImage(trayIcon.getImageURL());

      windowsTrayIcon = new WindowsTrayIcon(img,
                                            img.getWidth(null),
                                            img.getHeight(null));

      manager.logErr();

      windowsTrayIcon.setToolTipText(trayIcon.getName());
      manager.logErr();

      windowsTrayIcon.addMouseListener(trayIcon);
      manager.logErr();

      windowsTrayIcon.addActionListener(trayIcon);
      manager.logErr();

      /*
      windowsTrayIcon.addBalloonListener(trayIcon);
      manager.logErr();
      */


      jPopup = trayIcon.getTrayJPopupMenu();

      if(jPopup != null) {

        try {
          String javaVersion = System.getProperty("java.version");
          if(javaVersion.startsWith("1.3")) {
            Activator.log.info("workaround for java 1.3 menues");
            if(true) {
              TrayIconPopup popup = makeTrayIconPopup(trayIcon.getTrayJPopupMenu(), null);

              windowsTrayIcon.setPopup(popup);
            } else {

              windowsTrayIcon.addMouseListener(new MouseAdapter() {
                  public void mousePressed(MouseEvent ev) {
                    Component comp = windowsTrayIcon.getDummyComponent();
                    jPopup.show(comp,
                                ev.getX(),
                                ev.getY() - jPopup.getSize().height);
                  }
                });
            }

          } else {
            swingPopup = new SwingTrayPopup();

            MenuElement[] subMenus= jPopup.getSubElements();

            // Convert top level of menu to something the original
            // code understands.

            for(int i = 0; i < subMenus.length; i++) {
              if(subMenus[i] instanceof JMenuItem) {
                swingPopup.add((JMenuItem)subMenus[i]);
              }
            }

            swingPopup.setTrayIcon(windowsTrayIcon);
          }
        } catch (Exception e) {
          Activator.log.error("Failed to set menu");
        }
      }

      windowsTrayIcon.setVisible(true);
      manager.logErr();

      String startupMsg = trayIcon.getStartupMessage();
      if(startupMsg != null) {
        if(WindowsTrayIcon.supportsBalloonMessages()) {
          windowsTrayIcon.showBalloon(startupMsg,
                                      trayIcon.getName(),
                                      5 * 1000,
                                      WindowsTrayIcon.BALLOON_INFO);

        }
      }

      sendEvent(new TrayEvent(TrayEvent.OPENED));
    } catch (Exception e) {
      Activator.log.error("Failed to set up TrayIcon", e);
    }
  }

  void sendEvent(TrayEvent ev) {
    try {
      trayIcon.trayEvent(ev);
    } catch (Exception e) {
      Activator.log.error("event send failed", e);
    }
  }

  void close() {
    Activator.log.info("TrayIconWrapper.close() trayicon id=" + trayIcon.getId());
    if(trayIcon != null) {
      sendEvent(new TrayEvent(TrayEvent.CLOSED));
    }

    if(windowsTrayIcon != null) {
      Activator.log.info(" call freeIcon");
      windowsTrayIcon.freeIcon();
      Activator.log.info(" called freeIcon");
    }
    Activator.log.info("TrayIconWrapper.closed trayicon id=" + trayIcon.getId());

    trayIcon        = null;
    windowsTrayIcon = null;
  }

  Object imageLock = new Object();
  boolean bGotImage = false;

  Image loadImage(URL url) throws IOException {
    synchronized(imageLock) {
      Toolkit tk = Toolkit.getDefaultToolkit();
      Image img = tk.createImage(loadURL(url));

      img.getHeight(new ImageObserver() {
          public boolean imageUpdate(Image img, int infoflags, int x, int y, int width, int height) {
            if(0 != (infoflags & ImageObserver.HEIGHT)) {
              synchronized(imageLock) {
                bGotImage = true;
                imageLock.notifyAll();
              }
              return true;
            }
            return false;
          }
        });


      if(!bGotImage) {
        try {
          imageLock.wait(1000 * 10);
        } catch (InterruptedException e) {
          throw new IOException("Image load timed out: " + e);
        }
      }
      return img;
    }

  }

  TrayIconPopup makeTrayIconPopup(MenuElement el,
                                  String name) {

    if(el == null) {
      return null;
    }

    TrayIconPopup popup =
      name != null
      ? new TrayIconPopup(name)
      : new TrayIconPopup();

    MenuElement[] subMenus= el.getSubElements();

    /*
    System.out.println("wrapping " + el.getClass().getName() + ", name=" + name +
                       ", length=" + subMenus.length);
    */

    for(int i = 0; i < subMenus.length; i++) {
      if(subMenus[i] instanceof JCheckBoxMenuItem) {
        popup.addMenuItem(new TrayIconPopupCheckItemW((JCheckBoxMenuItem)subMenus[i]));
      } else if(subMenus[i] instanceof JMenu) {
        JMenu menu = (JMenu)subMenus[i];
        MenuElement[] subMenus2 = menu.getSubElements();
        if(subMenus2.length == 1) {
          popup.addMenuItem(makeTrayIconPopup(subMenus2[0],
                                              menu.getText()));
        }
      } else if(subMenus[i] instanceof JPopupMenu) {
        JPopupMenu menu = (JPopupMenu)subMenus[i];
        popup.addMenuItem(makeTrayIconPopup(menu, name));
      } else if(subMenus[i] instanceof JMenuItem) {
        popup.addMenuItem(new TrayIconPopupSimpleItemW((JMenuItem)subMenus[i]));
      } else {
        //      System.out.println("skip class=" + subMenus[i].getClass().getName());
      }
    }

    return popup;
  }

  /**
   * Load an URL into a byte array.
   */
  public static byte [] loadURL(URL url) throws IOException {
    byte [] buf     = new byte[1024];

    ByteArrayOutputStream bout = new ByteArrayOutputStream();
    BufferedInputStream   in   = new BufferedInputStream(url.openStream());
    int n;
    while ((n = in.read(buf)) > 0) {
      bout.write(buf, 0, n);
    }
    try { in.close(); } catch (Exception ignored) { }
    return bout.toByteArray();
  }

  public String toString() {
    return
      "TrayIconWrapper[" +
      "trayIcon=" + trayIcon +
      "windowsTrayIcon=" + windowsTrayIcon +
      "]";
  }
}


class TrayIconPopupSimpleItemW extends TrayIconPopupSimpleItem {
  JMenuItem jItem;

  TrayIconPopupSimpleItemW(JMenuItem jItem) {
    super(jItem.getText());

    this.jItem = jItem;

    //    System.out.println("wrapped JMenuItem " + jItem.getText());

    try {
      ActionListener[] la = (ActionListener[])jItem.getListeners(ActionListener.class);
      for(int i = 0; la != null && i < la.length; i++) {
        addActionListener(la[i]);
      }
    } catch (Exception e) {
      Activator.log.error("Failed to get actionlisteners", e);
    }

  }

  public String getName() {
    return jItem.getText();
  }
}

class TrayIconPopupCheckItemW extends TrayIconPopupCheckItem implements ChangeListener {
  JCheckBoxMenuItem jItem;

  TrayIconPopupCheckItemW(JCheckBoxMenuItem jItem) {
    super(jItem.getText());

    this.jItem = jItem;

    try {
      ActionListener[] la = (ActionListener[])jItem.getListeners(ActionListener.class);
      for(int i = 0; la != null && i < la.length; i++) {
        addActionListener(la[i]);
      }
    } catch (Exception e) {
      Activator.log.error("Failed to get actionlisteners", e);
    }


    jItem.addChangeListener(this);
  }

  public void stateChanged(ChangeEvent e) {
    //    System.out.println(getName() + ": change " + e);
    super.setCheck(getCheck());
  }

  public String getName() {
    return jItem.getText();
  }

  public boolean getCheck() {
    return jItem.getState();
  }

  public void setCheck(boolean b) {
    super.setCheck(b);
    jItem.setState(b);
  }
}