cz.abclinuxu.datoveschranky.tinyDB.DataBoxManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tiny-isds Show documentation
Library implementing subset of JavaISDS - basically readonly access.
The newest version!
package cz.abclinuxu.datoveschranky.tinyDB;

import cz.abclinuxu.datoveschranky.common.entities.DeliveryInfo;
import cz.abclinuxu.datoveschranky.common.entities.MessageStateChange;
import cz.abclinuxu.datoveschranky.common.Utils;
import cz.abclinuxu.datoveschranky.common.DataBoxException;
import cz.abclinuxu.datoveschranky.common.interfaces.AttachmentStorer;
import cz.abclinuxu.datoveschranky.common.entities.Hash;
import cz.abclinuxu.datoveschranky.common.entities.Message;
import cz.abclinuxu.datoveschranky.common.entities.MessageEnvelope;
import cz.abclinuxu.datoveschranky.common.entities.MessageState;
import cz.abclinuxu.datoveschranky.common.entities.MessageType;
import cz.abclinuxu.datoveschranky.common.Config;
import cz.abclinuxu.datoveschranky.common.interfaces.DataBoxAccessService;
import cz.abclinuxu.datoveschranky.common.interfaces.DataBoxDownloadService;
import cz.abclinuxu.datoveschranky.common.interfaces.DataBoxMessagesService;
import cz.abclinuxu.datoveschranky.common.interfaces.DataBoxSearchService;
import cz.abclinuxu.datoveschranky.common.interfaces.DataBoxServices;
import cz.abclinuxu.datoveschranky.common.interfaces.DataBoxUploadService;
import cz.abclinuxu.datoveschranky.tinyDB.responseparsers.DownloadReceivedMessage;
import cz.abclinuxu.datoveschranky.tinyDB.responseparsers.GetListOfReceivedMessages;
import cz.abclinuxu.datoveschranky.tinyDB.responseparsers.DownloadSignedReceivedMessage;
import cz.abclinuxu.datoveschranky.tinyDB.responseparsers.VerifyMessage;
import cz.abclinuxu.datoveschranky.tinyDB.responseparsers.AbstractResponseParser;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.ProtocolException;
import java.net.URL;
import java.util.Arrays;
import java.util.Date;
import java.util.EnumSet;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.net.ssl.HttpsURLConnection;
import javax.net.ssl.SSLSocketFactory;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.parsers.SAXParser;
import javax.xml.parsers.SAXParserFactory;
import org.apache.commons.codec.binary.Base64;
import org.xml.sax.SAXException;

/**
 * Tato třída umožnuje přihlášení k datové schránce a základní operace s ní, tzn.
 * stažení přijatých zpráv a stažení přijaté zprávy včetně příloh.
 * 
 * @author Vaclav Rosecky <xrosecky 'at' gmail 'dot' com>
 * 
 */
public class DataBoxManager implements DataBoxMessagesService, DataBoxDownloadService, DataBoxServices {
    
    private static final int MAX_REDIRECT_COUNT = 25;
    private static final List redirectionCodes = Arrays.asList(301, 302);
    private static final List OKCodes = Arrays.asList(200, 304);
    protected final Config config;
    protected String authCookie = null;
    protected SSLSocketFactory socketFactory = null;
    protected Logger logger = Logger.getLogger(this.getClass().getName());
    protected String authorization;

    private DataBoxManager(Config configuration) {
        this.config = configuration;
    }

    /**
     * Realizuje přihlášení do datové schránky pod daným uživatelským jménem
     * a heslem a při úspěšném přihlášení vrátí příslušnou instanci ISDSManageru
     * poskytující služby k této schránce.
     * 
     * @param userName   jméno uživatele
     * @param password   heslo uživatele
     * @throws DataBoxException   při přihlašování do DS došlo k chybě. Důvodem může
     * být špatné heslo či uživatelské jméno, zacyklení při přesměrování či absence
     * autorizační cookie.
     * 
     */
    public static DataBoxManager login(Config config, String userName, String password) throws Exception {
        DataBoxManager manager = new DataBoxManager(config);
        manager.loginImpl(userName, password);
        return manager;
    }

    public DataBoxDownloadService getDataBoxDownloadService() {
        return this;
    }

    public DataBoxMessagesService getDataBoxMessagesService() {
        return this;
    }

    public DataBoxUploadService getDataBoxUploadService() {
        throw new UnsupportedOperationException("Sluzba DataBoxUploadService neni pristupna.");
    }

    // metody z DataBoxMessages
    public List getListOfReceivedMessages(Date from, Date to,
            EnumSet state, int offset, int limit) {
        // tohle hrani se stringy je neefektivní, ale pro nase 
        // demonstracni ucely to vyhovuje.
        String resource = "/GetListOfReceivedMessages.xml";
        String post = Utils.readResourceAsString(this.getClass(), resource);
        post = post.replace("${DATE_FROM}", XMLUtils.toXmlDate(from).toString());
        post = post.replace("${DATE_TO}", XMLUtils.toXmlDate(to).toString());
        post = post.replace("${OFFSET}", String.valueOf(offset));
        post = post.replace("${LIMIT}", String.valueOf(limit));
        GetListOfReceivedMessages result = new GetListOfReceivedMessages();
        this.postAndParseResponse(post, "dx", result);
        return result.getMessages();
    }

    public List getListOfSentMessages(Date from,
            Date to, EnumSet state,  int offset, int limit) {
        throw new UnsupportedOperationException();
    }

    public Hash verifyMessage(MessageEnvelope envelope) {
        String resource = "/VerifyMessage.xml";
        String post = Utils.readResourceAsString(this.getClass(), resource);
        post = post.replace("${ID}", envelope.getMessageID());
        VerifyMessage parser = new VerifyMessage();
        this.postAndParseResponse(post, "dx", parser);
        return parser.getResult();
    }

    // metody z DataBoxDownload
    public Message downloadMessage(MessageEnvelope envelope,
            AttachmentStorer storer) {
        if (envelope.getType() != MessageType.RECEIVED) {
            throw new UnsupportedOperationException("Stahnout lze pouze prijatou zpravu");
        }
        String resource = "/DownloadReceivedMessage.xml";
        String post = Utils.readResourceAsString(this.getClass(), resource);
        post = post.replace("${ID}", envelope.getMessageID());
        DownloadReceivedMessage parser = new DownloadReceivedMessage(envelope, storer);
        this.postAndParseResponse(post, "dz", parser);
        return new Message(envelope, null, null, parser.getResult());
    }

    public void downloadSignedMessage(MessageEnvelope envelope, OutputStream os) {
        if (envelope.getType() != MessageType.RECEIVED) {
            throw new UnsupportedOperationException("Stahnout lze pouze prijatou zpravu");
        }
        String resource = "/DownloadSignedReceivedMessage.xml";
        String post = Utils.readResourceAsString(this.getClass(), resource);
        post = post.replace("${ID}", envelope.getMessageID());
        DownloadSignedReceivedMessage parser = new DownloadSignedReceivedMessage(os);
        this.postAndParseResponse(post, "dz", parser);
    }

    public DeliveryInfo getDeliveryInfo(MessageEnvelope arg0) {
        throw new UnsupportedOperationException("Operace getDeliveryInfo neni touto " +
                "knihovnou podporovana.");
    }

    public List GetMessageStateChanges(Date from, Date to) {
	throw new UnsupportedOperationException("Operace GetMessageStateChanges neni touto " +
                "knihovnou podporovana.");
    }

    public DataBoxSearchService getDataBoxSearchService() {
        throw new UnsupportedOperationException("Sluzba DataBoxSearchService neni pristupna.");
    }

    public void markMessageAsDownloaded(MessageEnvelope env) {
        throw new UnsupportedOperationException("Operace markMessageAsDownloaded neni " +
                "touto knihovnou podporovana.");
    }
    
    public DataBoxAccessService getDataBoxAccessService() {
        throw new UnsupportedOperationException("Operace getDataBoxAccessService neni " +
        "touto knihovnou podporovana.");
    }

    /**
     * Stáhne přijatou zprávu včetně SOAP obálky a příloh jako XML soubor. Vhodné pouze
     * pro debugovací účely, ne pro zálohování.
     * 
     * @param envelope  obálka zprávy, která se má stáhnout
     * @param os        kam přijde uložit
     * @throws DataBoxException
     * 
     */
    public void storeMessageAsXML(MessageEnvelope envelope, OutputStream os) {
        if (envelope.getType() != MessageType.RECEIVED) {
            throw new UnsupportedOperationException("Stahnout lze pouze prijatou zpravu");
        }
        String resource = "/resources/DownloadReceivedMessage.xml";
        String post = Utils.readResourceAsString(this.getClass(), resource);
        post = post.replace("${ID}", envelope.getMessageID());
        this.storeRequest(post, "dz", os);
    }

    public void getSignedDeliveryInfo(MessageEnvelope envelope, OutputStream os) {
        throw new UnsupportedOperationException("Not supported.");
    }

    private void loginImpl(String username, String password) throws Exception {
        String userPassword = username + ":" + password;
        Base64 base64 = new Base64(0, null, false);
        // základní HTTP autorizace
        authorization = "Basic " + new String(base64.encode(userPassword.getBytes()), "UTF-8");
        this.socketFactory = Utils.createSSLSocketFactory(config.getKeyStore());
    }

    private void postAndParseResponse(String post, String prefix,
            AbstractResponseParser rp) {
        HttpsURLConnection con = null;
        try {
            // udelame post
            URL url = new URL(config.getServiceURL() + prefix);
            con = (HttpsURLConnection) url.openConnection();
            this.configure(con);
            con.getOutputStream().write(post.getBytes("UTF-8"));
            checkHttpResponseCode(con);
            // zparsujeme výsledek SAX parserem
            SAXParserFactory factory = SAXParserFactory.newInstance();
            factory.setValidating(false);
            SAXParser parser = factory.newSAXParser();
            parser.parse(con.getInputStream(), new SimpleSAXParser(rp));
            // ověříme vrácený stav pri volani webove služby
            if (!rp.getStatus().ok()) {
                String message = String.format("Pozadavek selhal chybou %s:%s",
                        rp.getStatus().getStatusCode(), rp.getStatus().getStatusMesssage());
                logger.log(Level.SEVERE, message);
                throw new DataBoxException(message);
            }
        } catch (SAXException sax) {
            throw new DataBoxException("Chyba pri parsovani odpovedi.", sax);
        } catch (ParserConfigurationException pce) {
            throw new DataBoxException("Chyba pri konfiguraci SAX parseru.", pce);
        } catch (IOException ioe) {
            throw new DataBoxException("IO chyba pri cteni odpovedi.", ioe);
        } finally {
            close(con);
        }
    }

    private void storeRequest(String request, String prefix, OutputStream os) {
        HttpsURLConnection con = null;
        try {
            URL url = new URL(config.getServiceURL() + prefix);
            con = (HttpsURLConnection) url.openConnection();
            this.configure(con);
            con.getOutputStream().write(request.getBytes("UTF-8"));
            this.checkHttpResponseCode(con);
            InputStream is = con.getInputStream();
            Utils.copy(is, os);
        } catch (IOException ioe) {
            throw new DataBoxException("Nemohu ulozit zpravu", ioe);
        } finally {
            close(con);
        }
    }

    private void checkHttpResponseCode(HttpsURLConnection con) throws IOException {
        if (!OKCodes.contains(con.getResponseCode())) {
            String message = String.format("Pozadavek selhal se stavovym kodem %d %s.",
                    con.getResponseCode(), con.getResponseMessage());
            logger.log(Level.SEVERE, message);
            throw new DataBoxException(message);
        }
    }

    private void configure(HttpsURLConnection connect) throws ProtocolException {
        connect.setSSLSocketFactory(socketFactory);
        // connect.setRequestProperty("Authorization", authorization);
        connect.setRequestProperty("Authorization", authorization);
        // connect.setRequestProperty("Cookie", authCookie);
        connect.setRequestMethod("POST");
        connect.setDoOutput(true);
        connect.setRequestProperty("Content-Type", "text/xml; charset=utf-8");
        connect.setRequestProperty("Soapaction", "");
    }

    private void close(HttpsURLConnection con) {
        if (con != null) {
            con.disconnect();
        }
    }
    
    
}