| // 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. |
| |
| import com.google.enterprise.adaptor.AbstractAdaptor; |
| import com.google.enterprise.adaptor.Acl; |
| import com.google.enterprise.adaptor.AdaptorContext; |
| import com.google.enterprise.adaptor.DocId; |
| import com.google.enterprise.adaptor.DocIdPusher; |
| import com.google.enterprise.adaptor.GroupPrincipal; |
| import com.google.enterprise.adaptor.PollingIncrementalLister; |
| 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.IOException; |
| import java.io.OutputStreamWriter; |
| import java.io.Writer; |
| import java.net.URI; |
| import java.util.ArrayList; |
| import java.util.Collections; |
| import java.util.Date; |
| import java.util.HashMap; |
| import java.util.logging.Logger; |
| |
| /** |
| * Demonstrates what code is necessary for putting public content onto a GSA. |
| * The key operations are: |
| * <ol> |
| * <li>providing document ids |
| * <li>providing document bytes given a document id |
| * </ol> |
| */ |
| public class HelloWorldConnector extends AbstractAdaptor implements |
| PollingIncrementalLister { |
| |
| private static final Logger log = |
| Logger.getLogger(HelloWorldConnector.class.getName()); |
| int requestCount = 1, requestCount2 = 1; |
| |
| @Override |
| public void init(AdaptorContext context) throws Exception { |
| context.setPollingIncrementalLister(this); |
| HelloWorldAuthenticator authenticator = |
| new HelloWorldAuthenticator(context); |
| context.setAuthnAuthority(authenticator); |
| context.setAuthzAuthority(authenticator); |
| context.createHttpContext("/google-response", |
| new ResponseHandler(context)); |
| } |
| |
| /** Full crawl **/ |
| @Override |
| public void getDocIds(DocIdPusher pusher) throws InterruptedException { |
| ArrayList<DocId> mockDocIds = new ArrayList<DocId>(); |
| // push docids |
| mockDocIds.add(new DocId("")); |
| pusher.pushDocIds(mockDocIds); |
| // push records |
| DocIdPusher.Record record = new DocIdPusher.Record.Builder(new DocId( |
| "1009")).setCrawlImmediately(true).setLastModified(new Date()) |
| .build(); |
| pusher.pushRecords(Collections.singleton(record)); |
| // push named resources |
| HashMap<DocId, Acl> aclParent = new HashMap<DocId, Acl>(); |
| ArrayList<Principal> permits = new ArrayList<Principal>(); |
| permits.add(new UserPrincipal("user1", "Default")); |
| aclParent.put(new DocId("fakeID"), new Acl.Builder() |
| .setEverythingCaseInsensitive().setPermits(permits) |
| .setInheritanceType(Acl.InheritanceType.PARENT_OVERRIDES) |
| .build()); |
| pusher.pushNamedResources(aclParent); |
| //push groups |
| // pusher.pushGroupDefinitions(defs, caseSensitive, handler) |
| } |
| |
| /** Gives the bytes of a document referenced with id. */ |
| @Override |
| public void getDocContent(Request req, Response resp) throws IOException { |
| DocId id = req.getDocId(); |
| log.info("DocId '" + id.getUniqueId() + "'"); |
| |
| // Hard-coded list of our doc id's |
| if ("".equals(id.getUniqueId())) { |
| // this is a the root folder, write some URLs |
| Writer writer = new OutputStreamWriter(resp.getOutputStream()); |
| writer.write("<!DOCTYPE html>\n<html><body>"); |
| writer.write("<br></br>"); |
| writer.write("<a href=\"1001\">doc_not_changed</a>"); |
| writer.write("<br></b r>"); |
| writer.write("<a href=\"1002\">doc_changed</a>"); |
| writer.write("<br></br>"); |
| writer.write("<a href=\"1003\">doc_deleted</a>"); |
| writer.write("<br></br>"); |
| writer.write("<a href=\"1004\">doc_with_meta</a>"); |
| writer.write("<br></br>"); |
| writer.write("<a href=\"1005\">doc_with_ACL</a>"); |
| writer.write("<br></br>"); |
| writer.write("<a href=\"1006\">doc_with_ACL_Inheritance</a>"); |
| writer.write("<br></br>"); |
| writer.write("<a href=\"1007\">doc_with_Fragment</a>"); |
| writer.write("<br></br>"); |
| writer.write("<a href=\"1008\">doc_with_Fragment</a>"); |
| writer.write("<br></br>"); |
| writer.write("</body></html>"); |
| writer.close(); |
| } else if ("1001".equals(id.getUniqueId())) { |
| // Example with If-Modified-Since |
| req.hasChangedSinceLastAccess(new Date()); |
| if (req.getLastAccessTime() != null) { |
| // GSA is asking whether anything changed |
| log.info("asked for last access"); |
| resp.respondNotModified(); |
| } else { |
| log.info("No asked for last access"); |
| resp.setLastModified(new Date()); |
| Writer writer = new OutputStreamWriter(resp.getOutputStream()); |
| writer.write("Menu 1001 says latte"); |
| writer.close(); |
| } |
| } else if ("1002".equals(id.getUniqueId())) { |
| // Very basic doc |
| Writer writer = new OutputStreamWriter(resp.getOutputStream()); |
| writer.write("Menu 1002 says cappuccino"); |
| writer.close(); |
| } else if ("1003".equals(id.getUniqueId())) { |
| // Alternate between doc and a 404 response |
| if (requestCount2 == 1) { |
| Writer writer = new OutputStreamWriter(resp.getOutputStream()); |
| writer.write("Menu 1003 says machiato"); |
| writer.close(); |
| requestCount2 = 0; |
| } else { |
| resp.respondNotFound(); |
| requestCount2 = 1; |
| } |
| } else if ("1004".equals(id.getUniqueId())) { |
| // doc with metdata & different display URL |
| try { |
| resp.addMetadata("flavor", "vanilla"); |
| resp.addMetadata("flavor", "hazel nuts"); |
| resp.addMetadata("taste", "strawberry"); |
| resp.setDisplayUrl(new URI("http://fake.com/a")); |
| Writer writer = new OutputStreamWriter(resp.getOutputStream()); |
| writer.write("Menu 1004 says espresso"); |
| writer.close(); |
| } catch (Exception e) { |
| log.info(e.getMessage()); |
| } |
| } else if ("1005".equals(id.getUniqueId())) { |
| // doc with ACLs |
| try { |
| ArrayList<Principal> permits = new ArrayList<Principal>(); |
| permits.add(new UserPrincipal("user1", "Default")); |
| permits.add(new UserPrincipal("eric", "Default")); |
| permits.add(new GroupPrincipal("group1", "Default")); |
| ArrayList<Principal> denies = new ArrayList<Principal>(); |
| denies.add(new UserPrincipal("user2", "Default")); |
| denies.add(new GroupPrincipal("group2", "Default")); |
| |
| resp.setAcl(new Acl.Builder() |
| .setEverythingCaseInsensitive() |
| .setInheritanceType(Acl.InheritanceType.PARENT_OVERRIDES) |
| .setPermits(permits).setDenies(denies).build()); |
| |
| Writer writer = new OutputStreamWriter(resp.getOutputStream()); |
| writer.write("Menu 1005 says americano"); |
| writer.close(); |
| } catch (Exception e) { |
| log.info(e.getMessage()); |
| } |
| } else if ("1006".equals(id.getUniqueId())) { |
| // Inherit ACLs from 1005 |
| try { |
| ArrayList<Principal> permits = new ArrayList<Principal>(); |
| permits.add(new GroupPrincipal("group3", "Default")); |
| ArrayList<Principal> denies = new ArrayList<Principal>(); |
| denies.add(new GroupPrincipal("group3", "Default")); |
| |
| resp.setAcl(new Acl.Builder() |
| .setEverythingCaseInsensitive() |
| .setInheritanceType(Acl.InheritanceType.PARENT_OVERRIDES) |
| .setInheritFrom(new DocId("1005")).setPermits(permits) |
| .setDenies(denies).build()); |
| |
| Writer writer = new OutputStreamWriter(resp.getOutputStream()); |
| writer.write("Menu 1006 says misto"); |
| writer.close(); |
| } catch (Exception e) { |
| log.info(e.getMessage()); |
| } |
| } else if ("1007".equals(id.getUniqueId())) { |
| // Inherit ACLs from 1005 & 1006 |
| try { |
| ArrayList<Principal> permits = new ArrayList<Principal>(); |
| permits.add(new GroupPrincipal("group5", "Default")); |
| |
| resp.putNamedResource("Whatever", new Acl.Builder() |
| .setEverythingCaseInsensitive() |
| .setInheritFrom(new DocId("1006")) |
| .setPermits(permits) |
| .setInheritanceType(Acl.InheritanceType.PARENT_OVERRIDES) |
| .build()); |
| |
| ArrayList<Principal> permits2 = new ArrayList<Principal>(); |
| permits2.add(new GroupPrincipal("group4", "Default")); |
| ArrayList<Principal> denies = new ArrayList<Principal>(); |
| denies.add(new GroupPrincipal("group4", "Default")); |
| |
| resp.setAcl(new Acl.Builder() |
| .setEverythingCaseInsensitive() |
| .setInheritanceType(Acl.InheritanceType.PARENT_OVERRIDES) |
| .setInheritFrom(new DocId("1005")).setPermits(permits2) |
| .setDenies(denies).build()); |
| |
| Writer writer = new OutputStreamWriter(resp.getOutputStream()); |
| writer.write("Menu 1006 says frappuccino"); |
| writer.close(); |
| } catch (Exception e) { |
| log.info(e.getMessage()); |
| } |
| } else if ("1008".equals(id.getUniqueId())) { |
| // Inherit ACLs from 1007 |
| try { |
| ArrayList<Principal> denies = new ArrayList<Principal>(); |
| denies.add(new GroupPrincipal("group5", "Default")); |
| |
| resp.setAcl(new Acl.Builder() |
| .setEverythingCaseInsensitive() |
| .setInheritanceType(Acl.InheritanceType.PARENT_OVERRIDES) |
| .setInheritFrom(new DocId("1007"), "Whatever") |
| .setDenies(denies).build()); |
| Writer writer = new OutputStreamWriter(resp.getOutputStream()); |
| writer.write("Menu 1008 says coffee"); |
| writer.close(); |
| } catch (Exception e) { |
| log.info(e.getMessage()); |
| } |
| } else if ("1009".equals(id.getUniqueId())) { |
| // Late Binding (security handled by connector) |
| try { |
| resp.setSecure(true); |
| Writer writer = new OutputStreamWriter(resp.getOutputStream()); |
| writer.write("Menu 1009 says espresso"); |
| writer.close(); |
| } catch (Exception e) { |
| log.info(e.getMessage()); |
| } |
| } else { |
| resp.respondNotFound(); |
| } |
| } |
| |
| @Override |
| public void getModifiedDocIds(DocIdPusher pusher) throws IOException, |
| InterruptedException { |
| ArrayList<DocId> mockDocIds = new ArrayList<DocId>(); |
| mockDocIds.add(new DocId("1002")); |
| pusher.pushDocIds(mockDocIds); |
| } |
| |
| /** Call default main for adaptors. */ |
| public static void main(String[] args) { |
| AbstractAdaptor.main(new HelloWorldConnector(), args); |
| } |
| } |
