Finish (more or less) to CQL-to-PQF translator.

[cql-java-moved-to-github.git] / src / org / z3950 / zing / cql / CQLTermNode.java

diff --git a/src/org/z3950/zing/cql/CQLTermNode.java b/src/org/z3950/zing/cql/CQLTermNode.java
index 64a0a84..9e7a7b9 100644 (file)
--- a/src/org/z3950/zing/cql/CQLTermNode.java
+++ b/src/org/z3950/zing/cql/CQLTermNode.java
@@ -1,55 +1,137 @@
-// $Id: CQLTermNode.java,v 1.3 2002-10-25 16:56:43 mike Exp $
+// $Id: CQLTermNode.java,v 1.8 2002-11-06 20:13:45 mike Exp $
 
 package org.z3950.zing.cql;
+import java.util.Properties;
+import java.util.Vector;
 
 
 /**
- * Represents a terminal node in a CQL parse-tree ...
- * ###
+ * Represents a terminal node in a CQL parse-tree.
+ * A term node consists of the term String itself, together with,
+ * optionally, a qualifier string and a relation.  Neither or both of
+ * these must be provided - you can't have a qualifier without a
+ * relation or vice versa.
  *
- * @version    $Id: CQLTermNode.java,v 1.3 2002-10-25 16:56:43 mike Exp $
+ * @version    $Id: CQLTermNode.java,v 1.8 2002-11-06 20:13:45 mike Exp $
  */
 public class CQLTermNode extends CQLNode {
     private String qualifier;
-    private String relation;
+    private CQLRelation relation;
     private String term;
 
-    public CQLTermNode(String qualifier, String relation, String term) {
+    /**
+     * Creates a new term node with the specified <TT>qualifier</TT>,
+     * <TT>relation</TT> and <TT>term</TT>.  The first two may be
+     * <TT>null</TT>, but the <TT>term</TT> may not.
+     */
+    public CQLTermNode(String qualifier, CQLRelation relation, String term) {
        this.qualifier = qualifier;
        this.relation = relation;
        this.term = term;
     }
 
-    String toXCQL(int level) {
+    public String getQualifier() { return qualifier; }
+    public CQLRelation getRelation() { return relation; }
+    public String getTerm() { return term; }
+
+    public String toXCQL(int level) {
        return (indent(level) + "<searchClause>\n" +
-               indent(level+1) + "<index>" + qualifier + "<index>\n" +
-               indent(level+1) + "<relation>" + relation + "<relation>\n" +
-               indent(level+1) + "<term>" + term + "<term>\n" +
+               indent(level+1) + "<index>" + xq(qualifier) + "</index>\n" +
+               relation.toXCQL(level+1) +
+               indent(level+1) + "<term>" + xq(term) + "</term>\n" +
                indent(level) + "</searchClause>\n");
     }
 
-    String toCQL() {
-       String quotedTerm = term;
+    public String toCQL() {
+       String quotedQualifier = maybeQuote(qualifier);
+       String quotedTerm = maybeQuote(term);
+       String res = quotedTerm;
+
+       if (!qualifier.equalsIgnoreCase("srw.serverChoice")) {
+           // ### We don't always need spaces around `relation'.
+           res = quotedQualifier + " " + relation.toCQL() + " " + quotedTerm;
+       }
+
+       return res;
+    }
+
+    public String toPQF(Properties config) throws PQFTranslationException {
+       Vector attrs = new Vector();
+
+       String attr;
+       attr = config.getProperty("qualifier." + qualifier);
+       if (attr == null)
+           throw new UnknownQualifierException(qualifier);
+       attrs.add(attr);
+
+       String rel = relation.getBase();
+       if (rel.equals("=")) {
+           rel = "eq";
+       } else if (rel.equals("<=")) {
+           rel = "le";
+       } else if (rel.equals(">=")) {
+           rel = "ge";
+       }
+       // ### Handling "any" and "all" properly would involve breaking
+       // the string down into a bunch of individual words and ORring
+       // or ANDing them together.  Another day.
+       attr = config.getProperty("relation." + rel);
+       if (attr == null)
+           throw new UnknownRelationException(rel);
+       attrs.add(attr);
 
-       if (quotedTerm.indexOf('"') != -1) {
-           // ### precede each '"' with a '/'
+       String[] mods = relation.getModifiers();
+       for (int i = 0; i < mods.length; i++) {
+           attr = config.getProperty("relationModifier." + mods[i]);
+           if (attr == null)
+               throw new UnknownRelationModifierException(mods[i]);
+           attrs.add(attr);
        }
 
-       // ### There must be a better way ...
-       if (quotedTerm.indexOf('"') != -1 ||
-           quotedTerm.indexOf(' ') != -1 ||
-           quotedTerm.indexOf('\t') != -1 ||
-           quotedTerm.indexOf('=') != -1 ||
-           quotedTerm.indexOf('<') != -1 ||
-           quotedTerm.indexOf('>') != -1 ||
-           quotedTerm.indexOf('/') != -1 ||
-           quotedTerm.indexOf('(') != -1 ||
-           quotedTerm.indexOf(')') != -1) {
-           quotedTerm = '"' + quotedTerm + '"';
+       String pos = "unanchored";
+       String text = term;
+       if (text.length() > 0 && text.substring(0, 1).equals("^")) {
+           text = text.substring(1);
+           pos = "anchored";
+       }
+       attr = config.getProperty("position." + pos);
+       if (attr == null)
+           throw new UnknownPositionException(pos);
+       attrs.add(attr);
+
+       attr = config.getProperty("structure." + rel);
+       if (attr == null)
+           attr = config.getProperty("structure.*");
+       attrs.add(attr);
+
+       attr = config.getProperty("always");
+       if (attr != null)
+           attrs.add(attr);
+
+       String s = "";
+       for (int i = 0; i < attrs.size(); i++) {
+           attr = (String) attrs.get(i);
+           s += "@attr " + Utils.replaceString(attr, " ", " @attr ") + " ";
+       }
+
+       return s + maybeQuote(text);
+    }
+
+    static String maybeQuote(String str) {
+       // There _must_ be a better way to make this test ...
+       if (str.length() == 0 ||
+           str.indexOf('"') != -1 ||
+           str.indexOf(' ') != -1 ||
+           str.indexOf('\t') != -1 ||
+           str.indexOf('=') != -1 ||
+           str.indexOf('<') != -1 ||
+           str.indexOf('>') != -1 ||
+           str.indexOf('/') != -1 ||
+           str.indexOf('(') != -1 ||
+           str.indexOf(')') != -1) {
+           str = '"' + Utils.replaceString(str, "\"", "\\\"") + '"';
        }
 
-       // ### The qualifier may need quoting.
-       // ### We don't always need spaces around `relation'.
-       return qualifier + " " + relation + " " + quotedTerm;
+       return str;
     }
 }