package org.paneris.jammyjoes.mail;

 import java.io.IOException;

 import java.io.OutputStream;

 import java.util.Iterator;

 import java.util.List;

 import javax.mail.MessagingException;

 import com.quiotix.html.parser.HtmlDocument;

 import com.quiotix.html.parser.HtmlDumper;

 import com.quiotix.html.parser.HtmlDocument.Attribute;

 import com.quiotix.html.parser.HtmlDocument.Tag;

 public class ImageTranslator extends HtmlDumper {

   private int count;

   private NewsletterInterface newsletter;

   public ImageTranslator(OutputStream out, NewsletterInterface newsletter) {

     super(out);

     this.newsletter = newsletter;

   }

   public void visit(Tag t)  {

     if (t.tagName.equals("img")){

       out.write("<img ");

       Iterator attributes = t.attributeList.attributes.iterator();

       while (attributes.hasNext()) {

         Attribute element = (Attribute) attributes.next();

         if (element.name.equals("src")) {

           out.write("src=\"cid:image" + count + "\" ");

         } else { 

           out.write(element.toString() + " ");

         }

       }

       out.write(">");

         try {

           newsletter.addImage(findName(t.attributeList.attributes), count);

         } catch (MessagingException e) {

           throw new RuntimeException(e);

         } catch (IOException e) {

           throw new RuntimeException(e);

         }

         count++;

     } else {

       super.visit(t);

     }

   }

   public String findName(List vector) {

     Iterator iter = vector.iterator();

     while (iter.hasNext()) {

       HtmlDocument.Attribute element = (HtmlDocument.Attribute) iter.next();

       if (element.name.equals("src")) {

         return element.value.replaceAll("\"","");

       }

     }

     return null;

   }

 }