/*************************************************************************
 *
 * HTCd - Copyright (C) 1998-2006 Henrik Rydberg
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 */

#include <mime/mime.h>

const charset par_left("([{"),par_right("}])");

inline int is_field(const sref& s) {
  sref p=s.left_first(':');
  while(cset_spaces.test(p.back())) p=p.popb();
  return p.nempty()&&p.find_of(cset_spaces)<0;
}

inline int getquoted(const sref& s,char c) {
  int i,quote=0,par=0;
  for(i=0;i<s.size();i++) {
    if(s[i]=='\"') quote=!quote;
    else if(par_left.test(s[i])) par++;
    else if(par_right.test(s[i])) par--;
    else if(s[i]==c&&!quote&&!par) break;
  }
  return i;
}

void Mime::Init(const sref& message)
{
  mstring line;
  rrpile pile;
  sref first;
  head=body=message;
  while(Splitln(first,body)) {
    if(first.empty()) break;
    if(!is_field(first)) Special(first);
    else {
      line=first;
      while(cset_spaces.test(*body)) {
        Splitln(first,body);
        line.append(1,' ');
        line+=first.right_first_not_of(cset_spaces);
      }
      sref opt=line;
      int col=opt.find(':');
      sref name=Trim(opt.left(col),cset_quotes);
      opt=opt.adv(col).popf();
      sref rawval=Trim(opt);
      int semi=getquoted(opt,';');
      sref val=Trim(opt.left(semi),cset_quotes);
      if(getquoted(val,'=')<val.size()) val.clear();
      else opt=opt.adv(semi).popf();
      pile.clear();
      while(opt.nempty()) {
        sref optname=opt.left(getquoted(opt,'='));
        opt=opt.adv(optname).popf();
        semi=getquoted(opt,';');
        sref optval=opt.left(semi);
        opt=opt.adv(semi).popf();
        optname=Trim(optname,cset_quotes);
        optval=Trim(optval,cset_quotes);
        if(optname.nempty()) pile.push_back(rrstring(optname,optval));
      }
      Handle(name,rawval,val,pile);
    }
  }
  head=head.left(body-head);
}


///////////////////////////////////////////////////////////////////////////

Mime::Mime()
{
  mime_version="1.0";
}

int Mime::Special(sref line)
{
  return 0;
}

int Mime::Handle(const sref& name,
                 const sref& rawval,
                 const sref& val,
                 const rrpile& opts) {
  if(eqn(name,mime_version_tag)) mime_version=val;
  else if(eqn(name,cont_type_tag)) {
    (cont_type=val).tolower();
    for(rrpile::const_iterator p=opts.begin();p!=opts.end();p++) {
      if(eqn(p->first,charset_tag)) cont_type_charset=p->second;
      else if(eqn(p->first,name_tag)) cont_type_name=p->second;
      else if(eqn(p->first,boundary_tag)) cont_type_boundary=p->second;
    }
  }
  else if(eqn(name,cont_id_tag)) cont_id=val;
  else if(eqn(name,cont_desc_tag)) cont_desc=val;
  else if(eqn(name,cont_disp_tag)) {
    cont_disp=val;
    for(rrpile::const_iterator p=opts.begin();p!=opts.end();p++) {
      if(eqn(p->first,name_tag)) cont_disp_name=p->second;
      else if(eqn(p->first,file_tag)) cont_disp_file=p->second;
    }
  }
  else if(eqn(name,cont_trans_enc_tag)) cont_trans_enc=val;
  else if(eqn(name,cont_length_tag)) cont_length=val;
  else return 0;
  return 1;
}

///////////////////////////////////////////////////////////////////////////

void Mime::putHead(mstream& out) const
{
  if(putval(out,mime_version_tag,mime_version)) putend(out);
  if(putval(out,cont_type_tag,cont_type)) {
    putopt(out,charset_tag,cont_type_charset);
    putopt(out,name_tag,cont_type_name);
    putopt(out,boundary_tag,cont_type_boundary);
    putend(out);
  }
  if(putval(out,cont_id_tag,cont_id)) putend(out);
  if(putval(out,cont_desc_tag,cont_desc)) putend(out);
  if(putval(out,cont_disp_tag,cont_disp)) {
    putopt(out,name_tag,cont_disp_name);
    putopt(out,file_tag,cont_disp_file);
    putend(out);
  }
  if(putval(out,cont_trans_enc_tag,cont_trans_enc)) putend(out);
  if(putval(out,cont_length_tag,cont_length)) putend(out);
}


///////////////////////////////////////////////////////////////////////////

int Mime::Multipart(rpile& list) const
{
  if(cont_type_boundary.empty()) return 0;
  mstring bound=sref("--")+cont_type_boundary;
  sref rest=body.past_first(bound).past_nl();
  while(rest.nempty()) {
    int pos=rest.find(bound);
    if(pos>=0) {
      sref first=rest.left(pos);
      if(first.back()=='\n') {
        first=first.popb();
        if(first.back()=='\r') first=first.popb();
      }
      list.push_back(first);
      rest=rest.right(pos).adv(bound.size());
      if(*rest=='-') rest.clear(); else rest=rest.past_nl();
    }
    else {
      list.push_back(rest);
      rest.clear();
    }
  }
  return !list.empty();
}

void Mime::Pick(const sref& mime)
{
  rpile list;
  if(Multipart(list)) {
    if(eqn(cont_type,mime_multialt)) {
      int pos=0;
      for(int i=0;i<list.size();i++) {
        Mime msg(list[i]);
        if(eqn(msg.cont_type,mime)) { pos=i; break; }
      }
      Init(list[pos]);
    }
  }
}