templates.cc

/* Template handling functions for TagReport. */

#include "config.h"
#include "templates.h"
#include "util.h"

#include <cstdio>
#include <iostream>
#include <string>
#include <sstream>

#ifdef HAVE_LIBIBERTY_H
#include <libiberty.h>
#endif

#include "tagreport.h"

using namespace std;

#include "default.h"

extern int yyparse (void);
extern FILE *yyin;

/* Default templates */
key template_title (DEFAULT_TITLE), template_body_tag(DEFAULT_BODY_TAG);
key template_stats (DEFAULT_STATS), template_prebody(DEFAULT_PREBODY);
key template_body (DEFAULT_BODY), template_footer(DEFAULT_FOOTER);
key template_head_body, template_header; /* unset by default */

const struct replace_map header_map[] = {
  { "$count", COUNT, 6 },
  { "$date", DATE, 5 },
  { "$directory", DIRECTORY, 10 },
};

const struct replace_map body_map[] = {
  { "$artist", ARTIST, 7 },
  { "$title", TITLE, 6 },
  { "$num", NUMBER, 4 },
  { "$a-t", ARTIST_TITLE, 4 }
};

string replace_header (const string & in, int count, const string & directory)
{
  unsigned int i, j;
  string out = in;
  
  /* Replace all \n with actual newlines */
  while ((i = out.find ("\\n")) != string::npos)
    out.replace (i, 2, "\n"); 

  for (i = 0; i < ARRAY_SIZE(header_map); i++)
  {
    while ((j = out.find (header_map[i].term)) != string::npos)
    {
      ostringstream s;
      char* t;
      
      switch (header_map[i].i)
      {
        case COUNT:
          s << count;
          break;

        case DATE:
          t = get_time_string();
          s << t;
          delete[] t;
          break;

        case DIRECTORY:
          s << directory;
          break;

        default: /* huh?!? this better not happen */
          abort();
      }
      
      out.replace (j, header_map[i].term_length, s.str());
    }
  }

  return out;
}

string replace_body (const string & artist, const string & title, unsigned int n)
{
  string out = template_body.get();
  unsigned int i, j;

  /* replace \n with actual newlines */
  while ((i = out.find ("\\n")) != string::npos)
    out.replace (i, 2, "\n");

  /* Look for each term in the list of replacements */
  for (i = 0; i < ARRAY_SIZE(body_map); i++)
  {
    /* find all occurrences */
    while ((j = out.find(body_map[i].term)) != string::npos)
    {
      switch (body_map[i].i)
      {
        case ARTIST:
          out.replace (j, body_map[i].term_length, artist);
          break;
	  
        case TITLE:
          out.replace (j, body_map[i].term_length, title);
	  break;

        case ARTIST_TITLE:
	  if (artist == "") /* Left unset by traverse_dir */
	    out.replace (j, body_map[i].term_length, title);
	  else
	  {
            ostringstream s;
	    s << artist << " - " << title;
	    out.replace (j, body_map[i].term_length, s.str());
	  }
	  break;

	case NUMBER:
	{
          ostringstream s;
	  s << n;
          out.replace (j, body_map[i].term_length, s.str());
	  break;
	}

	default: /* wha? */
	  abort();
      }
    }
  }

  return out;
}

void read_template_file (char * template_fn)
{
  FILE * template_file = NULL;

  if ((template_file = fopen (template_fn, "r")) == NULL)
  {
    cerr << "Error reading template " << template_fn << ": " << strerror(errno) << endl;
    free(template_fn);
    if (!force)
      exit(1);
  }
  else
  {
    yyin = template_file;
    yyparse();
    fclose(template_file);
    free(template_fn);
  }
}