/**

  * SQL-DK

  * Copyright © 2014 František Kučera (frantovo.cz)

  *

  * 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 cz.frantovo.ejabberd.auth;

 import java.io.DataInputStream;

 import java.io.DataOutputStream;

 import java.io.FileInputStream;

 import java.io.FileOutputStream;

 import java.io.IOException;

 import java.io.InputStream;

 import java.io.OutputStream;

 import java.io.PrintStream;

 import java.sql.Connection;

 import java.sql.DriverManager;

 import java.sql.PreparedStatement;

 import java.sql.ResultSet;

 import java.sql.SQLException;

 import java.util.Properties;

 /**

  *

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

  */

 public class EJabberdAuth {

 	public static final String NASTAVENÍ_DB_URL = "db.url";

 	public static final String NASTAVENÍ_DB_JMÉNO = "db.jmeno";

 	public static final String NASTAVENÍ_DB_HESLO = "db.heslo";

 	public static final String NASTAVENÍ_DOMÉNA = "domena";

 	public static final String NASTAVENÍ_LOG = "log";

 	public static final String VÝCHOZÍ_LOG = "/tmp/eJabberd-auth.log";

 	private static final int ANO = 1;

 	private static final int NE = 0;

 	private final DataInputStream in;

 	private final DataOutputStream out;

 	private final PrintStream err;

 	private final String našeDoména;

 	private final String dbUrl;

 	private final String dbJméno;

 	private final String dbHeslo;

 	public static void main(String[] args) throws IOException, ChybaZápisuVýsledku, ChybaČteníVstupu {

 		try (FileInputStream nastaveníFis = new FileInputStream(args[0])) {

 			Properties nastavení = new Properties();

 			nastavení.load(nastaveníFis);

 			try (FileOutputStream err = new FileOutputStream(nastavení.getProperty(NASTAVENÍ_LOG, VÝCHOZÍ_LOG))) {

 				EJabberdAuth a = new EJabberdAuth(System.in, System.out, err, nastavení);

 				a.start();

 			}

 		}

 	}

 	public EJabberdAuth(InputStream in, OutputStream out, OutputStream err, Properties nastavení) {

 		this.in = new DataInputStream(in);

 		this.out = new DataOutputStream(out);

 		this.err = new PrintStream(err);

 		this.našeDoména = nastavení.getProperty(NASTAVENÍ_DOMÉNA);

 		this.dbUrl = nastavení.getProperty(NASTAVENÍ_DB_URL);

 		this.dbJméno = nastavení.getProperty(NASTAVENÍ_DB_JMÉNO);

 		this.dbHeslo = nastavení.getProperty(NASTAVENÍ_DB_HESLO);

 	}

 	public void start() throws ChybaZápisuVýsledku, ChybaČteníVstupu {

 		while (true) {

 			String text = načtiText();

 			zpracuj(text);

 		}

 	}

 	private Connection getDB() throws SQLException {

 		return DriverManager.getConnection(dbUrl, dbJméno, dbHeslo);

 	}

 	private String načtiText() throws ChybaČteníVstupu {

 		try {

 			short délka = in.readShort();

 			byte[] bajty = new byte[délka];

 			in.read(bajty);

 			return new String(bajty);

 		} catch (Exception e) {

 			throw new ChybaČteníVstupu(e);

 		}

 	}

 	private void zpracuj(String text) throws ChybaZápisuVýsledku {

 		String[] prvky = text.split(":");

 		final String operace;

 		final String uživatel;

 		final String doména;

 		final String heslo;

 		if (prvky.length < 3) {

 			zapišVýsledek(NE);

 			err.println("Chybný počet parametrů: " + prvky.length);

 		} else {

 			operace = prvky[0];

 			uživatel = prvky[1];

 			doména = prvky[2];

 			heslo = (prvky.length > 3) ? prvky[3] : null;

 			err.println(operace + ":" + uživatel + ":" + doména + (heslo == null ? "" : ":*"));

 			switch (operace) {

 				case "auth":

 					ověřHeslo(uživatel, doména, heslo);

 					break;

 				case "isuser":

 					ověřExistenci(uživatel, doména);

 					break;

 				case "setpass":

 				case "tryregister":

 				case "removeuser":

 				case "removeuser3":

 				default:

 					zapišVýsledek(NE);

 			}

 		}

 	}

 	private void zapišVýsledek(int výsledek) throws ChybaZápisuVýsledku {

 		try {

 			err.println("výsledek: " + výsledek);

 			out.writeShort(2);

 			out.writeShort(výsledek);

 			out.flush();

 		} catch (Exception e) {

 			throw new ChybaZápisuVýsledku(e);

 		}

 	}

 	private void ověřHeslo(String uživatel, String doména, String heslo) throws ChybaZápisuVýsledku {

 		if (ověřDoménu(doména)) {

 			try (Connection db = getDB()) {

 				try (PreparedStatement ps = db.prepareStatement("SELECT over_heslo(?,?)")) {

 					int i = 1;

 					ps.setString(i++, uživatel);

 					ps.setString(i++, heslo);

 					try (ResultSet rs = ps.executeQuery()) {

 						rs.next();

 						int výsledek = rs.getBoolean(1) ? ANO : NE;

 						zapišVýsledek(výsledek);

 					}

 				}

 			} catch (SQLException e) {

 				logujSQLException(e);

 				zapišVýsledek(NE);

 			}

 		} else {

 			zapišVýsledek(NE);

 		}

 	}

 	private void ověřExistenci(String uživatel, String doména) throws ChybaZápisuVýsledku {

 		if (ověřDoménu(doména)) {

 			try (Connection db = getDB()) {

 				try (PreparedStatement ps = db.prepareStatement("SELECT over_existenci_uzivatele(?)")) {

 					int i = 1;

 					ps.setString(i++, uživatel);

 					try (ResultSet rs = ps.executeQuery()) {

 						rs.next();

 						int výsledek = rs.getBoolean(1) ? ANO : NE;

 						zapišVýsledek(výsledek);

 					}

 				}

 			} catch (SQLException e) {

 				logujSQLException(e);

 				zapišVýsledek(NE);

 			}

 		} else {

 			zapišVýsledek(NE);

 		}

 	}

 	private boolean ověřDoménu(String doména) {

 		return našeDoména.equalsIgnoreCase(doména);

 	}

 	private void logujSQLException(SQLException e) {

 		err.println("-- SQLException:");

 		e.printStackTrace(err);

 		err.println("-- SQLException");

 	}

 }