RefreshAuthKeyUtil.java

/**
 * Copyright 2014 Sean Kavanagh - sean.p.kavanagh6@gmail.com
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.keybox.manage.util;

import com.keybox.common.util.AppConfig;
import java.util.Timer;
import java.util.TimerTask;

/**
 * sets authorized keys on systems with given refresh interval
 */
public class RefreshAuthKeyUtil {


	private Timer timer;
	private static Integer minute = Integer.valueOf(AppConfig.getProperty("authKeysRefreshInterval"));


	private RefreshAuthKeyUtil() {
		//set interval
		timer = new Timer();
		timer.schedule(new RefreshAllSystemsTimerTask(), minute * 60 * 1000);
	}

	/**
	 * start timer to refresh all systems
	 */
	public static void startRefreshAllSystemsTimerTask() {
		if (SSHUtil.keyManagementEnabled && minute > 0) {
			new RefreshAuthKeyUtil();
		}
	}

	/**
	 * Task for distributing keys to all systems
	 */
	public static void refreshAllSystems() {

		Runnable run = new RefreshAllSystemsTask();
		Thread thread = new Thread(run);
		thread.start();
	}

	/**
	 * Task for distributing keys to all systems based on profile
	 *
	 * @param profileId profile id
	 */
	public static void refreshProfileSystems(Long profileId) {

		Runnable run = new RefreshProfileSystemsTask(profileId);
		Thread thread = new Thread(run);
		thread.start();
	}

	/**
	 *Task for distributing keys to all systems based on user
	 *
	 * @param userId user id
	 */
	public static void refreshUserSystems(Long userId) {

		Runnable run = new RefreshUserSystemsTask(userId);
		Thread thread = new Thread(run);
		thread.start();

	}



	/**
	 * Timer task for distributing keys to all systems
	 */
	private class RefreshAllSystemsTimerTask extends TimerTask {

		@Override
		public void run() {
			//distribute all public keys
			SSHUtil.distributePubKeysToAllSystems();
			timer.cancel();

			//create new timer and set interval
			timer = new Timer();
			timer.schedule(new RefreshAllSystemsTimerTask(), minute * 60 * 1000);
		}
	}

}


/**
 * Task for distributing keys to all systems based on user
 */
class RefreshUserSystemsTask implements Runnable {

	Long userId;
	public RefreshUserSystemsTask(Long userId){
		this.userId=userId;
	}

	@Override
	public void run() {
		//distribute all public keys based on user
		SSHUtil.distributePubKeysToUser(this.userId);
	}
}


/**
 * Task for distributing keys to all systems assigned to a profile
 */
class RefreshProfileSystemsTask implements Runnable {

	Long profileId;
	public RefreshProfileSystemsTask(Long profileId){
		this.profileId=profileId;
	}

	@Override
	public void run() {
		//distribute all public keys based on profile
		SSHUtil.distributePubKeysToProfile(this.profileId);
	}
}


/**
 * Task for distributing keys to all systems
 */
class RefreshAllSystemsTask implements Runnable {

	@Override
	public void run() {
		//distribute all public keys
		SSHUtil.distributePubKeysToAllSystems();

	}
}