/*
* RHQ Management Platform
* Copyright (C) 2005-2010 Red Hat, Inc.
* All rights reserved.
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation version 2 of the License.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/
package org.rhq.enterprise.server.plugins.alertOperations;
import java.util.Date;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.rhq.core.domain.alert.Alert;
import org.rhq.core.domain.auth.Subject;
import org.rhq.core.domain.operation.OperationDefinition;
import org.rhq.core.domain.resource.Resource;
import org.rhq.enterprise.server.alert.AlertManagerLocal;
import org.rhq.enterprise.server.resource.ResourceManagerLocal;
import org.rhq.enterprise.server.util.LookupUtil;
/**
* Helper to replace tokens by their values
* @author Heiko W. Rupp
*/
public class AlertTokenReplacer {
private final Log log = LogFactory.getLog(AlertTokenReplacer.class);
public static final String NOT_YET_IMPLEMENTED = " - not yet implemented -";
protected static final String THE_QUICK_BROWN_FOX_JUMPS_OVER_THE_LAZY_DOG = "TheQuickBrownFoxJumpsOverTheLazyDOg";
private Alert alert;
private Pattern pattern;
private OperationDefinition operationDefinition;
private Resource targetResource;
public AlertTokenReplacer(Alert alert, OperationDefinition operationDefinition, Resource targetResource) {
this.alert = alert;
this.operationDefinition = operationDefinition;
this.targetResource = targetResource;
pattern = Pattern.compile("<%\\s*([a-zA-Z]+\\.[a-zA-Z0-9]+)\\s*%>");
}
/**
* Replace all tokens on the input line. If no tokens are found the input is returned.
* Tokens have the form '<i><% class.sub %></i>'
* @param input a line of text
* @return input with tokens replaced.
* @see org.rhq.enterprise.server.plugins.alertOperations.Token
* @see org.rhq.enterprise.server.plugins.alertOperations.TokenClass
*/
public String replaceTokens(String input) {
String work = input;
Matcher matcher = pattern.matcher(work);
if (!matcher.find()) {
if (log.isDebugEnabled()) {
log.debug("No tokens found in " + input);
}
return input;
}
matcher.reset();
do {
matcher = pattern.matcher(work);
if (!matcher.find()) {
break;
}
String replacement = replaceToken(matcher.group(1));
String s = matcher.replaceFirst(replacement);
work = s;
} while (true);
return work;
}
/**
* Replace the token string passed (without the token delimiters ) with the actual value
* @param tokenString Input like alert.id
* @return replacement string or the input if the token was not valid.
*/
public String replaceToken(String tokenString) {
// Ok, we have at least one token. Now split the tokenString and loop over the tokens
if (!tokenString.contains("."))
return tokenString;
String tmp = tokenString.substring(0, tokenString.indexOf("."));
TokenClass tc = TokenClass.getByText(tmp);
if (tc == null) {
log.warn("Unknown token class in [" + tokenString + "], not replacing tokens");
return tokenString;
}
Token token = Token.getByText(tokenString);
if (token == null) {
log.warn("No known token found in [" + tokenString + "], not replacing token");
return tokenString;
}
String ret = null;
switch (tc) {
case ALERT:
ret = replaceAlertToken(token, alert);
break;
case RESOURCE:
ret = replaceResourceToken(token, alert.getAlertDefinition().getResource());
break;
case TARGET_RESOURCE:
// Create a "pseudo" token to feed it into the plain resource replacement code
String text = "resource." + token.getName();
Token tok = Token.getByText(text);
ret = replaceResourceToken(tok, targetResource);
break;
case OPERATION:
ret = replaceOperationToken(token);
break;
case TEST:
switch (token) {
case TEST_ECHO:
ret = tokenString;
break;
case TEST_FIX:
ret = THE_QUICK_BROWN_FOX_JUMPS_OVER_THE_LAZY_DOG;
break;
case TEST_CAMEL:
ret = "camel";
break;
default:
ret = NOT_YET_IMPLEMENTED;
}
break;
}
return ret;
}
private String replaceAlertToken(Token token, Alert alert) {
AlertManagerLocal mgr = LookupUtil.getAlertManager();
switch (token) {
case ALERT_ID:
return String.valueOf(alert.getId());
case ALERT_FIRE_TIME:
return new Date(alert.getCtime()).toString(); // TODO use a specific impl here?
case ALERT_WILL_RECOVER:
return String.valueOf(alert.getAlertDefinition().getWillRecover());
case ALERT_WILL_DISABLE:
return String.valueOf(mgr.willDefinitionBeDisabled(alert));
case ALERT_DEF_NAME:
return alert.getAlertDefinition().getName();
case ALERT_DEF_DESC:
return alert.getAlertDefinition().getDescription();
case ALERT_DEF_PRIO:
return alert.getAlertDefinition().getPriority().getName();
case ALERT_URL:
return mgr.prettyPrintAlertURL(alert);
case ALERT_CONDITIONS:
return mgr.prettyPrintAlertConditions(alert, false);
default:
return NOT_YET_IMPLEMENTED;
}
}
private String replaceResourceToken(Token token, Resource resource) {
ResourceManagerLocal mgr = LookupUtil.getResourceManager();
Subject overlord = LookupUtil.getSubjectManager().getOverlord();
Resource parent;
Resource platform = mgr.getPlaformOfResource(overlord, resource.getId());
if (platform == null)
platform = resource;
switch (token) {
case RESOURCE_ID:
return String.valueOf(resource.getId());
case RESOURCE_NAME:
return resource.getName();
case RESOURCE_PARENT_ID:
parent = mgr.getParentResource(resource.getId());
if (parent == null)
return "0";
else
return String.valueOf(parent.getId());
case RESOURCE_PARENT_NAME:
parent = mgr.getParentResource(resource.getId());
if (parent == null)
return "N/A";
else
return parent.getName();
case RESOURCE_TYPE_ID:
return String.valueOf(resource.getResourceType().getId());
case RESOURCE_TYPE_NAME:
return resource.getResourceType().getName();
case RESOURCE_PLATFORM_ID:
return String.valueOf(platform.getId());
case RESOURCE_PLATFORM_NAME:
return platform.getName();
case RESOURCE_PLATFORM_TYPE:
return platform.getResourceType().getName();
default:
return NOT_YET_IMPLEMENTED;
}
}
private String replaceOperationToken(Token token) {
switch (token) {
case OPERATION_ID:
return String.valueOf(operationDefinition.getId());
case OPERATION_NAME:
return operationDefinition.getName();
default:
return NOT_YET_IMPLEMENTED;
}
}
}