forums.smartfoxserver.com

Code: Select all

public class ReadOnlyUserVariablesFilter extends SysControllerFilter
{
	private final Set<String> readOnlyVars;
	private final Zone zone;
	private final Logger log;
	
	public ReadOnlyUserVariablesFilter(Zone zone)
	{
		this(zone, null);
	}
	
	public ReadOnlyUserVariablesFilter(Zone zone, List<String> varNames)
    {
		this.log = LoggerFactory.getLogger(this.getClass());
		this.zone = zone;
		readOnlyVars = varNames == null ? new CopyOnWriteArraySet<String>() : new CopyOnWriteArraySet<String>(varNames);
    }
	
	public void addVariable(String varName)
	{
		readOnlyVars.add(varName);
	}
	
	public void removeVariable(String varName)
	{
		readOnlyVars.remove(varName);
	}
	
	public List<String> getFilteredVariables()
	{
		return new ArrayList<String>(readOnlyVars);
	}
	
	public void clearFilter()
	{
		readOnlyVars.clear();
	}
	
	@Override
	public FilterAction handleClientRequest(User sender, ISFSObject params) throws SFSException
	{		
		// Not this Zone, continue
	    if (sender.getZone() != this.zone)
	    	return FilterAction.CONTINUE;
	    
	    ISFSArray vars = params.getSFSArray(SetUserVariables.KEY_VAR_LIST);
	    
	    if (vars != null && vars.size() > 0)
	    {
	    	ISFSArray filteredVars = null;
	    	
	    	if (vars.size() == 1)
	    		filteredVars = filterSingleVariableUpdate(vars);
	    	else
	    		filteredVars = filterMultiVariableUpdate(vars);
	    	
	    	if (filteredVars.size() < 1)
	    		return FilterAction.HALT;
	    	
	    	params.putSFSArray(SetUserVariables.KEY_VAR_LIST, filteredVars);
	    }
	    
	    return FilterAction.CONTINUE;
	}
	
	private ISFSArray filterSingleVariableUpdate(ISFSArray vars)
	{
		ISFSArray var = (ISFSArray) vars.getElementAt(0);
		
		return isReadOnly(var.getUtfString(0)) ? new SFSArray() : vars;
	}
	
	private ISFSArray filterMultiVariableUpdate(ISFSArray vars)
	{
		ISFSArray filteredVars = new SFSArray();
		
		for (int i = 0; i < vars.size(); i++)
    	{
    		ISFSArray var = (ISFSArray) vars.getElementAt(i);
    		if (!isReadOnly(var.getUtfString(0)))
    			filteredVars.addSFSArray(var);
    	}
		
		return filteredVars;
	}
	
	private boolean isReadOnly(String varName)
	{
		boolean readOnly = readOnlyVars.contains(varName);
		
		if (log.isDebugEnabled() && readOnly)
			log.debug("Filtering read-only variable: " + varName);
		
		return readOnly;
	}
}

Code: Select all

public class MyExtension extends SFSExtension
{
    @Override
    public void init()
    {
        // Reset filter chain to clean previous filters
        getParentZone().resetSystemFilterChain();
         
        ISystemFilterChain filterChain = new SysControllerFilterChain();
	ReadOnlyUserVariablesFilter filter = new ReadOnlyUserVariablesFilter(getParentZone(), Arrays.asList("alpha", "beta"));
        filterChain.addFilter("ReadOnlyUserVars", filter);
         
        // Plug the filter chain    
        getParentZone().setFilterChain(SystemRequest.SetUserVariables, filterChain);
    }
     
    @Override
    public void destroy()
    {
        super.destroy();
    }
}