ServerProfile.java

/******************************************************************************
 *                                                                            *
 *                    Copyright 2016 Subterranean Security                    *
 *                                                                            *
 *  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.subterranean_security.crimson.sv.profile;

import java.util.ArrayList;
import java.util.Date;

import com.subterranean_security.crimson.core.attribute.Attribute;
import com.subterranean_security.crimson.core.attribute.UntrackedAttribute;
import com.subterranean_security.crimson.core.attribute.keys.AKeySimple;
import com.subterranean_security.crimson.core.proto.Delta.AttributeGroupContainer;
import com.subterranean_security.crimson.core.proto.Delta.EV_ProfileDelta;
import com.subterranean_security.crimson.core.proto.Delta.EV_ServerProfileDelta;
import com.subterranean_security.crimson.core.proto.Delta.EV_ViewerProfileDelta;
import com.subterranean_security.crimson.core.proto.Listener.ListenerConfig;
import com.subterranean_security.crimson.core.proto.Misc.AuthMethod;
import com.subterranean_security.crimson.core.util.ProtoUtil;
import com.subterranean_security.crimson.server.store.ListenerStore;
import com.subterranean_security.crimson.server.store.ProfileStore;
import com.subterranean_security.crimson.sv.net.Listener;
import com.subterranean_security.crimson.viewer.ui.UIStore;

public class ServerProfile extends Profile {

	private static final long serialVersionUID = 1L;

	private Date updateTimestamp = new Date();

	public ArrayList<ListenerConfig> listeners = new ArrayList<ListenerConfig>();
	public ArrayList<AuthMethod> authMethods = new ArrayList<AuthMethod>();
	public ArrayList<ViewerProfile> users = new ArrayList<ViewerProfile>();

	// General attributes
	private Attribute messageLatency;

	public ServerProfile() {
		super();
		messageLatency = new UntrackedAttribute();
	}

	public String getMessageLatency() {
		return messageLatency.get();
	}

	public void setMessageLatency(String messageLatency) {
		this.messageLatency.set(messageLatency);
	}

	public Date getUpdateTimestamp() {
		return updateTimestamp;
	}

	public void amalgamate(EV_ServerProfileDelta c) {
		super.amalgamate(c.getPd());

		for (ListenerConfig lc : c.getListenerList()) {
			for (ListenerConfig l : listeners) {
				if (lc.getId() == l.getId()) {
					listeners.remove(l);
					break;
				}
			}

			listeners.add(lc);
		}
		if (UIStore.netMan != null && c.getListenerCount() > 0) {
			UIStore.netMan.lp.lt.fireTableDataChanged();
		}

		for (EV_ViewerProfileDelta lc : c.getViewerUserList()) {
			boolean modified = false;
			for (ViewerProfile l : users) {
				if (l.get(AKeySimple.VIEWER_USER).equals(
						ProtoUtil.getGeneralGroup(lc).getAttributeOrDefault(AKeySimple.VIEWER_USER.getFullID(), ""))) {
					modified = true;
					l.amalgamate(lc);
					break;
				}
			}
			if (!modified) {

				ViewerProfile vp = new ViewerProfile();
				vp.amalgamate(lc);
				users.add(vp);
			}

		}
		for (AuthMethod am : c.getAuthMethodList()) {
			boolean modified = false;
			for (AuthMethod a : authMethods) {
				if (a.getId() == am.getId()) {
					a = AuthMethod.newBuilder().mergeFrom(a).mergeFrom(am).build();
					modified = true;
				}
			}
			if (!modified) {
				authMethods.add(am);
			}
		}

		if (UIStore.userMan != null && c.getAuthMethodCount() > 0) {
			UIStore.userMan.up.ut.fireTableDataChanged();
		}

	}

	public EV_ServerProfileDelta getUpdates(Date lastUpdate, ViewerProfile vp) {
		EV_ServerProfileDelta.Builder spd = EV_ServerProfileDelta.newBuilder();

		try {
			// add general attributes
			spd.setPd(EV_ProfileDelta.newBuilder(super.getUpdates(lastUpdate))
					.addGroup(AttributeGroupContainer.newBuilder().putAttribute(AKeySimple.SERVER_STATUS.getFullID(),
							ListenerStore.isRunning() ? "1" : "0")));

			// add listeners
			for (Listener l : ListenerStore.listeners) {
				spd.addListener(l.getConfig());
			}

			// add viewers
			for (Integer i : ProfileStore.getViewerKeyset()) {

				ViewerProfile vpi = ProfileStore.getViewer(i);
				if (vpi.equals(vp)) {
					// give full rights to the viewer's own profile
					spd.addViewerUser(vpi.gatherForServer(null));
				} else {
					spd.addViewerUser(vpi.gatherForServer(vp.getPermissions()));
				}

			}
		} catch (Exception e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}

		return spd.build();
	}

}