navi_email_sync 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: eb23348504b8e7954bdf7093b22ed565df930c368131827938098ab926aec925
+  data.tar.gz: 9232cb2fea5c1dbcb207ed450a827f8864f5094824d578ebee179a3eba94df32
+SHA512:
+  metadata.gz: 61d2b62886f6c3873721f5abe46ba3ba0733cf72cb6c26a656cf4b2a4cec6f8bad3c4dda5a47c6624de0336d4405c4876e5190042362e8eb4f6e5a11c5e9c0c0
+  data.tar.gz: 247f7c139a5d5beaec3b7e1a070f0c8bf115ba6a8a8e74d760b0301a571cd16c7acfac0d09d5ef34437a9149e7e03811b4794a1a01ea77be02ac639281ee81fe

data/.gitignore

@@ -0,0 +1,11 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+.idea/
+*.gem

data/.rspec

@@ -0,0 +1,2 @@
+--format documentation
+--color

data/.travis.yml

@@ -0,0 +1,4 @@
+language: ruby
+rvm:
+  - 2.3.0
+before_install: gem install bundler -v 1.11.2

data/CODE_OF_CONDUCT.md

@@ -0,0 +1,49 @@
+# Contributor Code of Conduct
+
+As contributors and maintainers of this project, and in the interest of
+fostering an open and welcoming community, we pledge to respect all people who
+contribute through reporting issues, posting feature requests, updating
+documentation, submitting pull requests or patches, and other activities.
+
+We are committed to making participation in this project a harassment-free
+experience for everyone, regardless of level of experience, gender, gender
+identity and expression, sexual orientation, disability, personal appearance,
+body size, race, ethnicity, age, religion, or nationality.
+
+Examples of unacceptable behavior by participants include:
+
+* The use of sexualized language or imagery
+* Personal attacks
+* Trolling or insulting/derogatory comments
+* Public or private harassment
+* Publishing other's private information, such as physical or electronic
+  addresses, without explicit permission
+* Other unethical or unprofessional conduct
+
+Project maintainers have the right and responsibility to remove, edit, or
+reject comments, commits, code, wiki edits, issues, and other contributions
+that are not aligned to this Code of Conduct, or to ban temporarily or
+permanently any contributor for other behaviors that they deem inappropriate,
+threatening, offensive, or harmful.
+
+By adopting this Code of Conduct, project maintainers commit themselves to
+fairly and consistently applying these principles to every aspect of managing
+this project. Project maintainers who do not follow or enforce the Code of
+Conduct may be permanently removed from the project team.
+
+This code of conduct applies both within project spaces and in public spaces
+when an individual is representing the project or its community.
+
+Instances of abusive, harassing, or otherwise unacceptable behavior may be
+reported by contacting a project maintainer at surya@whitehatengineering.com. All
+complaints will be reviewed and investigated and will result in a response that
+is deemed necessary and appropriate to the circumstances. Maintainers are
+obligated to maintain confidentiality with regard to the reporter of an
+incident.
+
+This Code of Conduct is adapted from the [Contributor Covenant][homepage],
+version 1.3.0, available at
+[http://contributor-covenant.org/version/1/3/0/][version]
+
+[homepage]: http://contributor-covenant.org
+[version]: http://contributor-covenant.org/version/1/3/0/

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in navi_email_sync.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2017 Surya
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.md

@@ -0,0 +1,46 @@
+# NaviEmailSync
+
+Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/navi_client`. To experiment with that code, run `bin/console` for an interactive prompt.
+
+TODO: Delete this and the text above, and describe your gem
+
+## Generate Documentation
+```
+$ rdoc
+```
+Browse html file inside the ```doc``` folder
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'navi_email_sync'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install navi_email_sync
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/navi_client. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [Contributor Covenant](http://contributor-covenant.org) code of conduct.
+
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](http://opensource.org/licenses/MIT).

data/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/bin/console

@@ -0,0 +1,12 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/config/navi_client.sample.yml

@@ -0,0 +1,9 @@
+# Just copy paste the following format in application.yml and configure accordingly
+
+development:
+  api_url: http://localhost:3020/
+  parser_url: http://localhost:9090
+
+production:
+  parser_url: http://34.214.134.104:9090
+

data/lib/client.rb

@@ -0,0 +1,259 @@
+require "net/imap"
+require "mail"
+require "time"
+
+require 'base64'
+require 'fileutils'
+require 'yaml'
+
+require "logger"
+
+require "httparty"
+require "http_service/naviai"
+require 'openssl'
+
+##
+# This module provides the common functionality that will be needed for local and cloud module.
+# @client_type defines if the gem is being used for local-lockbox or cloud-version.
+#
+module Client
+
+  attr_accessor :downloaded
+
+  attr_accessor :message_ids
+
+  attr_accessor :total_emails
+
+  @logglyTag
+
+  def logger
+    @logger
+  end
+
+  def errors
+    @errors
+  end
+
+  #
+  # imap_connection
+  #
+  # connect the app with imap server
+  #
+  def imap_connection(server, username, password, exitOnFail = true)
+    # connect to IMAP server
+    @imap = Net::IMAP.new server, ssl: true, certs: nil, verify: false
+
+    Net::IMAP.debug = @net_imap_debug
+
+    # http://ruby-doc.org/stdlib-2.1.5/libdoc/net/imap/rdoc/Net/IMAP.html#method-i-capability
+    capabilities = @imap.capability
+
+    @logger.debug("imap capabilities: #{capabilities.join(',')}") if @debug
+
+    if @client_type == 'local'
+      unless capabilities.include? "IDLE" || @debug
+        @logger.info "'IDLE' IMAP capability not available in server: #{server}"
+        @imap.disconnect
+        exit if exitOnFail
+      end
+    end
+
+    begin
+      # login
+      @imap.login username, password
+      @errors = nil
+    rescue Net::IMAP::NoResponseError => e
+      # mostly due to credentials error
+      @errors = {exception: e}
+    rescue Net::IMAP::BadResponseError => e
+      @errors = {exception: e}
+    rescue
+      @errors = {exception: e}
+    end
+
+    # return IMAP connection handler
+    @imap
+  end
+
+  #
+  # retrieve_emails
+  #
+  # retrieve any mail from a folder, followin specified serach condition
+  # for any mail retrieved call a specified block
+  #
+  def fetch_emails(&process_email_block)
+
+    @message_ids.each_with_index do |message_id, i|
+      begin
+        # fetch all the email contents
+        data = @imap.uid_fetch(message_id, "RFC822")
+        data.each do |d|
+          msg = d.attr['RFC822']
+          # instantiate a Mail object to avoid further IMAP parameters nightmares
+          mail = Mail.read_from_string msg
+
+          # call the block with mail object as param
+          process_email_block.call mail, i, i == @message_ids.length-1, message_id
+
+          # mark as read
+          if @mark_as_read
+            @imap.store(message_id, "+FLAGS", [:Seen])
+          end
+        end
+      rescue => e
+        unless logger.nil?
+          logger.info "Issue processing email for uuid ##{message_id}, #{e.message}"
+        end
+
+        logToLoggly({event:"EMAIL_SYNC_FAILED", env: @env, storage: @client_type, email: @current_user_email, uuid: message_id, error: e.message})
+        raise e
+      end
+    end
+  end
+
+
+  def init_messages label, search_condition = ['ALL']
+
+    # examine will read email and sets flags unread
+    # https://stackoverflow.com/questions/16516464/read-gmail-xoauth-mails-without-marking-it-read
+    @imap.examine label
+
+    @message_ids = @imap.uid_search(search_condition)
+
+    if search_condition.include?('ALL')
+      @all_emails = @message_ids if search_condition.include? 'ALL'
+    else
+      @all_emails = @all_emails | @message_ids
+    end
+
+    @total_emails = @message_ids.length if search_condition.include?('ALL')
+    downloaded_email = getMessageUUIds("#{@download_path}meta/")
+    downloaded_email = downloaded_email.collect{|i| i.to_i}
+
+    downloaded_email = @all_emails & downloaded_email
+
+    @downloaded = downloaded_email.length
+    # @downloaded = @downloaded - 2 if @downloaded > 1
+
+    @downloaded = [@downloaded, @total_emails].min
+
+    @message_ids = @message_ids - downloaded_email
+    @message_ids = @message_ids&.sort {|msg_a, msg_b| msg_b <=> msg_a}
+
+    if @debug
+      if @message_ids.empty?
+        @logger.info "No new emails found..."
+      elsif downloaded_email.any?
+        @logger.info "Found emails saved in your #{@client_type}. Downloading only #{@message_ids.count} new emails..."
+      else
+        @logger.info "Downloading #{@message_ids.count} emails."
+      end
+    end
+
+  end
+
+  ##
+  # Process each email downloaded from imap-server and creates meta file that will be sent over to the navi-ai service.
+  # Meta will content information like: ['from', 'to', 'cc', ..., 'body_url']. This information is then used by navi-ai to parse the body content.
+  #
+  def process_email(mail, uid, stamp = nil)
+    meta = Hash.new
+    custom_uid = (Time.now.to_f * 1000).to_s + "_" + mail.__id__.to_s
+
+    meta["from"] = mail[:from].to_s
+    meta["to"] = mail[:to].to_s
+    meta["cc"] = mail[:cc].to_s
+    meta["subject"] = mail.subject
+    meta["date"] = mail.date.to_s
+
+    if mail.multipart?
+      for i in 0...mail.parts.length
+        m = download(mail.parts[i], custom_uid)
+        meta.merge!(m) unless m.nil?
+      end
+    else
+      m = download(mail, custom_uid)
+      meta.merge!(m) unless m.nil?
+    end
+
+    if stamp.nil?
+      save(meta, "meta/#{uid.to_s + '_' + custom_uid}")
+    else
+      save(meta, "meta/#{uid.to_s + '_' + custom_uid}", stamp)
+    end
+
+  end
+
+  def encrypt(data)
+    Base64.encode64(data)
+  end
+
+  def depcrecated_encrypt(data)
+    cipher = OpenSSL::Cipher::AES.new(256, :CFB)
+    cipher.encrypt
+
+    yml_config = config
+
+    key_iv_exists = (yml_config['key'] && yml_config['iv']) ? (!yml_config['key'].empty? && !yml_config['iv'].empty? ) : false
+
+    if key_iv_exists
+      # this condition must be true for cloud version
+      cipher.key = Base64.decode64(File.read(yml_config['key']))
+      cipher.iv = Base64.decode64(File.read(yml_config['iv']))
+    else
+      cipher.key = key = cipher.random_key
+      cipher.iv = iv = cipher.random_iv
+
+      key_path, iv_path = save_aes_key_iv(key, iv)
+
+      yml_config['key'] = key_path
+      yml_config['iv'] = iv_path
+
+      update_config(yml_config)
+    end
+
+    encrypted = cipher.update(data)
+    Base64.encode64(encrypted)
+  end
+
+  def time_now
+    Time.now.utc.iso8601(3)
+  end
+
+  ##
+  # If the gem is being used for local-lockbox mode, after fetching all emails, the process will go idle mode. If user send the interrupt command, it will call shutdown to disconnect the connection with imap server
+  #
+  def shutdown(imap)
+    imap.idle_done
+    imap.logout unless imap.disconnected?
+    imap.disconnect
+
+    @logger.info "#{$0} has ended (crowd applauds)"
+    exit 0
+  end
+
+  def logToLoggly(messageBody)
+
+    if(@logglyTag)
+      begin
+        HTTParty.post("http://logs-01.loggly.com/bulk/0d67f93b-6568-4b00-9eca-97d0ea0bd5a1/tag/#{@logglyTag}/",
+                      body: messageBody.to_json,
+                      headers: { 'Content-Type' => 'application/json' } )
+      rescue
+        true
+      end
+
+    else
+      if @debug && !@logger.nil?
+        @logger.info "Logging to Loggly disabled"
+        @logger.info messageBody
+      end
+    end
+
+  end
+
+  def setupLoggly(tag)
+    @logglyTag = tag
+  end
+
+end

data/lib/cloud/navi_cloud_client.rb

@@ -0,0 +1,140 @@
+require Gem::Specification.find_by_name("navi_email_sync").gem_dir+"/lib/client"
+require 'concurrent'
+
+module NaviEmailSync
+
+  ##
+  # This class represents the client for cloud version.
+  # In cloud version, all the content will be saved and also being used from the s3
+  #
+  class Cloud
+    include Concurrent::Async
+    include Client
+
+    attr_reader :id
+
+    def initialize(sso_web_url = ENV['api_url'], env = Rails.env)
+      super()
+
+      # client-id used to track the process
+      @id = SecureRandom.uuid
+
+      # flag to print Ruby library debug info (very detailed)
+      @net_imap_debug = false
+
+      # flag to mark email as read after gets downloaded.
+      @mark_as_read = false
+      # flag to turn on/off debug mode.
+      @debug = false
+
+      @logger = nil
+
+      # sso_web (authentication) config.
+      @sso_web_url = sso_web_url
+      # authentication token received from sso_web used to authenticate the request to database_api
+      @token = nil
+
+      # client_type
+      @client_type = "cloud"
+
+      @download_path = config[:s3_download_folder] + '/'
+
+      # set email_address of current_user for s3 folder name
+      @current_user_email = nil
+      @env = env
+    end
+
+    def override_logger(logger)
+      @logger = logger
+    end
+
+    #
+    # login
+    #
+    # login to the navi-cloud and get the authentication token
+    #
+    def login(session_token)
+      @token = session_token
+    end
+
+    def set_current_user_email(email)
+      @current_user_email = email
+      @download_path = config[:s3_download_folder] + '/' + email + "/"
+    end
+
+    ##
+    # send bulk request to navi-ai service with list of input files downloaded from imap server.
+    #
+    def send_request(in_filenames = [], is_last: false)
+      unless in_filenames.blank?
+        download_path = config['s3_download_folder'] + "/" + @current_user_email
+        filepath = download_path + "/inputs/" + (Time.now.to_f * 1000).to_s
+        filename = upload_to_s3(filepath, in_filenames.join("\n"))
+
+        HTTPService::NaviAI.start(filepath: filename, client_id: @id, client_type: @client_type, token: @token, email: @current_user_email, is_last: is_last)
+      end
+    end
+
+    ##
+    # Downloads the email content from imap-server and save it to the download_path
+    #
+    def download(message, custom_uid)
+      download_path = config[:s3_download_folder] + "/" + @current_user_email
+      if ['text/plain', 'text/html'].include? message.mime_type
+
+        h = Hash.new
+        out_file = download_path + "/" + message.mime_type + "/"+custom_uid
+
+        s3_filepath = upload_to_s3(out_file, encrypt(message.decoded))
+        key = message.mime_type.split("/").join("_")
+
+        h[key] = s3_filepath
+        return h
+      end
+    end
+
+    ##
+    # save data to download_path with file named by filename params.
+    # Input is hashmap, and it save the hashmap as yml format.
+    #
+    def save(data={}, filename)
+      download_path = config[:s3_download_folder] + "/" + @current_user_email
+      filepath = download_path + "/" + filename + ".yml"
+
+      return upload_to_s3(filepath, data.to_yaml)
+    end
+
+    ##
+    # Helper function to upload the content to the s3
+    def upload_to_s3(file_path, content)
+      credentials = Aws::Credentials.new(config[:aws_key], config[:aws_secret])
+      s3 = Aws::S3::Client.new(credentials: credentials, region: config[:aws_region])
+      obj = s3.put_object({
+                            body: content,
+                            bucket: config[:s3_bucket],
+                            key: file_path
+                          })
+      return file_path if obj.successful?
+      return ""
+    end
+
+    def s3_resource_files(bucket_name, prefix)
+      files = []
+      credentials = Aws::Credentials.new(config[:aws_key], config[:aws_secret])
+      s3 = Aws::S3::Resource.new(credentials: credentials, region: config[:aws_region])
+      s3.bucket(bucket_name).objects(prefix: prefix).each do |obj|
+        files << obj.key
+      end
+      return files
+    end
+
+    def getMessageUUIds(prefix)
+      files = s3_resource_files(config[:s3_bucket], prefix)
+      files.empty? ? [] : files.map { |i| i.empty? ? 0 : i.split('/').last.split("_").first.to_i }
+    end
+
+    def config
+      YAML.load_file(Rails.root.join("config/navi_client.yml")).with_indifferent_access
+    end
+  end
+end

data/lib/gmail_client.rb

@@ -0,0 +1,224 @@
+require "mail"
+require "time"
+
+require 'base64'
+require 'fileutils'
+require 'yaml'
+
+require "logger"
+
+require "httparty"
+require "http_service/naviai"
+require 'openssl'
+require 'google/apis/gmail_v1'
+require 'googleauth'
+require 'googleauth/stores/file_token_store'
+
+##
+# This module provides the common functionality that will be needed for local and cloud module.
+# @client_type defines if the gem is being used for local-lockbox or cloud-version.
+#
+module GmailClient
+
+  OOB_URI = 'urn:ietf:wg:oauth:2.0:oob'.freeze
+  APPLICATION_NAME = 'Gmail Api Sync'.freeze
+
+  attr_accessor :downloaded
+
+  attr_accessor :message_ids
+
+  attr_accessor :total_emails
+
+  @logglyTag
+
+  def logger
+    @logger
+  end
+
+  def errors
+    @errors
+  end
+
+  def init client_path, user_cred_path, user_id
+
+    client_id = Google::Auth::ClientId.from_file(client_path)
+    token_store = Google::Auth::Stores::FileTokenStore.new(file: user_cred_path)
+    @authorizer = Google::Auth::UserAuthorizer.new(client_id, Google::Apis::GmailV1::AUTH_GMAIL_READONLY, token_store)
+
+    credentials = @authorizer.get_credentials(user_id)
+
+    unless credentials.nil?
+      init_apis credentials
+    end
+    credentials
+  end
+
+  def authorize code, user_id
+    credentials = @authorizer.get_and_store_credentials_from_code(
+        user_id: user_id, code: code, base_url: OOB_URI
+    )
+    unless credentials.nil?
+      init_apis credentials
+    end
+    credentials
+  end
+
+  def init_apis cred
+    @service = Google::Apis::GmailV1::GmailService.new
+    @service.client_options.application_name = APPLICATION_NAME
+    @service.authorization = cred
+  end
+
+  def authorization_url
+    @authorizer.get_authorization_url(base_url: OOB_URI)
+  end
+
+  #
+  # retrieve_emails
+  #
+  # retrieve any mail from a folder, followin specified serach condition
+  # for any mail retrieved call a specified block
+  #
+  def fetch_emails(&process_email_block)
+
+    @message_ids.each_with_index do |message_id, i|
+      begin
+        # fetch all the email contents
+        # data = imap.uid_fetch(message_id, "RFC822")
+        result = @service.get_user_message('me', message_id)
+
+
+        process_email_block.call result, i, i == @message_ids.length-1, message_id
+
+      rescue => e
+        unless logger.nil?
+          logger.info "Issue processing email for uuid ##{message_id}, #{e.message}"
+        end
+
+        logToLoggly({event:"EMAIL_SYNC_FAILED", env: @env, storage: @client_type, email: @current_user_email, uuid: message_id, error: e.message})
+        raise e
+      end
+    end
+
+  end
+
+
+  ##
+  # Process each email downloaded from imap-server and creates meta file that will be sent over to the navi-ai service.
+  # Meta will content information like: ['from', 'to', 'cc', ..., 'body_url']. This information is then used by navi-ai to parse the body content.
+  #
+  def process_email(mail, uid, stamp = nil)
+    meta = Hash.new
+    custom_uid = (Time.now.to_f * 1000).to_s + "_" + uid
+
+    payload = mail.payload
+
+    headers = payload.headers
+    date = headers.any? { |h| h.name == 'Date' } ? headers.find { |h| h.name == 'Date' }.value : ''
+    from = headers.any? { |h| h.name == 'From' } ? headers.find { |h| h.name == 'From' }.value : ''
+    to = headers.any? { |h| h.name == 'To' } ? headers.find { |h| h.name == 'To' }.value : ''
+    subject = headers.any? { |h| h.name == 'Subject' } ? headers.find { |h| h.name == 'Subject' }.value : ''
+    cc = headers.any? { |h| h.name == 'Cc' } ? headers.find { |h| h.name == 'Cc' }.value : ''
+
+    meta["from"] = from
+    meta["to"] = to
+    meta["cc"] = cc
+    meta["subject"] = subject
+    meta["date"] = date
+
+    if payload.parts&.any?
+      payload.parts.map { |part|
+        m = download({"decoded" => part.body.data, "mime_type" => part.mime_type}, custom_uid)
+        meta.merge!(m) unless m.nil?
+      }
+    else
+      m = download({"decoded" => payload.body.data, "mime_type" => "text/html"}, custom_uid)
+      meta.merge!(m) unless m.nil?
+    end
+
+    if stamp.nil?
+      save(meta, "meta/#{uid.to_s + '_' + custom_uid}")
+    else
+      save(meta, "meta/#{uid.to_s + '_' + custom_uid}", stamp)
+    end
+
+  end
+
+  def all_messages label
+    messages = []
+    next_page = nil
+
+    begin
+      result = @service.list_user_messages('me', max_results: 1000, page_token: next_page, label_ids: [label])
+      messages += result.messages
+      # break if messages.size >= 1
+      next_page = result.next_page_token
+    end while next_page
+
+    messages
+  end
+
+  def get_user_info user_id
+    @service.get_user_profile user_id
+  end
+
+  def all_message_ids label
+    mids = []
+    messages = all_messages label
+    messages.each { |m| mids << m.id }
+    mids
+  end
+
+  def init_messages label
+    @message_ids = all_message_ids label
+    @total_emails = @message_ids.length
+    downloaded_email = getMessageUUIds("#{@download_path}meta/")
+
+    downloaded_email = @message_ids & downloaded_email
+
+    @downloaded = downloaded_email.length
+
+    @downloaded = [@downloaded, @total_emails].min
+
+    @message_ids = @message_ids - downloaded_email
+
+    if @debug
+      if @message_ids.empty?
+        @logger.info "No new emails found..."
+      elsif downloaded_email.any?
+        @logger.info "Found emails saved in your #{@client_type}. Downloading only #{@message_ids.count} new emails..."
+      else
+        @logger.info "Downloading #{@message_ids.count} emails."
+      end
+    end
+
+  end
+
+  def logToLoggly(messageBody)
+
+    if(@logglyTag)
+      begin
+        HTTParty.post("http://logs-01.loggly.com/bulk/0d67f93b-6568-4b00-9eca-97d0ea0bd5a1/tag/#{@logglyTag}/",
+                      body: messageBody.to_json,
+                      headers: { 'Content-Type' => 'application/json' } )
+      rescue
+        true
+      end
+
+    else
+      if @debug && !@logger.nil?
+        @logger.info "Logging to Loggly disabled"
+        @logger.info messageBody
+      end
+    end
+
+  end
+
+  def setupLoggly(tag)
+    @logglyTag = tag
+  end
+
+  def encrypt(data)
+    Base64.encode64(data)
+  end
+end

data/lib/http_service/naviai.rb

@@ -0,0 +1,15 @@
+require "httparty"
+
+module HTTPService
+  class NaviAI
+
+    def self.start(filepath:, client_id: nil, client_type:, token:, email:, is_last: false)
+      go_url = "#{ENV['parser_url']}/v2/metas"
+      HTTParty.post(go_url, body: { client_id: client_id, client_type: client_type, list_meta_path: filepath, token: token, email: email, is_last: is_last }.to_json, timeout: 400)
+    end
+
+    def self.generate_csv(url, token)
+      HTTParty.post(url, headers: {"Authorization": token})
+    end
+  end
+end

data/lib/local/navi_local_client.rb

@@ -0,0 +1,246 @@
+require_relative '../client'
+require_relative '../gmail_client'
+
+module NaviEmailSync
+
+  ##
+  # This class represents the client for local lockbox mode.
+  # In local-lockbox mode, all the content will be saved and also being used from the local file system
+  #
+  class Local
+
+    CONFIG_PATH = File.join(ENV['HOME'], '/.navi/config.yml')
+
+    def initialize(method, client_type = "local", debug = true)
+      # flag to print Ruby library debug info (very detailed)
+      @net_imap_debug = false
+
+      # flag to mark email as read after gets downloaded.
+      @mark_as_read = false
+
+      # flag to turn on/off debug mode.
+      @debug = debug
+
+      # override the log file
+      mkdir_if_not_exist(config['client_log_file'])
+      @logger = Logger.new(config['client_log_file'])
+
+      # sso_web (authentication) config.
+      @sso_web_url = ENV['api_url']
+      # authentication token received from sso_web used to authenticate the request to database_api
+      @token = nil
+
+      @download_path = config['download_path'] + config['identifier'] + "/#{method}/"
+      @current_user_email = config['username']
+
+      # client_type
+      @client_type = client_type
+      @gen_csv_url = "#{ENV['api_url']}/v2/generate_csv_input"
+      @env = ENV['env']?ENV['env']:Rails.env
+    end
+
+    #
+    # login
+    #
+    # login to the navi-cloud and get the authentication token
+    #
+    def login
+      url = "#{@sso_web_url}/oauth/token"
+      provider_url = url
+      token = HTTParty.post(provider_url,
+                             body: {
+                               client_id: config["uid"], # get from sso_web application
+                               client_secret: config["secret_key"],
+                               grant_type: "client_credentials"
+                             }
+                            )['access_token']
+      @token = "Token: #{token}##{config['username']}"
+    end
+
+    def override_logger(logger)
+      @logger = logger
+    end
+
+    def set_token(token)
+      @token = token
+    end
+
+    ##
+    # Downloads the email content from imap-server and save it to the download_path
+    #
+    def download(message, custom_uid)
+
+      mime_type = message["mime_type"].nil? ? message.mime_type : message["mime_type"]
+
+      if ['text/plain', 'text/html', 'text/alternative'].include? mime_type
+
+        h = Hash.new
+        out_file = @download_path + mime_type + "/"+custom_uid
+        mkdir_if_not_exist(out_file)
+
+        File.open(out_file, 'w') { |file| file.write(encrypt(message["decoded"].nil? ? message.decoded : message["decoded"])) }
+        key = mime_type.split("/").join("_")
+
+        h[key] = out_file
+        return h
+      end
+    end
+
+    ##
+    # save data to download_path with file named by filename params.
+    # Input is hashmap, and it save the hashmap as yml format.
+    #
+    def save(data={}, filename, stamp)
+      temp_path = @download_path + "inputs/temp-#{stamp}"
+      filepath = @download_path + filename + ".yml"
+
+      mkdir_if_not_exist(filepath)
+      mkdir_if_not_exist(temp_path)
+      File.open(filepath, 'w') do |f|
+        f.puts(data.to_yaml)
+      end
+
+      File.open(temp_path, 'a') do |f|
+        f.puts(filepath)
+      end
+
+      return filepath
+    end
+
+    ##
+    # send bulk request to navi-ai service with list of input files downloaded from imap server.
+    #
+    def send_request(in_filenames = [], stamp, is_last: false)
+      unless in_filenames.empty?
+        filename = @download_path + "inputs/#{stamp}"
+
+        mkdir_if_not_exist(filename)
+
+        File.open(filename, 'w') do |f|
+          in_filenames.each { |element| f.puts(element) }
+        end
+
+        response = HTTPService::NaviAI.start(filepath: filename, client_type: @client_type, token: @token, email: config['username'], is_last: is_last)
+        return response
+      end
+    end
+
+    def mkdir_if_not_exist(filepath)
+      dirname = File.dirname(filepath)
+      unless File.directory?(dirname)
+        FileUtils.mkdir_p(dirname)
+      end
+    end
+
+    #
+    # idle_loop
+    #
+    # check for any further mail with "real-time" responsiveness.
+    # retrieve any mail from a folder, following specified search condition
+    # for any mail retrieved call a specified block
+    #
+    def idle_loop(search_condition, folder, server, username, password, callback = nil, navi_control = nil)
+
+      @logger.info "\nwaiting new mails (IDLE loop)..."
+
+      loop do
+        begin
+          @imap.select folder
+          @imap.idle do |resp|
+
+            # You'll get all the things from the server. For new emails (EXISTS)
+            if resp.kind_of?(Net::IMAP::UntaggedResponse) and resp.name == "EXISTS"
+
+              # @logger.debug resp.inspect if @debug
+              # Got something. Send DONE. This breaks you out of the blocking call
+              @logger.debug "New mail received" if @debug
+              @imap.idle_done
+            end
+          end
+
+          # We're out, which means there are some emails ready for us.
+          # Go do a search for UNSEEN and fetch them.
+          filenames = []
+          stamp = (Time.now.to_f * 1000).to_s
+          init_messages folder, search_condition
+
+          fetch_emails() { |mail, i, isLast, id|
+            callback.call(mail, i, id, false) unless callback.nil?
+            filenames << process_email(mail, id, stamp)
+            callback.call(mail, i, id, true) unless callback.nil?
+          }
+
+          @logger.info "Sending Request for #{filenames.size} emails to Navi AI."
+          navi_control.call(true) unless navi_control.nil?
+          self.send_request(filenames, stamp, is_last: true)
+
+          # HTTPService::NaviAI.generate_csv("#{@gen_csv_url}?email=#{username}", @token)
+
+          @logger.debug "Process Completed." if @debug
+          navi_control.call(false) unless navi_control.nil?
+
+        rescue SignalException => e
+          # http://stackoverflow.com/questions/2089421/capturing-ctrl-c-in-ruby
+          @logger.info "Signal received at #{time_now}: #{e.class}. #{e.message}"
+          shutdown @imap
+
+        rescue Net::IMAP::Error => e
+          @logger.error "Net::IMAP::Error at #{time_now}: #{e.class}. #{e.message}"
+          @imap = imap_connection(server, username, password) #if e.message == 'connection closed'
+          @logger.info "reconnected to server: #{server}"
+
+        # rescue Exception => e
+        #   @logger.error "Something went wrong at #{time_now}: #{e.class}. #{e.message}"
+        #
+        #   @imap = imap_connection(server, username, password)
+        #   @logger.info "reconnected to server: #{server}"
+        end
+      end
+    end
+
+    def initate_csv_generation(username)
+      HTTPService::NaviAI.generate_csv("#{@gen_csv_url}?email=#{username}", @token)
+    end
+
+    def config
+      YAML.load_file(CONFIG_PATH)
+    end
+
+    def save_aes_key_iv(key, iv)
+      aes_key_path = File.join(ENV['HOME'] + '/.navi/.aes.key')
+      iv_path = File.join(ENV['HOME'] + '/.navi/.aes.iv')
+
+      File.open(aes_key_path, "w") { |f| f.write(Base64.encode64(key)) }
+      File.open(iv_path, "w") { |f| f.write(Base64.encode64(iv)) }
+      return aes_key_path, iv_path
+    end
+
+    def getMessageUUIds(path)
+      File.directory?(path) ? (Dir.entries path).map { |i| i.split("_").first } : []
+    end
+
+    def update_config(data)
+      File.write(CONFIG_PATH, YAML.dump(data))
+    end
+
+    def parse_prev_failed_emails
+      changes = false
+      Dir[@download_path + 'inputs/temp*'].each do |file_name|
+        if File.file? file_name
+          HTTPService::NaviAI.start(filepath: file_name, client_type: @client_type, token: @token, email: @current_user_email )
+          changes = true
+        end
+      end
+      return changes
+    end
+
+  end
+
+  class GmailApi < Local
+    include GmailClient
+  end
+
+  class ImapApi < Local
+    include Client
+  end
+end

data/lib/navi_client/version.rb

@@ -0,0 +1,3 @@
+module NaviEmailSync
+  VERSION = "0.0.1"
+end

data/lib/navi_email_sync.rb

@@ -0,0 +1,9 @@
+require "local/navi_local_client"
+require "cloud/navi_cloud_client"
+
+##
+# This module represents the client that helps connection between navi and the imap server.
+#
+module NaviEmailSync
+
+end

data/navi_email_sync.gemspec

@@ -0,0 +1,37 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'navi_client/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "navi_email_sync"
+  spec.version       = NaviEmailSync::VERSION
+  spec.authors       = ["Sanjib"]
+  spec.email         = ["sanjib@whitehatengineering.com"]
+
+  spec.summary       = %q{ Write a short summary, because Rubygems requires one.}
+  spec.description   = %q{ Write a longer description or delete this line.}
+  spec.homepage      = "http://navihq.com"
+  spec.license       = "MIT"
+
+  # Prevent pushing this gem to RubyGems.org by setting 'allowed_push_host', or
+  # delete this section to allow pushing this gem to any host.
+  if spec.respond_to?(:metadata)
+    # spec.metadata['allowed_push_host'] = "Set to 'https://rubygems.org'"
+  else
+    raise "RubyGems 2.0 or newer is required to protect against public gem pushes."
+  end
+
+  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.11"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "rspec", "~> 3.0"
+  spec.add_runtime_dependency "httparty"
+  spec.add_runtime_dependency "mail"
+  spec.add_runtime_dependency "google-api-client", ">=0.23.0"
+  spec.add_runtime_dependency "concurrent-ruby", "~> 1.0.5"
+end

metadata

@@ -0,0 +1,162 @@
+--- !ruby/object:Gem::Specification
+name: navi_email_sync
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Sanjib
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2018-06-13 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.11'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.11'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: httparty
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: mail
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: google-api-client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.23.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.23.0
+- !ruby/object:Gem::Dependency
+  name: concurrent-ruby
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.5
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.5
+description: " Write a longer description or delete this line."
+email:
+- sanjib@whitehatengineering.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- CODE_OF_CONDUCT.md
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/.DS_Store
+- bin/console
+- bin/setup
+- config/navi_client.sample.yml
+- lib/client.rb
+- lib/cloud/navi_cloud_client.rb
+- lib/gmail_client.rb
+- lib/http_service/naviai.rb
+- lib/local/navi_local_client.rb
+- lib/navi_client/version.rb
+- lib/navi_email_sync.rb
+- navi_email_sync.gemspec
+homepage: http://navihq.com
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.7.3
+signing_key: 
+specification_version: 4
+summary: Write a short summary, because Rubygems requires one.
+test_files: []