src/org/z3950/zing/cql/CQLTermNode.java

   1 // $Id: CQLTermNode.java,v 1.13 2002-12-06 12:35:42 mike Exp $
   2
   3 package org.z3950.zing.cql;
   4 import java.util.Properties;
   5 import java.util.Vector;
   6
   7
   8 /**
   9  * Represents a terminal node in a CQL parse-tree.
  10  * A term node consists of the term String itself, together with,
  11  * optionally, a qualifier string and a relation.  Neither or both of
  12  * these must be provided - you can't have a qualifier without a
  13  * relation or vice versa.
  14  *
  15  * @version     $Id: CQLTermNode.java,v 1.13 2002-12-06 12:35:42 mike Exp $
  16  */
  17 public class CQLTermNode extends CQLNode {
  18     private String qualifier;
  19     private CQLRelation relation;
  20     private String term;
  21
  22     /**
  23      * Creates a new term node with the specified <TT>qualifier</TT>,
  24      * <TT>relation</TT> and <TT>term</TT>.  The first two may be
  25      * <TT>null</TT>, but the <TT>term</TT> may not.
  26      */
  27     public CQLTermNode(String qualifier, CQLRelation relation, String term) {
  28         this.qualifier = qualifier;
  29         this.relation = relation;
  30         this.term = term;
  31     }
  32
  33     public String getQualifier() { return qualifier; }
  34     public CQLRelation getRelation() { return relation; }
  35     public String getTerm() { return term; }
  36
  37     public String toXCQL(int level, Vector prefixes) {
  38         return (indent(level) + "<searchClause>\n" +
  39                 renderPrefixes(level+1, prefixes) +
  40                 indent(level+1) + "<index>" + xq(qualifier) + "</index>\n" +
  41                 relation.toXCQL(level+1, new Vector()) +
  42                 indent(level+1) + "<term>" + xq(term) + "</term>\n" +
  43                 indent(level) + "</searchClause>\n");
  44     }
  45
  46     public String toCQL() {
  47         String quotedQualifier = maybeQuote(qualifier);
  48         String quotedTerm = maybeQuote(term);
  49         String res = quotedTerm;
  50
  51         if (!qualifier.equalsIgnoreCase("srw.serverChoice")) {
  52             // ### We don't always need spaces around `relation'.
  53             res = quotedQualifier + " " + relation.toCQL() + " " + quotedTerm;
  54         }
  55
  56         return res;
  57     }
  58
  59     // ### Interaction between this and its callers is not good as
  60     //  regards truncation of the term and generation of truncation
  61     //  attributes.  Change the interface to fix this.
  62     private Vector getAttrs(Properties config) throws PQFTranslationException {
  63         Vector attrs = new Vector();
  64
  65         // Do this first so that if any other truncation or
  66         // completeness attributes are generated, they "overwrite"
  67         // those specified here.
  68         //
  69         //  ### This approach relies on an unpleasant detail of Index
  70         //      Data's (admittedly definitive) implementation of PQF,
  71         //      and should not relied upon.
  72         //
  73         String attr = config.getProperty("always");
  74         if (attr != null)
  75             attrs.add(attr);
  76
  77         attr = config.getProperty("qualifier." + qualifier);
  78         if (attr == null)
  79             throw new UnknownQualifierException(qualifier);
  80         attrs.add(attr);
  81
  82         String rel = relation.getBase();
  83         if (rel.equals("=")) {
  84             rel = "eq";
  85         } else if (rel.equals("<=")) {
  86             rel = "le";
  87         } else if (rel.equals(">=")) {
  88             rel = "ge";
  89         }
  90         // ### Handling "any" and "all" properly would involve breaking
  91         // the string down into a bunch of individual words and ORring
  92         // or ANDing them together.  Another day.
  93         attr = config.getProperty("relation." + rel);
  94         if (attr == null)
  95             throw new UnknownRelationException(rel);
  96         attrs.add(attr);
  97
  98         String[] mods = relation.getModifiers();
  99         for (int i = 0; i < mods.length; i++) {
 100             attr = config.getProperty("relationModifier." + mods[i]);
 101             if (attr == null)
 102                 throw new UnknownRelationModifierException(mods[i]);
 103             attrs.add(attr);
 104         }
 105
 106         String pos = "any";
 107         String text = term;
 108         if (text.length() > 0 && text.substring(0, 1).equals("^")) {
 109             text = text.substring(1);
 110             pos = "first";
 111         }
 112         // ### add back the last-in-field stuff
 113         attr = config.getProperty("position." + pos);
 114         if (attr == null)
 115             throw new UnknownPositionException(pos);
 116         attrs.add(attr);
 117
 118         attr = config.getProperty("structure." + rel);
 119         if (attr == null)
 120             attr = config.getProperty("structure.*");
 121         attrs.add(attr);
 122
 123         return attrs;
 124     }
 125
 126     public String toPQF(Properties config) throws PQFTranslationException {
 127         Vector attrs = getAttrs(config);
 128
 129         String attr, s = "";
 130         for (int i = 0; i < attrs.size(); i++) {
 131             attr = (String) attrs.get(i);
 132             s += "@attr " + Utils.replaceString(attr, " ", " @attr ") + " ";
 133         }
 134
 135         String text = term;
 136         if (text.length() > 0 && text.substring(0, 1).equals("^"))
 137             text = text.substring(1);
 138         return s + maybeQuote(text);
 139     }
 140
 141     static String maybeQuote(String str) {
 142         // There _must_ be a better way to make this test ...
 143         if (str.length() == 0 ||
 144             str.indexOf('"') != -1 ||
 145             str.indexOf(' ') != -1 ||
 146             str.indexOf('\t') != -1 ||
 147             str.indexOf('=') != -1 ||
 148             str.indexOf('<') != -1 ||
 149             str.indexOf('>') != -1 ||
 150             str.indexOf('/') != -1 ||
 151             str.indexOf('(') != -1 ||
 152             str.indexOf(')') != -1) {
 153             str = '"' + Utils.replaceString(str, "\"", "\\\"") + '"';
 154         }
 155
 156         return str;
 157     }
 158
 159     /**
 160      * ### Document this!
 161      */
 162     public byte[] toType1(Properties config) throws PQFTranslationException {
 163         String text = term;
 164         if (text.length() > 0 && text.substring(0, 1).equals("^"))
 165             text = text.substring(1);
 166         String attr, attrList, term = maybeQuote(text);
 167         System.out.println("in CQLTermNode.toType101(): PQF=" + toPQF(config));
 168         byte[] operand = new byte[text.length()+100];
 169         int i, j, offset, type, value;
 170         offset = putTag(CONTEXT, 0, CONSTRUCTED, operand, 0); // op
 171         operand[offset++]=(byte)(0x80&0xff); // indefinite length
 172         offset = putTag(CONTEXT, 102, CONSTRUCTED, operand, offset); // AttributesPlusTerm
 173         operand[offset++] = (byte)(0x80&0xff); // indefinite length
 174         offset = putTag(CONTEXT, 44, CONSTRUCTED, operand, offset); // AttributeList
 175         operand[offset++] = (byte)(0x80&0xff); // indefinite length
 176         offset = putTag(UNIVERSAL, SEQUENCE, CONSTRUCTED, operand, offset);
 177         operand[offset++] = (byte)(0x80&0xff);
 178
 179         Vector attrs = getAttrs(config);
 180         for(i = 0; i < attrs.size(); i++) {
 181             attrList = (String) attrs.get(i);
 182             java.util.StringTokenizer st =
 183                 new java.util.StringTokenizer(attrList);
 184             while (st.hasMoreTokens()) {
 185                 attr = st.nextToken();
 186                 j = attr.indexOf('=');
 187                 offset = putTag(CONTEXT, 120, PRIMITIVE, operand, offset);
 188                 type = Integer.parseInt(attr.substring(0, j));
 189                 offset = putLen(numLen(type), operand, offset);
 190                 offset = putNum(type, operand, offset);
 191
 192                 offset = putTag(CONTEXT, 121, PRIMITIVE, operand, offset);
 193                 value = Integer.parseInt(attr.substring(j+1));
 194                 offset = putLen(numLen(value), operand, offset);
 195                 offset = putNum(value, operand, offset);
 196             }
 197         }
 198         operand[offset++] = 0x00; // end of SEQUENCE
 199         operand[offset++] = 0x00;
 200         operand[offset++] = 0x00; // end of AttributeList
 201         operand[offset++] = 0x00;
 202
 203         offset = putTag(CONTEXT, 45, PRIMITIVE, operand, offset); // general Term
 204         byte[] t = term.getBytes();
 205         offset = putLen(t.length, operand, offset);
 206         System.arraycopy(t, 0, operand, offset, t.length);
 207         offset += t.length;
 208
 209         operand[offset++] = 0x00; // end of AttributesPlusTerm
 210         operand[offset++] = 0x00;
 211         operand[offset++] = 0x00; // end of Operand
 212         operand[offset++] = 0x00;
 213         byte[] o = new byte[offset];
 214         System.arraycopy(operand, 0, o, 0, offset);
 215         return o;
 216     }
 217 }