package org.apache.torque.mojo.morph;

 import org.codehaus.plexus.util.StringUtils;

 public class DataMorpher extends Morpher {

         // Ant prologue

         String antPrologue = "<?xml version=\"1.0\"?>";

         // New prologue

         String newPrologue = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>";

         // The hard coded Ant impex document type

         String antDocType = "<!DOCTYPE dataset SYSTEM \"data.dtd\">";

         public DataMorpher() {

                 this(null, null);

         }

         public DataMorpher(MorphRequest morphRequest, String artifactId) {

                 super(morphRequest, artifactId);

         }

         protected String getNewDocType() {

                 return "<!DOCTYPE dataset SYSTEM \"" + getArtifactId() + ".dtd\">";

         }

         protected String getMorphedContents(String contents) {

                 contents = StringUtils.replace(contents, antPrologue, newPrologue);

                 return StringUtils.replace(contents, antDocType, getNewDocType());

         }

         /**

          * Return true if we need to morph this file

          */

         protected boolean isMorphNeeded(String contents) {

                 // Look for the DTD the Maven Impex Plugin uses

                 int pos = contents.indexOf(getNewDocType());

                 if (pos == -1) {

                         // It isn't there so we should morph

                         return true;

                 } else {

                         // It is already there, we are good to go

                         return false;

                 }

         }

         public String getAntPrologue() {

                 return antPrologue;

         }

         public void setAntPrologue(String antPrologue) {

                 this.antPrologue = antPrologue;

         }

         public String getNewPrologue() {

                 return newPrologue;

         }

         public void setNewPrologue(String newPrologue) {

                 this.newPrologue = newPrologue;

         }

 }