com.foosbar.mailsnag.smtp.Server.java Source code

Introduction

Here is the source code for com.foosbar.mailsnag.smtp.Server.java

Source

/*******************************************************************************
 * Copyright (c) 2010-2013 Foos-Bar.com
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 * Kevin Kelley - initial API and implementation
 * Enrico - Server & Message Event Listeners
 *******************************************************************************/
package com.foosbar.mailsnag.smtp;

import java.io.IOException;
import java.net.BindException;
import java.net.ServerSocket;
import java.util.HashSet;
import java.util.Set;

import org.eclipse.jface.preference.IPreferenceStore;

import com.foosbar.mailsnag.Activator;
import com.foosbar.mailsnag.events.ServerStateListener;
import com.foosbar.mailsnag.preferences.PreferenceConstants;

public class Server implements Runnable {

    private ServerState status;
    private ServerSocket serverSocket;

    private final Set<ServerStateListener> listeners = new HashSet<ServerStateListener>();

    public Server() {
        status = ServerState.STOPPED;
    }

    public void run() {

        setStatus(ServerState.STARTING);

        // Get Preferences
        IPreferenceStore pStore = Activator.getDefault().getPreferenceStore();

        // Port to listen on.
        int port = pStore.getInt(PreferenceConstants.PARAM_PORT);

        try {
            serverSocket = new ServerSocket(port);

            setStatus(ServerState.LISTENING);

            if (Activator.isDebugMode()) {
                System.out.println("MailSnag Server listening on port " + port);
            }
        } catch (BindException e) {
            System.err.println(String.format(Activator.getResourceBundle().getString("exception.port.bind"), port));
            // Make sure server is marked as stopped
            setStatus(ServerState.STOPPED);
            // Exit thread execution
            throw new RuntimeException(e);
        } catch (IOException e) {
            // Print error to System.err
            e.printStackTrace(System.err);
            // Make sure server is marked as stopped
            setStatus(ServerState.STOPPED);
            // Exit thread execution
            throw new RuntimeException(e);
        }

        try {
            while (ServerState.LISTENING == status) {
                new MailHandler(serverSocket.accept()).start();
            }
        } catch (IOException e) {
            if (Activator.isDebugMode()) {
                // Print error to System.err
                e.printStackTrace(System.err);
            }
        } finally {
            close();
        }
    }

    public void close() {
        try {
            if (serverSocket != null && !serverSocket.isClosed()) {

                if (Activator.isDebugMode()) {
                    System.out.println("Shutting down MailSnag Server");
                }

                serverSocket.close();
            }
        } catch (Exception e) {
            if (Activator.isDebugMode()) {
                e.printStackTrace(System.err);
            }
        } finally {
            setStatus(ServerState.STOPPED);
        }
    }

    public ServerState getStatus() {
        return status;
    }

    private void setStatus(ServerState status) {
        this.status = status;
        for (ServerStateListener listener : listeners) {
            listener.serverStateChanged(status);
        }
    }

    public void addServerStateListener(ServerStateListener listener) {
        listeners.add(listener);
    }

    public void removeServerStateListener(ServerStateListener listener) {
        listeners.remove(listener);
    }
}