embulk-input-gmail 0.1.0

data/gradle/wrapper/gradle-wrapper.properties

@@ -0,0 +1,5 @@
+distributionBase=GRADLE_USER_HOME
+distributionPath=wrapper/dists
+zipStoreBase=GRADLE_USER_HOME
+zipStorePath=wrapper/dists
+distributionUrl=https\://services.gradle.org/distributions/gradle-4.1-bin.zip

data/gradlew

@@ -0,0 +1,172 @@
+#!/usr/bin/env sh
+
+##############################################################################
+##
+##  Gradle start up script for UN*X
+##
+##############################################################################
+
+# Attempt to set APP_HOME
+# Resolve links: $0 may be a link
+PRG="$0"
+# Need this for relative symlinks.
+while [ -h "$PRG" ] ; do
+    ls=`ls -ld "$PRG"`
+    link=`expr "$ls" : '.*-> \(.*\)$'`
+    if expr "$link" : '/.*' > /dev/null; then
+        PRG="$link"
+    else
+        PRG=`dirname "$PRG"`"/$link"
+    fi
+done
+SAVED="`pwd`"
+cd "`dirname \"$PRG\"`/" >/dev/null
+APP_HOME="`pwd -P`"
+cd "$SAVED" >/dev/null
+
+APP_NAME="Gradle"
+APP_BASE_NAME=`basename "$0"`
+
+# Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
+DEFAULT_JVM_OPTS=""
+
+# Use the maximum available, or set MAX_FD != -1 to use that value.
+MAX_FD="maximum"
+
+warn () {
+    echo "$*"
+}
+
+die () {
+    echo
+    echo "$*"
+    echo
+    exit 1
+}
+
+# OS specific support (must be 'true' or 'false').
+cygwin=false
+msys=false
+darwin=false
+nonstop=false
+case "`uname`" in
+  CYGWIN* )
+    cygwin=true
+    ;;
+  Darwin* )
+    darwin=true
+    ;;
+  MINGW* )
+    msys=true
+    ;;
+  NONSTOP* )
+    nonstop=true
+    ;;
+esac
+
+CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar
+
+# Determine the Java command to use to start the JVM.
+if [ -n "$JAVA_HOME" ] ; then
+    if [ -x "$JAVA_HOME/jre/sh/java" ] ; then
+        # IBM's JDK on AIX uses strange locations for the executables
+        JAVACMD="$JAVA_HOME/jre/sh/java"
+    else
+        JAVACMD="$JAVA_HOME/bin/java"
+    fi
+    if [ ! -x "$JAVACMD" ] ; then
+        die "ERROR: JAVA_HOME is set to an invalid directory: $JAVA_HOME
+
+Please set the JAVA_HOME variable in your environment to match the
+location of your Java installation."
+    fi
+else
+    JAVACMD="java"
+    which java >/dev/null 2>&1 || die "ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
+
+Please set the JAVA_HOME variable in your environment to match the
+location of your Java installation."
+fi
+
+# Increase the maximum file descriptors if we can.
+if [ "$cygwin" = "false" -a "$darwin" = "false" -a "$nonstop" = "false" ] ; then
+    MAX_FD_LIMIT=`ulimit -H -n`
+    if [ $? -eq 0 ] ; then
+        if [ "$MAX_FD" = "maximum" -o "$MAX_FD" = "max" ] ; then
+            MAX_FD="$MAX_FD_LIMIT"
+        fi
+        ulimit -n $MAX_FD
+        if [ $? -ne 0 ] ; then
+            warn "Could not set maximum file descriptor limit: $MAX_FD"
+        fi
+    else
+        warn "Could not query maximum file descriptor limit: $MAX_FD_LIMIT"
+    fi
+fi
+
+# For Darwin, add options to specify how the application appears in the dock
+if $darwin; then
+    GRADLE_OPTS="$GRADLE_OPTS \"-Xdock:name=$APP_NAME\" \"-Xdock:icon=$APP_HOME/media/gradle.icns\""
+fi
+
+# For Cygwin, switch paths to Windows format before running java
+if $cygwin ; then
+    APP_HOME=`cygpath --path --mixed "$APP_HOME"`
+    CLASSPATH=`cygpath --path --mixed "$CLASSPATH"`
+    JAVACMD=`cygpath --unix "$JAVACMD"`
+
+    # We build the pattern for arguments to be converted via cygpath
+    ROOTDIRSRAW=`find -L / -maxdepth 1 -mindepth 1 -type d 2>/dev/null`
+    SEP=""
+    for dir in $ROOTDIRSRAW ; do
+        ROOTDIRS="$ROOTDIRS$SEP$dir"
+        SEP="|"
+    done
+    OURCYGPATTERN="(^($ROOTDIRS))"
+    # Add a user-defined pattern to the cygpath arguments
+    if [ "$GRADLE_CYGPATTERN" != "" ] ; then
+        OURCYGPATTERN="$OURCYGPATTERN|($GRADLE_CYGPATTERN)"
+    fi
+    # Now convert the arguments - kludge to limit ourselves to /bin/sh
+    i=0
+    for arg in "$@" ; do
+        CHECK=`echo "$arg"|egrep -c "$OURCYGPATTERN" -`
+        CHECK2=`echo "$arg"|egrep -c "^-"`                                 ### Determine if an option
+
+        if [ $CHECK -ne 0 ] && [ $CHECK2 -eq 0 ] ; then                    ### Added a condition
+            eval `echo args$i`=`cygpath --path --ignore --mixed "$arg"`
+        else
+            eval `echo args$i`="\"$arg\""
+        fi
+        i=$((i+1))
+    done
+    case $i in
+        (0) set -- ;;
+        (1) set -- "$args0" ;;
+        (2) set -- "$args0" "$args1" ;;
+        (3) set -- "$args0" "$args1" "$args2" ;;
+        (4) set -- "$args0" "$args1" "$args2" "$args3" ;;
+        (5) set -- "$args0" "$args1" "$args2" "$args3" "$args4" ;;
+        (6) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" ;;
+        (7) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" ;;
+        (8) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" ;;
+        (9) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" "$args8" ;;
+    esac
+fi
+
+# Escape application args
+save () {
+    for i do printf %s\\n "$i" | sed "s/'/'\\\\''/g;1s/^/'/;\$s/\$/' \\\\/" ; done
+    echo " "
+}
+APP_ARGS=$(save "$@")
+
+# Collect all arguments for the java command, following the shell quoting and substitution rules
+eval set -- $DEFAULT_JVM_OPTS $JAVA_OPTS $GRADLE_OPTS "\"-Dorg.gradle.appname=$APP_BASE_NAME\"" -classpath "\"$CLASSPATH\"" org.gradle.wrapper.GradleWrapperMain "$APP_ARGS"
+
+# by default we should be in the correct project dir, but when run from Finder on Mac, the cwd is wrong
+if [ "$(uname)" = "Darwin" ] && [ "$HOME" = "$PWD" ]; then
+  cd "$(dirname "$0")"
+fi
+
+exec "$JAVACMD" "$@"

data/gradlew.bat

@@ -0,0 +1,84 @@
+@if "%DEBUG%" == "" @echo off
+@rem ##########################################################################
+@rem
+@rem  Gradle startup script for Windows
+@rem
+@rem ##########################################################################
+
+@rem Set local scope for the variables with windows NT shell
+if "%OS%"=="Windows_NT" setlocal
+
+set DIRNAME=%~dp0
+if "%DIRNAME%" == "" set DIRNAME=.
+set APP_BASE_NAME=%~n0
+set APP_HOME=%DIRNAME%
+
+@rem Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
+set DEFAULT_JVM_OPTS=
+
+@rem Find java.exe
+if defined JAVA_HOME goto findJavaFromJavaHome
+
+set JAVA_EXE=java.exe
+%JAVA_EXE% -version >NUL 2>&1
+if "%ERRORLEVEL%" == "0" goto init
+
+echo.
+echo ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
+echo.
+echo Please set the JAVA_HOME variable in your environment to match the
+echo location of your Java installation.
+
+goto fail
+
+:findJavaFromJavaHome
+set JAVA_HOME=%JAVA_HOME:"=%
+set JAVA_EXE=%JAVA_HOME%/bin/java.exe
+
+if exist "%JAVA_EXE%" goto init
+
+echo.
+echo ERROR: JAVA_HOME is set to an invalid directory: %JAVA_HOME%
+echo.
+echo Please set the JAVA_HOME variable in your environment to match the
+echo location of your Java installation.
+
+goto fail
+
+:init
+@rem Get command-line arguments, handling Windows variants
+
+if not "%OS%" == "Windows_NT" goto win9xME_args
+
+:win9xME_args
+@rem Slurp the command line arguments.
+set CMD_LINE_ARGS=
+set _SKIP=2
+
+:win9xME_args_slurp
+if "x%~1" == "x" goto execute
+
+set CMD_LINE_ARGS=%*
+
+:execute
+@rem Setup the command line
+
+set CLASSPATH=%APP_HOME%\gradle\wrapper\gradle-wrapper.jar
+
+@rem Execute Gradle
+"%JAVA_EXE%" %DEFAULT_JVM_OPTS% %JAVA_OPTS% %GRADLE_OPTS% "-Dorg.gradle.appname=%APP_BASE_NAME%" -classpath "%CLASSPATH%" org.gradle.wrapper.GradleWrapperMain %CMD_LINE_ARGS%
+
+:end
+@rem End local scope for the variables with windows NT shell
+if "%ERRORLEVEL%"=="0" goto mainEnd
+
+:fail
+rem Set variable GRADLE_EXIT_CONSOLE if you need the _script_ return code instead of
+rem the _cmd.exe /c_ return code!
+if  not "" == "%GRADLE_EXIT_CONSOLE%" exit 1
+exit /b 1
+
+:mainEnd
+if "%OS%"=="Windows_NT" endlocal
+
+:omega

data/lib/embulk/input/gmail.rb

@@ -0,0 +1,3 @@
+Embulk::JavaPlugin.register_input(
+  "gmail", "org.embulk.input.gmail.GmailInputPlugin",
+  File.expand_path('../../../../classpath', __FILE__))

data/src/main/java/org/embulk/input/gmail/GmailInputPlugin.java

@@ -0,0 +1,280 @@
+package org.embulk.input.gmail;
+
+import java.io.IOException;
+import java.security.GeneralSecurityException;
+import java.util.List;
+
+import com.google.common.base.Optional;
+
+import org.embulk.config.Config;
+import org.embulk.config.ConfigDefault;
+import org.embulk.config.ConfigDiff;
+import org.embulk.config.ConfigInject;
+import org.embulk.config.ConfigSource;
+import org.embulk.config.Task;
+import org.embulk.config.TaskReport;
+import org.embulk.config.TaskSource;
+import org.embulk.spi.BufferAllocator;
+import org.embulk.spi.Column;
+import org.embulk.spi.ColumnVisitor;
+import org.embulk.spi.Exec;
+import org.embulk.spi.InputPlugin;
+import org.embulk.spi.PageBuilder;
+import org.embulk.spi.PageOutput;
+import org.embulk.spi.Schema;
+import org.embulk.spi.SchemaConfig;
+import org.embulk.spi.time.Timestamp;
+import org.embulk.spi.time.TimestampParseException;
+import org.embulk.spi.time.TimestampParser;
+import org.embulk.spi.util.Timestamps;
+
+import org.slf4j.Logger;
+
+public class GmailInputPlugin
+        implements InputPlugin
+{
+    public interface PluginTask
+            extends Task, TimestampParser.Task
+    {
+        // Gmail client secret json path. (required string)
+        @Config("client_secret")
+        public String getClientSecretPath();
+
+        // Gmail API tokens directory. (required string)
+        @Config("tokens_directory")
+        public String getTokensDirectory();
+
+        // Gmail search user. (optional, default: "me")
+        @Config("user")
+        @ConfigDefault("\"me\"")
+        public String getUser();
+
+        // Gmail search query. (optional, default: "")
+        @Config("query")
+        @ConfigDefault("\"\"")
+        public String getQuery();
+
+        // Gmail search query "after: xxx". (optional, default: null)
+        // Concat this config string, after "query" config string.
+        // You use if '-o' option.
+        @Config("after")
+        @ConfigDefault("null")
+        public Optional<String> getAfter();
+
+        // schema
+        @Config("columns")
+        public SchemaConfig getColumns();
+
+        // 謎。バッファアロケーターの実装を定義？
+        @ConfigInject
+        public BufferAllocator getBufferAllocator();
+    }
+
+    private Logger log = Exec.getLogger(GmailInputPlugin.class);
+
+    @Override
+    public ConfigDiff transaction(ConfigSource config,
+            InputPlugin.Control control)
+    {
+        PluginTask task = config.loadConfig(PluginTask.class);
+
+        Schema schema = task.getColumns().toSchema();
+        int taskCount = 1;  // number of run() method calls
+
+        return resume(task.dump(), schema, taskCount, control);
+    }
+
+    @Override
+    public ConfigDiff resume(TaskSource taskSource,
+            Schema schema, int taskCount,
+            InputPlugin.Control control)
+    {
+        control.run(taskSource, schema, taskCount);
+
+        // ConfigDiff を作成し、after に now をセット
+        List<TaskReport> taskReportList =
+                control.run(taskSource, schema, taskCount);
+        ConfigDiff configDiff = Exec.newConfigDiff();
+        for (TaskReport taskReport : taskReportList) {
+            final String label = "after";
+            final String after = taskReport.get(String.class, label, null);
+            if (after != null) {
+                configDiff.set(label, after);
+            }
+        }
+        return configDiff;
+    }
+
+    @Override
+    public void cleanup(TaskSource taskSource,
+            Schema schema, int taskCount,
+            List<TaskReport> successTaskReports)
+    {
+    }
+
+    @Override
+    public TaskReport run(TaskSource taskSource,
+            Schema schema, int taskIndex,
+            PageOutput output)
+    {
+        PluginTask task = taskSource.loadTask(PluginTask.class);
+
+        log.info("Try login use '{}' and '{}'.", task.getClientSecretPath(), task.getTokensDirectory());
+
+        // user 取得
+        String user = task.getUser();
+        log.info("Query user: '{}'", user);
+
+        // query 文字列組み立て
+        String query = task.getQuery();
+
+        String now = String.valueOf(System.currentTimeMillis() / 1000L);
+        Optional<String> after = task.getAfter();
+        for (String p : after.asSet()) {
+            query += " before:" + now;
+            query += " after:" + p;
+        }
+
+        log.info("Send query : '{}'", query);
+
+        // Visitor 作成
+        BufferAllocator allocator = task.getBufferAllocator();
+        PageBuilder pageBuilder = new PageBuilder(allocator, schema, output);
+
+        GmailWrapper.Result result;
+        try {
+            GmailWrapper gmail = new GmailWrapper(
+                    task.getClientSecretPath(),
+                    task.getTokensDirectory());
+            result = gmail.search(user, query);
+        } catch (IOException|GeneralSecurityException e) {
+            log.error("{}", e.getClass(), e);
+
+            TaskReport taskReport = Exec.newTaskReport();
+            return taskReport;
+        }
+
+
+        // success messages.
+        for (GmailWrapper.Message message : result.getSuccessMessages()) {
+            // Visitor 作成
+            ColumnVisitor visitor = new ColumnVisitorImpl(message, task, pageBuilder);
+            // スキーマ解析
+            schema.visitColumns(visitor);
+            // 編集したレコードを追加
+            pageBuilder.addRecord();
+        }
+        pageBuilder.finish();
+
+        // TODO: output failed message info to log.
+
+        TaskReport taskReport = Exec.newTaskReport();
+        taskReport.set("after", now);
+
+        return taskReport;
+    }
+
+    @Override
+    public ConfigDiff guess(ConfigSource config)
+    {
+        return Exec.newConfigDiff();
+    }
+
+    class ColumnVisitorImpl implements ColumnVisitor {
+        private final GmailWrapper.Message message;
+        private final TimestampParser[] timestampParsers;
+        private final PageBuilder pageBuilder;
+
+        ColumnVisitorImpl(GmailWrapper.Message message, PluginTask task, PageBuilder pageBuilder) {
+            this.message = message;
+            this.pageBuilder = pageBuilder;
+
+            this.timestampParsers = Timestamps.newTimestampColumnParsers(
+                    task, task.getColumns());
+        }
+
+        @Override
+        public void booleanColumn(Column column) {
+            String value = message.getHeaders().get(column.getName());
+            if (value == null) {
+                pageBuilder.setNull(column);
+            } else {
+                pageBuilder.setBoolean(column, Boolean.parseBoolean(value));
+            }
+        }
+
+        @Override
+        public void longColumn(Column column) {
+            String value = message.getHeaders().get(column.getName());
+            if (value == null) {
+                pageBuilder.setNull(column);
+            } else {
+                try {
+                    pageBuilder.setLong(column, Long.parseLong(value));
+                } catch (NumberFormatException e) {
+                    log.error("NumberFormatError: Header: `{}:{}`", column.getName(), value);
+                    log.error("{}", e);
+                    pageBuilder.setNull(column);
+                }
+            }
+        }
+
+        @Override
+        public void doubleColumn(Column column) {
+            String value = message.getHeaders().get(column.getName());
+            if (value == null) {
+                pageBuilder.setNull(column);
+            } else {
+                try {
+                    pageBuilder.setDouble(column, Double.parseDouble(value));
+                } catch (NumberFormatException e) {
+                    log.error("NumberFormatError: Header: `{}:{}`", column.getName(), value);
+                    log.error("{}", e);
+                    pageBuilder.setNull(column);
+                }
+            }
+        }
+
+        @Override
+        public void stringColumn(Column column) {
+            if (column.getName().equals("Body")) {
+                String value = message.getBody().orElse(null);
+                if (value == null) {
+                    pageBuilder.setNull(column);
+                } else {
+                    pageBuilder.setString(column, value);
+                }
+            } else {
+                String value = message.getHeaders().get(column.getName());
+                if (value == null) {
+                    pageBuilder.setNull(column);
+                } else {
+                    pageBuilder.setString(column, value);
+                }
+            }
+        }
+
+        @Override
+        public void jsonColumn(Column column) {
+            throw new UnsupportedOperationException("This plugin doesn't support json type. Please try to upgrade version of the plugin using 'embulk gem update' command. If the latest version still doesn't support json type, please contact plugin developers, or change configuration of input plugin not to use json type.");
+        }
+
+        @Override
+        public void timestampColumn(Column column) {
+            String value = message.getHeaders().get(column.getName());
+            if (value == null) {
+                pageBuilder.setNull(column);
+            } else {
+                try {
+                    Timestamp timestamp = timestampParsers[column.getIndex()]
+                            .parse(value);
+                    pageBuilder.setTimestamp(column, timestamp);
+                } catch (TimestampParseException e) {
+                    log.error("TimestampParseError: Header: `{}:{}`", column.getName(), value);
+                    log.error("{}", e);
+                    pageBuilder.setNull(column);
+                }
+            }
+        }
+    }
+}