/* Copyright (C) 2008 Vincent Penquerc'h.
   This file is part of the Kate codec library.
   Written by Vincent Penquerc'h.

   Use, distribution and reproduction of this library is governed
   by a BSD style source license included with this source in the
   file 'COPYING'. Please read these terms before distributing. */


#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
#include <stdarg.h>
#include <stdio.h>
#ifdef HAVE_STDLIB_H
#include <stdlib.h>
#endif
#include <errno.h>
#ifdef HAVE_STRING_H
#include <string.h>
#endif
#include <ctype.h>
#include <ogg/ogg.h>
#include "kate/kate.h"
#include "kstream.h"

static int is_ok_for_filename(const char *s)
{
  int c;
  while ((c=*s++)) {
    if (isalnum(c)) continue;
    if (strchr("-_",c)) continue;
    return 0;
  }
  return 1;
}

static void fcats(char **ptr,const char *s,size_t s_len)
{
  if (s && *s) {
    size_t ptr_len=(*ptr)?strlen(*ptr):0;
    *ptr=(char*)kate_realloc(*ptr,ptr_len+s_len+1);
    memcpy((*ptr)+ptr_len,s,s_len);
    (*ptr)[ptr_len+s_len]=0;
  }
}

static void fcat(char **ptr,const char *s)
{
  if (s && *s) {
    fcats(ptr,s,strlen(s));
  }
}

static int is_filename_used(const char *filename,const kate_stream_set *streams)
{
  size_t n;
  if (streams) {
    for (n=0;n<streams->n_kate_streams;++n) {
      const kate_stream *ks=streams->kate_streams[n];
      if (ks->filename && !strcmp(filename,ks->filename)) return 1;
    }
  }
  return 0;
}

char *get_filename(const char *basename,const kate_stream *ks,const kate_stream_set *streams)
{
  char tmp[32];
  char *filename=NULL;
  const char *ptr=basename;

  if (!basename) return NULL;
  if (!ks) return NULL;
  while (*ptr) {
    const char *percent=strchr(ptr,'%');
    if (!percent) {
      fcat(&filename,ptr);
      break;
    }
    if (percent>ptr) {
      fcats(&filename,ptr,percent-ptr);
    }
    ptr=percent+1;
    if (!*ptr) {
      fprintf(stderr,"absent format tag in filename\n");
      exit(-1);
    }
    switch (*ptr) {
      case '%':
        fcats(&filename,"%",1);
        break;
      case 'c':
        if (!ks->ki.category) {
          fcat(&filename,"-");
        }
        else if (!is_ok_for_filename(ks->ki.category)) {
          fprintf(stderr,"Category '%s' not suitable for using in a filename, using 'INVCAT'",ks->ki.category);
          fcat(&filename,"INVCAT");
        }
        else {
          fcat(&filename,ks->ki.category);
        }
        break;
      case 'l':
        if (!ks->ki.language) {
          fcat(&filename,"-");
        }
        else if (!is_ok_for_filename(ks->ki.language)) {
          fprintf(stderr,"Language '%s' not suitable for using in a filename, using 'INVLANG'",ks->ki.language);
          fcat(&filename,"INVLANG");
        }
        else {
          fcat(&filename,ks->ki.language);
        }
        break;
      case 's':
        snprintf(tmp,sizeof(tmp),"%08x",(ogg_uint32_t)(ks?ks->os.serialno:0));
        fcat(&filename,tmp);
        break;
      case 'i':
        snprintf(tmp,sizeof(tmp),"%d",ks?ks->stream_index:0);
        fcat(&filename,tmp);
        break;
      default:
        fprintf(stderr,"unknown format tag in filename: %c\n",*ptr);
        exit(-1);
        break;
    }
    ++ptr;
  }

  if (is_filename_used(filename,streams)) {
    kate_free(filename);
    return NULL;
  }

  return filename;
}

int kstream_init(kate_stream *ks,ogg_page *og,int stream_index)
{
  int ret;

  ogg_stream_init(&ks->os,ogg_page_serialno(og));
  ret=kate_info_init(&ks->ki);
  if (ret<0) {
    fprintf(stderr,"failed to init info\n");
    ogg_stream_clear(&ks->os);
    return ret;
  }
  kate_info_no_limits(&ks->ki,1);
  ret=kate_comment_init(&ks->kc);
  if (ret<0) {
    fprintf(stderr,"failed to init comments\n");
    ogg_stream_clear(&ks->os);
    return ret;
  }
  ks->init=kstream_uninitialized;
  ks->filename=NULL;
  ks->fout=NULL;
  ks->stream_index=stream_index;
  ks->event_index=0;
  ks->ret=0;

  return 0;
}

kate_stream *add_kate_stream(kate_stream_set *streams,ogg_page *og,int stream_index)
{
  int ret;
  kate_stream **new_streams;
  kate_stream *ks;

  ks=(kate_stream*)kate_malloc(sizeof(kate_stream));
  if (!ks) {
    fprintf(stderr,"failed to allocate %zu bytes\n",sizeof(kate_stream));
    exit(-1);
  }
  new_streams=(kate_stream**)kate_realloc(streams->kate_streams,(streams->n_kate_streams+1)*sizeof(kate_stream*));
  if (!new_streams) {
    kate_free(ks);
    fprintf(stderr,"failed to allocate %zu bytes\n",(streams->n_kate_streams+1)*sizeof(kate_stream*));
    exit(-1);
  }
  streams->kate_streams=new_streams;
  streams->kate_streams[streams->n_kate_streams]=ks;
  ret=kstream_init(ks,og,stream_index);
  if (ret<0) return NULL;
  ++streams->n_kate_streams;
  return ks;
}

int kstream_clear(kate_stream *ks)
{
  int ret;

  ogg_stream_clear(&ks->os);
  if (ks->init==kstream_data || ks->init==kstream_eos) {
    ret=kate_clear(&ks->k);
    if (ret<0) {
      fprintf(stderr,"kate_clear failed: %d\n",ret);
      /* continue anyway */
    }
  }
  ret=kate_info_clear(&ks->ki);
  if (ret<0) {
    fprintf(stderr,"kate_info_clear failed: %d\n",ret);
    /* continue anyway */
  }
  ret=kate_comment_clear(&ks->kc);
  if (ret<0) {
    fprintf(stderr,"kate_comment_clear failed: %d\n",ret);
    /* continue anyway */
  }
  if (ks->fout && ks->fout!=stdout) {
    ret=fclose(ks->fout);
    if (ret<0) {
      fprintf(stderr,"fclose failed (%d) - file %s might be corrupted\n",ret,ks->filename);
    }
    if (ks->ret<0) {
      ret=unlink(ks->filename);
      if (ret<0) {
        fprintf(stderr,"unlink failed (%d) - corrupt file %s will not be deleted\n",ret,ks->filename);
      }
    }
    kate_free(ks->filename);
  }

  return 0;
}

void init_kate_stream_set(kate_stream_set *streams)
{
  streams->kate_streams=NULL;
  streams->n_kate_streams=0;
}

kate_stream *find_kate_stream_for_page(kate_stream_set *streams,ogg_page *og)
{
  size_t n;
  for (n=0;n<streams->n_kate_streams;++n) {
    kate_stream *ks=streams->kate_streams[n];
    int pagein_ret=ogg_stream_pagein(&ks->os,og);
    if (pagein_ret==0) {
      return ks;
    }
  }
  return NULL;
}

void remove_kate_stream(const kate_stream *ks,kate_stream_set *streams)
{
  size_t n;
  for (n=0;n<streams->n_kate_streams;++n) {
    if (ks==streams->kate_streams[n]) {
      if (n!=streams->n_kate_streams-1) {
        memmove(streams->kate_streams+n,streams->kate_streams+n+1,(streams->n_kate_streams-1-n)*sizeof(kate_stream*));
      }
      --streams->n_kate_streams;
    }
  }
}

void clear_and_remove_kate_stream(kate_stream *ks,kate_stream_set *streams)
{
  kstream_clear(ks);
  remove_kate_stream(ks,streams);
}

void clear_kate_stream_set(kate_stream_set *streams)
{
  size_t n;
  for (n=0;n<streams->n_kate_streams;++n) {
    kstream_clear(streams->kate_streams[n]);
    kate_free(streams->kate_streams[n]);
  }
  kate_free(streams->kate_streams);
  streams->n_kate_streams=0;
  streams->kate_streams=NULL;
}

kate_stream *find_kate_stream_for_file(kate_stream_set *streams,FILE *f)
{
  size_t n;

  for (n=0;n<streams->n_kate_streams;++n) {
    if (streams->kate_streams[n]->fout==f) {
      return streams->kate_streams[n];
    }
  }

  return NULL;
}