framework/pull-agent/src/main/java/org/apache/manifoldcf/crawler/DefineRepositoryConnection.java

/* $Id: DefineRepositoryConnection.java 988245 2010-08-23 18:39:35Z kwright $ */

/**
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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 org.apache.manifoldcf.crawler;

import java.io.*;
import org.apache.manifoldcf.core.interfaces.*;
import org.apache.manifoldcf.crawler.interfaces.*;
import org.apache.manifoldcf.crawler.system.*;
import java.util.*;

/** This class is used during testing.
*/
public class DefineRepositoryConnection
{
  public static final String _rcsid = "@(#)$Id: DefineRepositoryConnection.java 988245 2010-08-23 18:39:35Z kwright $";

  private DefineRepositoryConnection()
  {
  }


  public static void main(String[] args)
  {
    if (args.length < 5)
    {
      System.err.println("Usage: DefineRepositoryConnection <connection_name> <description> <connector_class> <authority_name> <pool_max> <param1>=<value1> ...");
      System.exit(1);
    }

    String connectionName = args[0];
    String description = args[1];
    String connectorClass = args[2];
    String authorityName = args[3];
    String poolMax = args[4];


    try
    {
      IThreadContext tc = ThreadContextFactory.make();
      ManifoldCF.initializeEnvironment(tc);
      IRepositoryConnectionManager mgr = RepositoryConnectionManagerFactory.make(tc);
      IRepositoryConnection conn = mgr.create();
      conn.setName(connectionName);
      conn.setDescription(description);
      conn.setClassName(connectorClass);
      if (authorityName.length() > 0)
        conn.setACLAuthority(authorityName);
      conn.setMaxConnections(new Integer(poolMax).intValue());
      ConfigParams x = conn.getConfigParams();
      int i = 5;
      while (i < args.length)
      {
        String arg = args[i++];
        // Parse
        int pos = arg.indexOf("=");
        if (pos == -1)
          throw new ManifoldCFException("Argument missing =");
        String name = arg.substring(0,pos);
        String value = arg.substring(pos+1);
        if (name.endsWith("assword"))
          x.setObfuscatedParameter(name,value);
        else
          x.setParameter(name,value);
      }

      // Now, save
      mgr.save(conn);

    }
    catch (Exception e)
    {
      e.printStackTrace();
      System.exit(2);
    }
  }

}