#!/iw/perl/current/bin/perl
#
#  file: gen-ticker-map.pl
#  desc: generate various output formats of stock ticker data
#

eval 'exec /iw/perl/current/bin/perl -s $0 ${1+"$@"}'
  if 0;

# we want perl 5.00x or later
require 5.004;

# who am i?
($prog = $0) =~ s#.*/##;

use Getopt::Std;

# process command line options, if any
getopts('D:PT');

# generate perl data structure of ticker data?
my($do_perl) = defined($opt_P);

# generate Tcl data structure of ticker data?
my($do_tcl) = defined($opt_T);

if (! $do_perl && ! $do_tcl) {
	print usage();
	exit 1;
}


# slurp in the ticker data
my(%ticker_map) = ();
while(<>) {
	chomp;
	my($ticker, $uri, $company, $cik) = split(/\|/);

	# ignore ftp://ftp.sec.gov/... URIs since they are links to skulked documents
	# and not to the companies
	$uri = "" if ($uri =~ m@^ftp://ftp.sec.gov/@);

	$ticker_map{$ticker} = {uri => $uri, company => $company, cik => $cik};
}


if ($do_perl) {

	# generate perl library file
	print <<EoI;

#
#  NOTE: this file is auto-generated.  DO NOT EDIT.
#
#  file: edgar-ticker.pl
#  desc: SEC EDGAR stock ticker symbol support
#

# limit scope
package EDGAR::ticker;


# stock ticker lookup table
my(\$ticker_map) = (
EoI

	foreach $ticker (sort keys(%ticker_map)) {
		my($cik) = $ticker_map{$ticker}{'cik'};
		my($company) = $ticker_map{$ticker}{'company'};
		my($uri) = $ticker_map{$ticker}{'uri'};

		# escape quotes
		$cik =~     s/'/\\'/g;
		$company =~ s/'/\\'/g;
		$uri =~     s/'/\\'/g;

		print "\t'$ticker' => {",
		        "cik => '",
		        $cik,
		        "', company => '",
		        $company,
		        "', uri => '",
		        $uri,
		        "'",
		      "},\n";
	}

	print ");\n";

	print <<EoI;

#
# lookup stock ticker
#
sub lookup {
	my(\$cik) = shift;

	my(\$symbol) = "";

	# find company to ticker mappings
   
	return \$symbol;
} 

EoI

} elsif ($do_tcl) {

	# generate ticker data
	print <<EoI;

#
#  NOTE: this file is auto-generated.  DO NOT EDIT.
#
#  file: edgar-ticker.tcl
#  desc: SEC EDGAR CIK to stock ticker symbol support
#

package provide edgar::ticker 1.0

namespace eval edgar::ticker {

	# CIK to ticker symbol maps
	variable ticker_map
	array set ticker_map \\
		{
EoI

	foreach $ticker (sort keys(%ticker_map)) {
		my($cik) = $ticker_map{$ticker}{'cik'};
		my($company) = $ticker_map{$ticker}{'company'};
		my($uri) = $ticker_map{$ticker}{'uri'};

		# escape quotes
		$cik =~     s/"/\\"/g;
		$company =~ s/"/\\"/g;
		$uri =~     s/"/\\"/g;

		print "\t\t\t$ticker {",
		        ' cik "',     $cik, '"',
		        ' company "', $company, '"',
		        ' uri "',     $uri, '"',
		      " }\n";
	}

	# end of ticker array
	print <<EoI;
		}
   
	namespace export ticker_lookup uri_lookup
}
 
 
# lookup stock ticker by CIK
proc edgar::ticker::ticker_lookup {cik} {
	variable ticker_map

	set tickers ""

	set user_cik [string trimleft \$cik 0]
   
	foreach t [array names ticker_map] {
		array set ticker \$ticker_map(\$t)

		set ticker_cik [string trimleft \$ticker(cik) 0]

		if {\$ticker_cik == \$user_cik} {
			lappend tickers \$t
		}
	}

	return [join \$tickers " "]
} 

# lookup uri by CIK
proc edgar::ticker::uri_lookup {cik} {
	variable ticker_map

	set uris ""
   
	set user_cik [string trimleft \$cik 0]
   
	foreach t [array names ticker_map] {
		array set ticker \$ticker_map(\$t)

		set ticker_cik [string trimleft \$ticker(cik) 0]

		if {\$ticker_cik == \$user_cik} {
			lappend uris \$ticker(uri)
		}
	}

	return [join \$uris " "]
} 

EoI

}


# c'ya
exit 0;


#
# generate usage message
#
sub usage {
	my($msg) = "Usage: $prog ";
	$msg .= "-P | -T" . "\n";
	$msg .= "\tP = Perl mode\n";
	$msg .= "\tT = Tcl mode\n";

	$msg;
}