/*******************************************************************************
 * Copyright (c) 2014, 2016 Association for Decentralized Information Management
 * in Industry THTH ry.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     VTT Technical Research Centre of Finland - initial API and implementation
 *******************************************************************************/
package org.simantics.r.scl;

import java.util.UUID;
import java.util.concurrent.ConcurrentHashMap;

import org.rosuda.REngine.Rserve.RConnection;
import org.rosuda.REngine.Rserve.RserveException;
import org.simantics.db.layer0.variable.NodeSupport;
import org.simantics.scl.runtime.function.Function;

public class RSessionManager {
    static ConcurrentHashMap<String, RSession> CONNECTIONS =
            new ConcurrentHashMap<String, RSession>(); 

    static ConcurrentHashMap<String, NodeSupport<String>> SUPPORTS =
            new ConcurrentHashMap<String, NodeSupport<String>>(); 
    
    public static RSession getSession(String id) {
        // CONNECTIONS is ConcurrentHashMap so no synchronization is needed here
        return CONNECTIONS.get(id);
    }
    
    public static RSession createSession(RSessionConfiguration configuration) throws RserveException {
        synchronized(CONNECTIONS) {
            String id = UUID.randomUUID().toString();
            return createSession(configuration, id);
        }
    }
    
    public static Object withConfiguration(RSessionConfiguration configuration, @SuppressWarnings("rawtypes") Function f) throws RserveException {
        RSession session = createSession(configuration);
        try {
            return session.syncExec(f);
        } catch (InterruptedException e) {
            throw new RuntimeException(e);
        } finally {
            session.close();
        }
    }
    
    public static RSession getOrCreateSession(RSessionConfiguration configuration, String id) throws RserveException {
        synchronized(CONNECTIONS) {
            RSession session = getSession(id);
            if(session == null)
                return createSession(configuration, id);
            else
                return session;
        }
    }
    
    private static RSession createSession(RSessionConfiguration configuration, String id) throws RserveException {
        synchronized(CONNECTIONS) {
            RConnection connection = new RConnection(configuration.host, configuration.port);
            if(configuration.username != null && !configuration.username.isEmpty())
                connection.login(configuration.username, configuration.password);
            RSession managedConnection = new RSession(connection, id);
            CONNECTIONS.put(id, managedConnection);
            return managedConnection;
        }
    }
    
    public static NodeSupport<String> getOrCreateNodeSupport(String id) {
        synchronized(SUPPORTS) {
            NodeSupport<String> result = SUPPORTS.get(id);
            if(result == null) {
                RSession session = getSession(id);
                result = new NodeSupport<String>(session.getNodeManager());
                SUPPORTS.put(id, result);
            }
            return result;
        }
    }
}