summaryrefslogtreecommitdiffstats
path: root/branches/sca-java-1.x/modules/domain-search/src/main/java/org/apache/tuscany/sca/domain/search/impl/ContributionDocumentProcessor.java
blob: 615a84b6dc3d06d20c540abe1e177a030110c55f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
package org.apache.tuscany.sca.domain.search.impl;

import org.apache.lucene.document.Field;

import org.apache.tuscany.sca.assembly.Composite;
import org.apache.tuscany.sca.contribution.Artifact;
import org.apache.tuscany.sca.contribution.Contribution;
import org.apache.tuscany.sca.domain.search.DocumentMap;
import org.apache.tuscany.sca.domain.search.DocumentProcessor;
import org.apache.tuscany.sca.domain.search.Result;

public class ContributionDocumentProcessor implements DocumentProcessor {

	public Document process(DocumentProcessor parentProcessor,
			DocumentMap documents, Object object, Document doc, String parent) {

		if (object instanceof Contribution) {
			Contribution contribution = (Contribution) object;
			String uri = contribution.getURI();

			if (uri != null) {

				if (uri.length() == 0) {
					uri = null;

				} else {
					
					parent += DomainPathAnalyzer.PATH_SEPARATOR
							+ SearchFields.CONTRIBUTION_FIELD
							+ DomainPathAnalyzer.TYPE_SEPARATOR + uri;
					
				}

			}

			if (uri != null && doc == null) {
				doc = documents.get(uri);

				doc.add(new Field(SearchFields.CONTRIBUTION_FIELD, uri,
						Field.Store.YES, Field.Index.ANALYZED));

			} else {
				doc = FAKE_DOCUMENT;
			}

			for (Artifact artifact : contribution.getArtifacts()) {
				Document artifactDoc = parentProcessor.process(parentProcessor,
						documents, artifact, null, parent);

				if (uri != null) {

					artifactDoc.add(new Field(SearchFields.PARENT_FIELD,
							parent, Field.Store.YES, Field.Index.ANALYZED));

				}

			}

			// for (Import imprt : contribution.getImports()) {
			// Document importDoc = processors.process(processors, documents,
			// imprt, null);
			//
			// if (uri != null) {
			//
			// importDoc.add(new Field(SearchFields.IMPORTEDBY_FIELD, uri,
			// Field.Store.YES, Field.Index.ANALYZED));
			//
			// }
			//
			// }
			//
			// for (Export export : contribution.getExports()) {
			// Document exportDoc = processors.process(processors, documents,
			// export, null);
			//
			// if (uri != null) {
			//
			// exportDoc.add(new Field(SearchFields.EXPORTEDBY_FIELD, uri,
			// Field.Store.YES, Field.Index.ANALYZED));
			//
			// }
			//
			// }

			if (!object.getClass().getSimpleName().contains("Workspace")) {

				for (Composite composite : contribution.getDeployables()) {
					Document compositeDoc = parentProcessor.process(parentProcessor,
							documents, composite, null, parent);

					if (uri != null) {

						compositeDoc.add(new Field(SearchFields.PARENT_FIELD,
								parent, Field.Store.YES, Field.Index.ANALYZED));

					}

				}

			}

			return doc;

		}

		throw new IllegalArgumentException();

	}

	public Object getDocumentKey(Object object) {

		if (object instanceof Contribution) {
			Contribution contribution = (Contribution) object;
			String uri = contribution.getURI();

			if (uri != null && uri.length() == 0) {
				return null;
			}

			return uri;

		}

		throw new IllegalArgumentException();

	}

	public Result processDocument(org.apache.lucene.document.Document document,
			Result result) {
		String contributionName = document.get(SearchFields.CONTRIBUTION_FIELD);

		if (contributionName != null) {

			if (result == null) {
				result = new ContributionResult(contributionName);

			} else if (!(result instanceof ContributionResult)) {
				throw new IllegalArgumentException();
			}

			result.setName(contributionName);

		}

		return result;

	}

}