/*
	Reads connection rules and extracts the one related to the horizons provided.
*/
#include <stdio.h>
#include <string.h>
#include <misc.h>
#include <tlmplib.h>
#include <string>
#include <vector>
#include <set>

using namespace std;

<mod>
int main (int argc, char *argv[])
{
	glocal int ret = -1;
	glocal vector<string> horizons;
	glocal const char *horizonfile = NULL;
	glocal const char *rulefile = NULL;
	glocal bool addnear = false;
	glocal.ret = <call tlmpprogram>(argc,argv);
	<f init>
		setproginfo ("blackhole-rulefilter",VERSION
			,"Filter rule that only applies to some horizons\n"
			 "(and will be handle by a sub-blackhole)\n");
		setarg ('f',"horizonfile","File holding Horizon names",glocal.horizonfile,false);
		setarg ('h',"horizon","Horizon name",glocal.horizons,false);
		setarg ('r',"rulefile","Rules to filter (or - for stdin)",glocal.rulefile,true);
		setarg (' ',"addnear","Add the near keyword at the end of the horizon command",glocal.addnear,false);
	</f>
	<f main_noarg>
		int ret = -1;
		glocal.horizons.push_back("DROP");	// This is a special target horizon
							// so if a left side points to it, it matches.
		if (glocal.horizonfile != NULL){
			<call loadfile>(glocal.horizonfile,true);
			<f oneline>	
				glocal.horizons.push_back(line);
				return 0;
			</f>
			</call>
		}
		glocal set<string> hors;
		for (unsigned i=0; i<glocal.horizons.size(); i++) glocal.hors.insert (glocal.horizons[i]);
		<call loadfile>(glocal.rulefile,true);
		<f oneline>
			vector<string> words;
			int n = str_splitline (line,' ',words);
			if (n == 2 && strcmp(words[0].c_str(),"horizon")==0){
				if (glocal.hors.count(words[1])){
					printf ("%s%s\n",line,glocal.addnear ? " near" : "");
				}
			}else if (n == 8 && (strcmp(words[0].c_str(),"rule")==0
				|| strcmp(words[0].c_str(),"arule")==0)){
				if (glocal.hors.count(words[1]) != 0
					&& glocal.hors.count(words[5]) != 0){
					printf ("%s\n",line);
				}
			}	
			return 0;
		</f>
		</call>
		return ret;
		
	</f>
	</call>
	return glocal.ret;
}
</mod>