87a651dbc64d717bbc9aa0ea3266da5908386c6b
[cql-java-moved-to-github.git] / src / org / z3950 / zing / cql / CQLTermNode.java
1 // $Id: CQLTermNode.java,v 1.14 2002-12-09 16:55:19 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.14 2002-12-09 16:55:19 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); // ### change not seen by caller
110             pos = "first";
111         }
112         int len = text.length();
113         if (len > 0 && text.substring(len-1, len).equals("^")) {
114             text = text.substring(0, len-1); // ### change not seen by caller
115             pos = pos.equals("first") ? "firstAndLast" : "last";
116             // ### in the firstAndLast case, the standard
117             //  pqf.properties file specifies that we generate a
118             //  completeness=whole-field attributem, which means that
119             //  we don't generate a position attribute at all.  Do we
120             //  care?  Does it matter?
121         }
122
123         attr = config.getProperty("position." + pos);
124         if (attr == null)
125             throw new UnknownPositionException(pos);
126         attrs.add(attr);
127
128         attr = config.getProperty("structure." + rel);
129         if (attr == null)
130             attr = config.getProperty("structure.*");
131         attrs.add(attr);
132
133         return attrs;
134     }
135
136     public String toPQF(Properties config) throws PQFTranslationException {
137         Vector attrs = getAttrs(config);
138
139         String attr, s = "";
140         for (int i = 0; i < attrs.size(); i++) {
141             attr = (String) attrs.get(i);
142             s += "@attr " + Utils.replaceString(attr, " ", " @attr ") + " ";
143         }
144
145         String text = term;
146         if (text.length() > 0 && text.substring(0, 1).equals("^"))
147             text = text.substring(1);
148         int len = text.length();
149         if (len > 0 && text.substring(len-1, len).equals("^"))
150             text = text.substring(0, len-1);
151
152         return s + maybeQuote(text);
153     }
154
155     static String maybeQuote(String str) {
156         // There _must_ be a better way to make this test ...
157         if (str.length() == 0 ||
158             str.indexOf('"') != -1 ||
159             str.indexOf(' ') != -1 ||
160             str.indexOf('\t') != -1 ||
161             str.indexOf('=') != -1 ||
162             str.indexOf('<') != -1 ||
163             str.indexOf('>') != -1 ||
164             str.indexOf('/') != -1 ||
165             str.indexOf('(') != -1 ||
166             str.indexOf(')') != -1) {
167             str = '"' + Utils.replaceString(str, "\"", "\\\"") + '"';
168         }
169
170         return str;
171     }
172
173     public byte[] toType1(Properties config) throws PQFTranslationException {
174         String text = term;
175         if (text.length() > 0 && text.substring(0, 1).equals("^"))
176             text = text.substring(1);
177         int len = text.length();
178         if (len > 0 && text.substring(len-1, len).equals("^"))
179             text = text.substring(0, len-1);
180
181         String attr, attrList, term = maybeQuote(text);
182         System.out.println("in CQLTermNode.toType101(): PQF=" + toPQF(config));
183         byte[] operand = new byte[text.length()+100];
184         int i, j, offset, type, value;
185         offset = putTag(CONTEXT, 0, CONSTRUCTED, operand, 0); // op
186         operand[offset++]=(byte)(0x80&0xff); // indefinite length
187         offset = putTag(CONTEXT, 102, CONSTRUCTED, operand, offset); // AttributesPlusTerm
188         operand[offset++] = (byte)(0x80&0xff); // indefinite length
189         offset = putTag(CONTEXT, 44, CONSTRUCTED, operand, offset); // AttributeList
190         operand[offset++] = (byte)(0x80&0xff); // indefinite length
191         offset = putTag(UNIVERSAL, SEQUENCE, CONSTRUCTED, operand, offset);
192         operand[offset++] = (byte)(0x80&0xff);
193
194         Vector attrs = getAttrs(config);
195         for(i = 0; i < attrs.size(); i++) {
196             attrList = (String) attrs.get(i);
197             java.util.StringTokenizer st =
198                 new java.util.StringTokenizer(attrList);
199             while (st.hasMoreTokens()) {
200                 attr = st.nextToken();
201                 j = attr.indexOf('=');
202                 offset = putTag(CONTEXT, 120, PRIMITIVE, operand, offset);
203                 type = Integer.parseInt(attr.substring(0, j));
204                 offset = putLen(numLen(type), operand, offset);
205                 offset = putNum(type, operand, offset);
206
207                 offset = putTag(CONTEXT, 121, PRIMITIVE, operand, offset);
208                 value = Integer.parseInt(attr.substring(j+1));
209                 offset = putLen(numLen(value), operand, offset);
210                 offset = putNum(value, operand, offset);
211             }
212         }
213         operand[offset++] = 0x00; // end of SEQUENCE
214         operand[offset++] = 0x00;
215         operand[offset++] = 0x00; // end of AttributeList
216         operand[offset++] = 0x00;
217
218         offset = putTag(CONTEXT, 45, PRIMITIVE, operand, offset); // general Term
219         byte[] t = term.getBytes();
220         offset = putLen(t.length, operand, offset);
221         System.arraycopy(t, 0, operand, offset, t.length);
222         offset += t.length;
223
224         operand[offset++] = 0x00; // end of AttributesPlusTerm
225         operand[offset++] = 0x00;
226         operand[offset++] = 0x00; // end of Operand
227         operand[offset++] = 0x00;
228         byte[] o = new byte[offset];
229         System.arraycopy(operand, 0, o, 0, offset);
230         return o;
231     }
232 }