org.nustaq.kontraktor.apputil.FileExposerMixin Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of service-support Show documentation
service registry
There is a newer version: 5.2.0
package org.nustaq.kontraktor.apputil;

import com.google.common.io.Files;
import io.undertow.server.HttpServerExchange;
import io.undertow.util.Headers;
import org.nustaq.kontraktor.Actor;
import org.nustaq.kontraktor.IPromise;
import org.nustaq.kontraktor.annotations.CallerSideMethod;
import org.nustaq.kontraktor.remoting.http.undertow.builder.BldFourK;

import java.io.File;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.nio.ByteBuffer;
import java.util.UUID;
import java.util.concurrent.ConcurrentHashMap;

import static org.nustaq.kontraktor.Actors.resolve;

class ExposeEntry {
    File f;
    String type;
    Crypter crypter;

    public ExposeEntry(File f, String type) {
        this.f = f;
        this.type = type;
    }

    public ExposeEntry f(final File f) {
        this.f = f;
        return this;
    }

    public ExposeEntry type(final String type) {
        this.type = type;
        return this;
    }

    public ExposeEntry crypter(final Crypter crypter) {
        this.crypter = crypter;
        return this;
    }

    public byte[] getBytes() throws IOException {
        byte[] bytes = Files.toByteArray(f);
        if ( crypter != null )
            bytes = crypter.decrypt(bytes);
        return bytes;
    }
}

public interface FileExposerMixin> {

    String URLPREFIX = "exposed";
    ConcurrentHashMap exposedDocuments = new ConcurrentHashMap<>();

    static void auto(BldFourK bld, Object mixin) {
        bld.httpHandler(URLPREFIX, httpServerExchange ->  {
            httpServerExchange.dispatch();
            httpServerExchange
                .setResponseCode(200);
            ((FileExposerMixin)mixin).handleExposedFileAccess(httpServerExchange);
        });
    }

    default void handleExposedFileAccess(HttpServerExchange httpServerExchange) {
        try {
            String requestPath = httpServerExchange.getRequestPath();
            String[] split = requestPath.split("/");
            String docId = split[split.length - 1];
            ExposeEntry exposeEntry = takeFile(docId);
            httpServerExchange.getResponseHeaders().put(Headers.CONTENT_TYPE, exposeEntry.type+"; charset=utf-8");
            httpServerExchange.getResponseSender().send(ByteBuffer.wrap(exposeEntry.getBytes()));
            takeFile(docId);
        } catch (Exception e) {
            httpServerExchange.getResponseHeaders().put(Headers.CONTENT_TYPE, "text/html; charset=utf-8");
            try {
                httpServerExchange.getResponseSender().send( ByteBuffer.wrap("Unbekannte Resource".getBytes("UTF-8") ));
            } catch (Exception e1) {
                e1.printStackTrace();
            }
        }
    }

    @CallerSideMethod
    default String exposeFile(File localPath, String type, Crypter cr) {
        String uid = UUID.randomUUID().toString();
        exposedDocuments.put(uid,new ExposeEntry(localPath,type).crypter(cr));
        return uid;
    }

    @CallerSideMethod
    default ExposeEntry takeFile(String id) {
        ExposeEntry entry = exposedDocuments.get(id);
        exposedDocuments.remove(id);
        return entry;
    }

}