/*

 *   SONEWS News Server

 *   see AUTHORS for the list of contributors

 *

 *   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 <http://www.gnu.org/licenses/>.

 */

package org.sonews.web;

import java.io.BufferedReader;

import java.io.IOException;

import java.io.InputStreamReader;

import java.io.PrintWriter;

import java.net.Socket;

import javax.servlet.http.HttpServlet;

import org.sonews.util.StringTemplate;

import org.sonews.util.io.Resource;

/**

 * Base class for all sonews servlets.

 * @author Christian Lins

 * @since sonews/0.5.0

 */

public class AbstractSonewsServlet extends HttpServlet

{

  public static final String TemplateRoot = "org/sonews/web/tmpl/";

  protected String        hello = null;

  private BufferedReader  in     = null;

  private PrintWriter     out    = null;

  private Socket          socket = null;

  protected void connectToNewsserver()

    throws IOException

  {

    // Get sonews port from properties

    String port = System.getProperty("sonews.port", "9119");

    String host = System.getProperty("sonews.host", "localhost");

    try

    {

      this.socket = new Socket(host, Integer.parseInt(port));

      this.in     = new BufferedReader(

        new InputStreamReader(socket.getInputStream()));

      this.out = new PrintWriter(socket.getOutputStream());

      hello = in.readLine(); // Read hello message

    }

    catch(IOException ex)

    {

      System.out.println("sonews.host=" + host);

      System.out.println("sonews.port=" + port);

      System.out.flush();

      throw ex;

    }

  }

  protected void disconnectFromNewsserver()

  {

    try

    {

      printlnToNewsserver("QUIT");

      out.close();

      readlnFromNewsserver(); // Wait for bye message

      in.close();

      socket.close();

    }

    catch(IOException ex)

    {

      ex.printStackTrace();

    }

  }

  protected StringTemplate getTemplate(String res)

  {

    StringTemplate tmpl = new StringTemplate(

      Resource.getAsString(TemplateRoot + "AbstractSonewsServlet.tmpl", true));

    String content    = Resource.getAsString(TemplateRoot + res, true);

    String stylesheet = System.getProperty("sonews.web.stylesheet", "style.css");

    tmpl.set("CONTENT", content);

    tmpl.set("STYLESHEET", stylesheet);

    return new StringTemplate(tmpl.toString());

  }

  protected void printlnToNewsserver(final String line)

  {

    this.out.println(line);

    this.out.flush();

  }

  protected String readlnFromNewsserver()

    throws IOException

  {

    return this.in.readLine();

  }

}