package org.argeo.slc.jsch;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.beans.factory.DisposableBean;
import org.springframework.beans.factory.InitializingBean;

import com.jcraft.jsch.Session;

/** Caches a JSCH session in the the ssh target. */
public class JschContextSession extends AbstractJschTask implements
		InitializingBean, DisposableBean {
	private final static Log log = LogFactory.getLog(JschContextSession.class);
	private Boolean autoconnect = false;

	@Override
	void run(Session session) {
		// clear();
		getSshTarget().setSession(session);
		if (log.isDebugEnabled())
			log.debug("Cached SSH context session to " + getSshTarget());
	}

	public void afterPropertiesSet() throws Exception {
		// if (log.isDebugEnabled())
		// log.debug(getClass() + ".afterPropertiesSet(), " + beanName + ", "
		// + this);
		if (autoconnect)
			try {
				run();
			} catch (Exception e) {
				log.error("Could not automatically open session", e);
			}
	}

	public void destroy() throws Exception {
		clear();
	}

	public void clear() {
		SshTarget sshTarget = getSshTarget();
		synchronized (sshTarget) {
			if (sshTarget.getSession() != null) {
				sshTarget.getSession().disconnect();
				sshTarget.setSession(null);
				if (log.isDebugEnabled())
					log.debug("Cleared cached SSH context session to "
							+ getSshTarget());
			}
		}
	}

	public void setAutoconnect(Boolean autoconnect) {
		this.autoconnect = autoconnect;
	}

}