filemgr/src/main/java/org/apache/oodt/cas/filemgr/ingest/CmdLineIngester.java

/*
 * 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.oodt.cas.filemgr.ingest;

//OODT imports
import org.apache.oodt.cas.filemgr.structs.exceptions.IngestException;
import org.apache.oodt.cas.metadata.MetExtractor;
import org.apache.oodt.cas.metadata.Metadata;
import org.apache.oodt.cas.metadata.SerializableMetadata;
import org.apache.oodt.cas.metadata.util.GenericMetadataObjectFactory;

//JDK imports
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.List;
import java.util.Vector;
import java.util.logging.Level;
import java.util.logging.Logger;

/**
 * @author mattmann
 * @author bfoster
 * @version $Revision$
 * 
 * <p>
 * A command line I/F for {@link Product} ingestion.
 * </p>.
 */
public class CmdLineIngester extends StdIngester {

    /* our log stream */
    private static final Logger LOG = Logger.getLogger(CmdLineIngester.class
            .getName());

    public CmdLineIngester(String serviceFactory) {
        super(serviceFactory);
    }

    /**
     * @param args
     * @throws IngestException
     * @throws MalformedURLException
     */
    public static void main(String[] args) throws Exception {
        String usage = CmdLineIngester.class.getName()
                + " --url <filemgr url> [options]\n"
                + "[--extractor <met extractor class name> <met conf file path>|"
                + "--metFile <met file path>]\n"
                + "--transferer <data transfer service factory>\n"
                + "[--file <full path> | --in reads list of files from STDIN]\n";

        String fmUrlStr = null, extractorClassName = null, filePath = null;
        String transferServiceFactory = null, metConfFilePath = null;
        String metFilePath = null;
        boolean readFromStdin = false;

        for (int i = 0; i < args.length; i++) {
            if (args[i].equals("--url")) {
                fmUrlStr = args[++i];
            } else if (args[i].equals("--extractor")) {
                extractorClassName = args[++i];
                metConfFilePath = args[++i];
            } else if (args[i].equals("--file")) {
                filePath = args[++i];
            } else if (args[i].equals("--in")) {
                readFromStdin = true;
            } else if (args[i].equals("--transferer")) {
                transferServiceFactory = args[++i];
            } else if (args[i].equals("--metFile")) {
                metFilePath = args[++i];
            }
        }

        if (fmUrlStr == null
                || ((extractorClassName == null || metConfFilePath == null) && metFilePath == null)
                || (metFilePath != null && (metConfFilePath != null || extractorClassName != null))
                || transferServiceFactory == null
                || (filePath == null && !readFromStdin)
                || (readFromStdin && metFilePath != null)
                || (readFromStdin && filePath != null)) {
            System.err.println(usage);
            System.exit(1);
        }

        CmdLineIngester ingester = new CmdLineIngester(transferServiceFactory);
        MetExtractor extractor = null;
        if (readFromStdin) {
            List prods = readProdFilesFromStdin();
            extractor = GenericMetadataObjectFactory
                    .getMetExtractorFromClassName(extractorClassName);
            ingester.ingest(new URL(fmUrlStr), prods, extractor, new File(
                    metConfFilePath));
        } else {
            String productID = null;
            if (metFilePath != null) {
                productID = ingester.ingest(new URL(fmUrlStr), new File(
                        filePath), new SerializableMetadata(
                        new FileInputStream(metFilePath)));
            } else {
                extractor = GenericMetadataObjectFactory
                        .getMetExtractorFromClassName(extractorClassName);
                productID = ingester.ingest(new URL(fmUrlStr), new File(
                        filePath), extractor, new File(metConfFilePath));
            }

            System.out.println("Result: " + productID);
        }

    }

    private static List<String> readProdFilesFromStdin() {
        List<String> prodFiles = new Vector<String>();
        BufferedReader br = null;

        br = new BufferedReader(new InputStreamReader(System.in));

        String line = null;

        try {
            while ((line = br.readLine()) != null) {
                prodFiles.add(line);
            }
        } catch (IOException e) {
            LOG.log(Level.WARNING, "Error reading prod file: line: [" + line
                    + "]: Message: " + e.getMessage(), e);
        }

        return prodFiles;
    }
}