/*

  *   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.storage.impl;

 import java.io.UnsupportedEncodingException;

 import java.sql.Connection;

 import java.sql.DriverManager;

 import java.sql.PreparedStatement;

 import java.sql.ResultSet;

 import java.sql.SQLException;

 import java.sql.Statement;

 import java.util.ArrayList;

 import java.util.Collections;

 import java.util.Date;

 import java.util.List;

 import java.util.logging.Level;

 import java.util.logging.Logger;

 import javax.mail.internet.MailDateFormat;

 import javax.mail.internet.MimeUtility;

 import org.apache.commons.codec.net.BCodec;

 import org.apache.commons.codec.net.QuotedPrintableCodec;

 import org.sonews.config.Config;

 import org.sonews.feed.Subscription;

 import org.sonews.storage.Article;

 import org.sonews.storage.ArticleHead;

 import org.sonews.storage.Group;

 import org.sonews.storage.Storage;

 import org.sonews.storage.StorageBackendException;

 import org.sonews.util.Pair;

 /**

  *

  * @author František Kučera (frantovo.cz)

  */

 public class DrupalDatabase implements Storage {

 	private static final Logger log = Logger.getLogger(DrupalDatabase.class.getName());

 	public static final String CHARSET = "UTF-8";

 	public static final String CRLF = "\r\n";

 	public static final int MAX_RESTARTS = 2;

 	/** How many times the database connection was reinitialized */

 	protected int restarts = 0;

 	protected Connection conn = null;

 	private QuotedPrintableCodec qpc = new QuotedPrintableCodec(CHARSET);

 	// TODO: správná doména

 	private String myDomain = "kinderporno.cz";

 	public DrupalDatabase() throws StorageBackendException {

 		connectDatabase();

 	}

 	private void connectDatabase() throws StorageBackendException {

 		try {

 			// Load database driver

 			String driverClass = Config.inst().get(Config.LEVEL_FILE, Config.STORAGE_DBMSDRIVER, "java.lang.Object");

 			Class.forName(driverClass);

 			// Establish database connection

 			String url = Config.inst().get(Config.LEVEL_FILE, Config.STORAGE_DATABASE, "<not specified>");

 			String username = Config.inst().get(Config.LEVEL_FILE, Config.STORAGE_USER, "root");

 			String password = Config.inst().get(Config.LEVEL_FILE, Config.STORAGE_PASSWORD, "");

 			conn = DriverManager.getConnection(url, username, password);

 			conn.setTransactionIsolation(Connection.TRANSACTION_SERIALIZABLE);

 			if (conn.getTransactionIsolation() != Connection.TRANSACTION_SERIALIZABLE) {

 				log.warning("Database is NOT fully serializable!");

 			}

 		} catch (Exception e) {

 			throw new StorageBackendException(e);

 		}

 	}

 	protected static void close(Connection connection, Statement statement, ResultSet resultSet) {

 		if (resultSet != null) {

 			try {

 				resultSet.close();

 			} catch (Exception e) {

 			}

 		}

 		if (statement != null) {

 			try {

 				statement.close();

 			} catch (Exception e) {

 			}

 		}

 		if (connection != null) {

 			try {

 				connection.close();

 			} catch (Exception e) {

 			}

 		}

 	}

 	/**

 	 * 

 	 * @param messageID <{0}-{1}-{2}@domain.tld> where {0} is nntp_id and {1} is group_id and {2} is group_name

 	 * @return array where [0] = nntp_id and [1] = group_id and [2] = group_name or returns null if messageID is invalid

 	 */

 	private static String[] parseMessageID(String messageID) {

 		if (messageID.matches("<[0-9]+\\-[0-9]+\\-[a-z0-9\\.]+@.+>")) {

 			return messageID.substring(1).split("@")[0].split("\\-");

 		} else {

 			return null;

 		}

 	}

 	private static Long parseArticleID(String messageID) {

 		String[] localPart = parseMessageID(messageID);

 		if (localPart == null) {

 			return null;

 		} else {

 			return Long.parseLong(localPart[0]);

 		}

 	}

 	private static Long parseGroupID(String messageID) {

 		String[] localPart = parseMessageID(messageID);

 		if (localPart == null) {

 			return null;

 		} else {

 			return Long.parseLong(localPart[1]);

 		}

 	}

 	private static String parseGroupName(String messageID) {

 		String[] localPart = parseMessageID(messageID);

 		if (localPart == null) {

 			return null;

 		} else {

 			return localPart[2];

 		}

 	}

 	private static String constructMessageId(int articleID, int groupID, String groupName, String domainName) {

 		StringBuilder sb = new StringBuilder();

 		sb.append("<");

 		sb.append(articleID);

 		sb.append("-");

 		sb.append(groupID);

 		sb.append("-");

 		sb.append(groupName);

 		sb.append("@");

 		sb.append(domainName);

 		sb.append(">");

 		return sb.toString();

 	}

 	/**

 	 * 

 	 * @param sb header list to be appended with new header. List must be terminated by line end.

 	 * @param key header name (without : and space)

 	 * @param value header value

 	 * @param encode true if value should be encoded/escaped before appending

 	 * @throws UnsupportedEncodingException 

 	 */

 	private static void addHeader(StringBuilder sb, String key, String value, boolean encode) throws UnsupportedEncodingException {

 		sb.append(key);

 		sb.append(": ");

 		if (encode) {

 			sb.append(MimeUtility.encodeWord(value));

 		} else {

 			sb.append(value);

 		}

 		sb.append(CRLF);

 	}

 	private String constructHeaders(ResultSet rs) throws SQLException, UnsupportedEncodingException {

 		StringBuilder sb = new StringBuilder();

 		addHeader(sb, "Message-id", constructMessageId(rs.getInt("id"), rs.getInt("group_id"), rs.getString("group_name"), myDomain), false);

 		addHeader(sb, "From", MimeUtility.encodeWord(rs.getString("sender_name")) + " <>", false);

 		addHeader(sb, "Subject", rs.getString("subject"), true);

 		/** TODO: správný formát data: */

 		addHeader(sb, "Date", MailDateFormat.getInstance().format(new Date(rs.getLong("created") * 1000)), false);

 		addHeader(sb, "Content-Type", "text/html; charset=" + CHARSET, false);

 		addHeader(sb, "Content-Transfer-Encoding", "quoted-printable", false);

 		//addHeader(sb, "Content-Transfer-Encoding", "base64", false);

 		Integer parentID = rs.getInt("parent_id");

 		if (parentID != null && parentID > 0) {

 			String parentMessageID = constructMessageId(parentID, rs.getInt("group_id"), rs.getString("group_name"), myDomain);

 			addHeader(sb, "In-Reply-To", parentMessageID, false);

 			addHeader(sb, "References", parentMessageID, false);

 		}

 		return sb.toString();

 	}

 	@Override

 	public List<Group> getGroups() throws StorageBackendException {

 		PreparedStatement ps = null;

 		ResultSet rs = null;

 		try {

 			ps = conn.prepareStatement("SELECT * FROM nntp_group");

 			rs = ps.executeQuery();

 			List<Group> skupiny = new ArrayList<Group>();

 			while (rs.next()) {

 				skupiny.add(new Group(rs.getString("name"), rs.getInt("id"), Group.READONLY));

 			}

 			return skupiny;

 		} catch (Exception e) {

 			throw new StorageBackendException(e);

 		} finally {

 			close(null, ps, rs);

 		}

 	}

 	@Override

 	public Group getGroup(String name) throws StorageBackendException {

 		PreparedStatement ps = null;

 		ResultSet rs = null;

 		try {

 			ps = conn.prepareStatement("SELECT * FROM nntp_group WHERE name = ?");

 			ps.setString(1, name);

 			rs = ps.executeQuery();

 			while (rs.next()) {

 				return new Group(rs.getString("name"), rs.getInt("id"), Group.READONLY);

 			}

 			return null;

 		} catch (Exception e) {

 			throw new StorageBackendException(e);

 		} finally {

 			close(null, ps, rs);

 		}

 	}

 	@Override

 	public boolean isGroupExisting(String groupname) throws StorageBackendException {

 		return getGroup(groupname) != null;

 	}

 	@Override

 	public Article getArticle(String messageID) throws StorageBackendException {

 		Long articleID = parseArticleID(messageID);

 		Long groupID = parseGroupID(messageID);

 		if (articleID == null || groupID == null) {

 			log.log(Level.SEVERE, "Invalid messageID: {0}", new Object[]{messageID});

 			return null;

 		} else {

 			return getArticle(articleID, groupID);

 		}

 	}

 	@Override

 	public Article getArticle(long articleID, long groupID) throws StorageBackendException {

 		PreparedStatement ps = null;

 		ResultSet rs = null;

 		try {

 			ps = conn.prepareStatement("SELECT * FROM nntp_article WHERE id = ? AND group_id = ?");

 			ps.setLong(1, articleID);

 			ps.setLong(2, groupID);

 			rs = ps.executeQuery();

 			if (rs.next()) {

 				String headers = constructHeaders(rs);

 				// TODO: fold?

 				BCodec bc = new BCodec(CHARSET);

 				byte[] body = qpc.encode(rs.getString("text")).getBytes();

 				//byte[] body = bc.encode(rs.getString("text")).getBytes();

 				return new Article(headers, body);

 			} else {

 				return null;

 			}

 		} catch (Exception e) {

 			throw new StorageBackendException(e);

 		} finally {

 			close(null, ps, rs);

 		}

 	}

 	@Override

 	public List<Pair<Long, ArticleHead>> getArticleHeads(Group group, long first, long last) throws StorageBackendException {

 		PreparedStatement ps = null;

 		ResultSet rs = null;

 		try {

 			// TODO: je nutné řazení?

 			ps = conn.prepareStatement("SELECT * FROM nntp_article WHERE group_id = ? AND id >= ? AND id <= ? ORDER BY id");

 			ps.setLong(1, group.getInternalID());

 			ps.setLong(2, first);

 			ps.setLong(3, last);

 			rs = ps.executeQuery();

 			List<Pair<Long, ArticleHead>> heads = new ArrayList<Pair<Long, ArticleHead>>();

 			while (rs.next()) {

 				String headers = constructHeaders(rs);

 				heads.add(new Pair<Long, ArticleHead>(rs.getLong("id"), new ArticleHead(headers)));

 			}

 			return heads;

 		} catch (Exception e) {

 			throw new StorageBackendException(e);

 		} finally {

 			close(null, ps, rs);

 		}

 	}

 	@Override

 	public List<Pair<Long, String>> getArticleHeaders(Group group, long start, long end, String header, String pattern) throws StorageBackendException {

 		log.log(Level.SEVERE, "TODO: getArticleHeaders {0} / {1} / {2} / {3} / {4}", new Object[]{group, start, end, header, pattern});

 		/** TODO: */

 		return Collections.emptyList();

 	}

 	@Override

 	public long getArticleIndex(Article article, Group group) throws StorageBackendException {

 		Long id = parseArticleID(article.getMessageID());

 		if (id == null) {

 			throw new StorageBackendException("Invalid messageID: " + article.getMessageID());

 		} else {

 			return id;

 		}

 	}

 	@Override

 	public List<Long> getArticleNumbers(long groupID) throws StorageBackendException {

 		PreparedStatement ps = null;

 		ResultSet rs = null;

 		try {

 			ps = conn.prepareStatement("SELECT id FROM nntp_article WHERE group_id = ?");

 			ps.setLong(1, groupID);

 			rs = ps.executeQuery();

 			List<Long> articleNumbers = new ArrayList<Long>();

 			while (rs.next()) {

 				articleNumbers.add(rs.getLong(1));

 			}

 			return articleNumbers;

 		} catch (Exception e) {

 			throw new StorageBackendException(e);

 		} finally {

 			close(null, ps, rs);

 		}

 	}

 	@Override

 	public int getFirstArticleNumber(Group group) throws StorageBackendException {

 		PreparedStatement ps = null;

 		ResultSet rs = null;

 		try {

 			ps = conn.prepareStatement("SELECT min(id) FROM nntp_article WHERE group_id = ?");

 			ps.setLong(1, group.getInternalID());

 			rs = ps.executeQuery();

 			rs.next();

 			return rs.getInt(1);

 		} catch (Exception e) {

 			throw new StorageBackendException(e);

 		} finally {

 			close(null, ps, rs);

 		}

 	}

 	@Override

 	public int getLastArticleNumber(Group group) throws StorageBackendException {

 		PreparedStatement ps = null;

 		ResultSet rs = null;

 		try {

 			ps = conn.prepareStatement("SELECT max(id) FROM nntp_article WHERE group_id = ?");

 			ps.setLong(1, group.getInternalID());

 			rs = ps.executeQuery();

 			rs.next();

 			return rs.getInt(1);

 		} catch (Exception e) {

 			throw new StorageBackendException(e);

 		} finally {

 			close(null, ps, rs);

 		}

 	}

 	@Override

 	public boolean isArticleExisting(String messageID) throws StorageBackendException {

 		Long articleID = parseArticleID(messageID);

 		Long groupID = parseGroupID(messageID);

 		if (articleID == null || groupID == null) {

 			return false;

 		} else {

 			PreparedStatement ps = null;

 			ResultSet rs = null;

 			try {

 				ps = conn.prepareStatement("SELECT count(*) FROM nntp_article WHERE id = ? AND group_id = ?");

 				ps.setLong(1, articleID);

 				ps.setLong(2, groupID);

 				rs = ps.executeQuery();

 				rs.next();

 				return rs.getInt(1) == 1;

 			} catch (Exception e) {

 				throw new StorageBackendException(e);

 			} finally {

 				close(null, ps, rs);

 			}

 		}

 	}

 	//

 	// --- zatím neimplementovat ---

 	//

 	@Override

 	public void addArticle(Article art) throws StorageBackendException {

 		log.log(Level.SEVERE, "TODO: addArticle {0}", new Object[]{art});

 	}

 	@Override

 	public void addEvent(long timestamp, int type, long groupID) throws StorageBackendException {

 		log.log(Level.SEVERE, "TODO: addEvent {0} / {1} / {2}", new Object[]{timestamp, type, groupID});

 	}

 	@Override

 	public void addGroup(String groupname, int flags) throws StorageBackendException {

 		log.log(Level.SEVERE, "TODO: addGroup {0} / {1}", new Object[]{groupname, flags});

 	}

 	@Override

 	public int countArticles() throws StorageBackendException {

 		PreparedStatement ps = null;

 		ResultSet rs = null;

 		try {

 			ps = conn.prepareStatement("SELECT count(*) FROM nntp_article");

 			rs = ps.executeQuery();

 			rs.next();

 			return rs.getInt(1);

 		} catch (Exception e) {

 			throw new StorageBackendException(e);

 		} finally {

 			close(null, ps, rs);

 		}

 	}

 	@Override

 	public int countGroups() throws StorageBackendException {

 		PreparedStatement ps = null;

 		ResultSet rs = null;

 		try {

 			ps = conn.prepareStatement("SELECT count(*) FROM nntp_group");

 			rs = ps.executeQuery();

 			rs.next();

 			return rs.getInt(1);

 		} catch (Exception e) {

 			throw new StorageBackendException(e);

 		} finally {

 			close(null, ps, rs);

 		}

 	}

 	@Override

 	public void delete(String messageID) throws StorageBackendException {

 		log.log(Level.SEVERE, "TODO: delete {0}", new Object[]{messageID});

 	}

 	@Override

 	public String getConfigValue(String key) throws StorageBackendException {

 		//log.log(Level.SEVERE, "TODO: getConfigValue {0}", new Object[]{key});

 		return null;

 	}

 	@Override

 	public int getEventsCount(int eventType, long startTimestamp, long endTimestamp, Group group) throws StorageBackendException {

 		log.log(Level.SEVERE, "TODO: getEventsCount {0} / {1} / {2} / {3}", new Object[]{eventType, startTimestamp, endTimestamp, group});

 		return 0;

 	}

 	@Override

 	public double getEventsPerHour(int key, long gid) throws StorageBackendException {

 		log.log(Level.SEVERE, "TODO: getEventsPerHour {0} / {1}", new Object[]{key, gid});

 		return 0;

 	}

 	@Override

 	public List<String> getGroupsForList(String listAddress) throws StorageBackendException {

 		log.log(Level.SEVERE, "TODO: getGroupsForList {0}", new Object[]{listAddress});

 		return Collections.emptyList();

 	}

 	@Override

 	public List<String> getListsForGroup(String groupname) throws StorageBackendException {

 		log.log(Level.SEVERE, "TODO: getListsForGroup {0}", new Object[]{groupname});

 		return Collections.emptyList();

 	}

 	@Override

 	public String getOldestArticle() throws StorageBackendException {

 		log.log(Level.SEVERE, "TODO: getOldestArticle");

 		return null;

 	}

 	@Override

 	public int getPostingsCount(String groupname) throws StorageBackendException {

 		PreparedStatement ps = null;

 		ResultSet rs = null;

 		try {

 			ps = conn.prepareStatement("SELECT count(*) FROM nntp_article WHERE group_name = ?");

 			ps.setString(1, groupname);

 			rs = ps.executeQuery();

 			rs.next();

 			return rs.getInt(1);

 		} catch (Exception e) {

 			throw new StorageBackendException(e);

 		} finally {

 			close(null, ps, rs);

 		}

 	}

 	@Override

 	public List<Subscription> getSubscriptions(int type) throws StorageBackendException {

 		log.log(Level.SEVERE, "TODO: getSubscriptions {0}", new Object[]{type});

 		return Collections.emptyList();

 	}

 	@Override

 	public void purgeGroup(Group group) throws StorageBackendException {

 		log.log(Level.SEVERE, "TODO: purgeGroup {0}", new Object[]{group});

 	}

 	@Override

 	public void setConfigValue(String key, String value) throws StorageBackendException {

 		log.log(Level.SEVERE, "TODO: setConfigValue {0} = {1}", new Object[]{key, value});

 	}

 	@Override

 	public boolean update(Article article) throws StorageBackendException {

 		log.log(Level.SEVERE, "TODO: update {0}", new Object[]{article});

 		throw new StorageBackendException("Not implemented yet.");

 	}

 	@Override

 	public boolean update(Group group) throws StorageBackendException {

 		log.log(Level.SEVERE, "TODO: update {0}", new Object[]{group});

 		throw new StorageBackendException("Not implemented yet.");

 	}

 }