tool for feeding groups from csv
diff --git a/build.xml b/build.xml
index ee1f494..5957194 100644
--- a/build.xml
+++ b/build.xml
@@ -415,6 +415,23 @@
</java>
</target>
+ <target name="groups_from_csv" depends="build"
+ description="Run groups from csv">
+ <java classpath="${build-src.dir}" fork="true"
+ classname="${adaptor.pkg.name}.examples.GroupDefinitionsFromCsv">
+ <classpath refid="adaptorlib.run.classpath"/>
+ <sysproperty key="java.util.logging.config.file"
+ value="${src.dir}/logging.properties"/>
+ <sysproperty key="javax.net.ssl.keyStore" file="keys.jks"/>
+ <sysproperty key="javax.net.ssl.keyStoreType" value="jks"/>
+ <sysproperty key="javax.net.ssl.keyStorePassword" value="changeit"/>
+ <sysproperty key="javax.net.ssl.trustStore" file="cacerts.jks"/>
+ <sysproperty key="javax.net.ssl.trustStoreType" value="jks"/>
+ <sysproperty key="javax.net.ssl.trustStorePassword" value="changeit"/>
+ <arg line="${adaptor.args}"/>
+ </java>
+ </target>
+
<target name="sim" depends="build" description="run simulator">
<java classpath="${build-src.dir}" fork="true"
classname="com.google.enterprise.adaptor.experimental.Sim">
diff --git a/src/com/google/enterprise/adaptor/examples/GroupDefinitionsFromCsv.java b/src/com/google/enterprise/adaptor/examples/GroupDefinitionsFromCsv.java
new file mode 100644
index 0000000..25b301f
--- /dev/null
+++ b/src/com/google/enterprise/adaptor/examples/GroupDefinitionsFromCsv.java
@@ -0,0 +1,139 @@
+// Copyright 2014 Google Inc. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package com.google.enterprise.adaptor.examples;
+
+import com.google.enterprise.adaptor.AbstractAdaptor;
+import com.google.enterprise.adaptor.AdaptorContext;
+import com.google.enterprise.adaptor.Config;
+import com.google.enterprise.adaptor.DocIdPusher;
+import com.google.enterprise.adaptor.GroupPrincipal;
+import com.google.enterprise.adaptor.Principal;
+import com.google.enterprise.adaptor.Request;
+import com.google.enterprise.adaptor.Response;
+import com.google.enterprise.adaptor.UserPrincipal;
+
+import java.io.BufferedReader;
+import java.io.File;
+import java.io.FileReader;
+import java.io.IOException;
+import java.nio.charset.Charset;
+import java.util.ArrayList;
+import java.util.LinkedList;
+import java.util.List;
+import java.util.Map;
+import java.util.TreeMap;
+import java.util.logging.Logger;
+
+/** Reads memberships from CSV, forms groups, and sends to to GSA. */
+public class GroupDefinitionsFromCsv extends AbstractAdaptor {
+ private static final Logger log
+ = Logger.getLogger(GroupDefinitionsFromCsv.class.getName());
+ private static Charset encoding = Charset.forName("UTF-8");
+
+ private File csvFile;
+ private String domain;
+
+ @Override
+ public void initConfig(Config config) {
+ config.addKey("csv.filename", null);
+ config.addKey("csv.domain", null);
+ }
+
+ @Override
+ public void init(AdaptorContext context) throws Exception {
+ String fname = context.getConfig().getValue("csv.filename");
+ csvFile = new File(fname);
+ if (!csvFile.exists() || !csvFile.isFile()) {
+ throw new IllegalStateException("cannot find file: " + fname);
+ }
+ domain = context.getConfig().getValue("csv.domain");
+ }
+
+ @Override
+ public void getDocIds(DocIdPusher pusher) throws InterruptedException {
+ try {
+ getDocIdsHelper(pusher);
+ } catch (IOException e) {
+ throw new RuntimeException(e);
+ }
+ }
+
+ private void getDocIdsHelper(DocIdPusher pusher) throws InterruptedException,
+ IOException {
+ BufferedReader br = new BufferedReader(new FileReader(csvFile));
+ String line = br.readLine();
+ if (null == line) {
+ throw new IllegalStateException("no content in csv file");
+ }
+ if (!"GROUPID,ENTITYID".equalsIgnoreCase(line)) {
+ throw new IllegalStateException("invalid csv header line");
+ }
+
+ Map<String, List<String>> groupdefs
+ = new TreeMap<String, List<String>>();
+ while ((line = br.readLine()) != null) {
+ String parts[] = line.split(",", -1);
+ if (2 != parts.length) {
+ throw new IllegalStateException("invalid csv line: " + line);
+ }
+ String groupid = parts[0];
+ if (!groupdefs.containsKey(groupid)) {
+ groupdefs.put(groupid, new LinkedList<String>());
+ }
+ if (!"".equals(parts[1])) {
+ String entity = parts[1];
+ groupdefs.get(groupid).add(entity);
+ }
+ }
+
+ boolean caseSensitive = true;
+ pusher.pushGroupDefinitions(convert(groupdefs), caseSensitive);
+ }
+
+ private String makeName(String id) {
+ return "" + id + "@" + domain;
+ }
+
+ private Map<GroupPrincipal, List<Principal>> convert(
+ Map<String, List<String>> src) {
+ Map<GroupPrincipal, List<Principal>> dest
+ = new TreeMap<GroupPrincipal, List<Principal>>();
+ for (Map.Entry<String, List<String>> e : src.entrySet()) {
+ int nentities = e.getValue().size();
+ List<Principal> entities = new ArrayList<Principal>(nentities);
+ for (int i = 0; i < nentities; i++) {
+ String id = e.getValue().get(i);
+ String name = makeName(id);
+ boolean entityIsGroup = src.containsKey(id);
+ if (entityIsGroup) {
+ entities.add(new GroupPrincipal(name));
+ } else {
+ entities.add(new UserPrincipal(name));
+ }
+ }
+ dest.put(new GroupPrincipal(makeName(e.getKey())), entities);
+ }
+ return dest;
+ }
+
+ @Override
+ public void getDocContent(Request req, Response res) throws IOException {
+ res.respondNotFound();
+ }
+
+ public static void main(String args[]) {
+ AbstractAdaptor.main(new GroupDefinitionsFromCsv(), args);
+ }
+}
