#include <stdio.h>
#include <string.h>
#include <misc.h>
#include <tlmplib.h>
#include "fdpass.h"
#include <string>
#include <vector>
#include <set>

using namespace std;

<mod>
int main (int argc, char *argv[])
{
	glocal int ret = -1;
	glocal const char *secretfile = "/etc/blackhole.secrets";
	glocal bool replace = false;
	glocal.ret = <call tlmpprogram>(argc,argv);
	<f init>
		setproginfo ("blackhole-gensecret",VERSION
			,"Gererate secrets used to communicate with horizon, wormhole and sub-blackhole\n"
			 "\n"
			 "blackhole-gensecret [ options ] horizon-name ...\n");
		setarg (' ',"secretfile","File holding secrets",glocal.secretfile,false);
		setarg ('r',"replace","Replace old secrets in file",glocal.replace,false);
	</f>
	<f main>
		glocal vector<string> lines;
		glocal set<string> names;
		for (int i=0; i<argc; i++) glocal.names.insert(argv[i]);
		<call loadfile>(glocal.secretfile,true);
		<f oneline>
			glocal.lines.push_back(line);
			return 0;
		</f>
		</call>
		<call savefile>(glocal.secretfile,false);
		<f dowrite>
			for (unsigned i=0; i<glocal.lines.size(); i++){
				const char *line = glocal.lines[i].c_str();
				string name;
				str_copyword(name,line);
				if (glocal.names.count(name) != 0){
					if (glocal.replace){
						char out[32+1];
						fdpass_makesalt (out);
						fprintf (fout,"%s %s\n",name.c_str(),out);	
					}else{
						tlmp_warning ("Name %s already in file, secret not replaced (Use option --replace to override)\n",name.c_str());
						fprintf (fout,"%s\n",line);
					}
					glocal.names.erase(name);
				}else{
					fprintf (fout,"%s\n",line);
				}
			}
			for (set<string>::iterator it=glocal.names.begin(); it != glocal.names.end(); it++){
				char out[32+1];
				fdpass_makesalt(out);
				fprintf (fout,"%s %s\n",it->c_str(),out);
			}
			return 0;
		</f>
		</call>
		return 0;
	</f>
	</call>
	return glocal.ret;
}
</mod>