curry_house_twitter.rb

require 'twitter'
require 'rufus-scheduler'

OpenSSL::SSL::VERIFY_PEER = OpenSSL::SSL::VERIFY_NONE

def init
  #curryhouse02
  config = {
      :consumer_key => 'NNtgfDPbs3e2RfYogLwozovGn',
      :consumer_secret => 'rZAtPRt3IlTJ0gUjMALihOYbDCqytVhse58lGPgn863gq5oLss',
      :access_token => '705790849393758208-DkouWibaCQ6xfnjygbm78Gfh9HxA1uB',
      :access_token_secret => 'ZJGwvPMmkwV5f6YaVZrUVgztiGcV1iXupncKGmqhXKXWb'
  }

  $client = Twitter::REST::Client.new(config)


  @db = SQLite3::Database.new './curry_house.sqlite'
  # @caught_tweets = @db.execute('SELECT * FROM tweets')
end

#search for tweets in the timeline that include 'order' or 'cancel' command
def search_timeline
  tweets = $client.mentions_timeline()
  most_recent = tweets.take(20)
  new_orders = []
  most_recent.each do |tweet|
    #regex= contains any alphabet character or contains only spaces
    valid_format = tweet.text.include?('order') && (/([a-zA-z]+)|(\A\s*\z)/ =~ tweet.text.partition('order')[2]).nil?
    unprocessed = @db.get_first_value('SELECT COUNT(*) FROM tweets WHERE id = ?',tweet.id)==0
    valid_cancellation = tweet.text.include?('cancel')&& !tweet.attrs[:user][:screen_name].equal?('curryhouse02')

    if (valid_format && unprocessed) || valid_cancellation
      new_orders.push tweet
    end
  end
  return new_orders
end

#Saves the order information to the database and tweets back to the customer.
def process_order tweet
  twitter_username = tweet.attrs[:user][:screen_name]
  adr = @db.get_first_value('SELECT address FROM customer WHERE twitterAcc=?',twitter_username)
  account_registered = !(adr.nil? || adr.eql?('')) ? true : false
  account_blacklisted = @db.get_first_value('SELECT blacklisted FROM customer WHERE twitterAcc=?',twitter_username)==1;
  order_text = tweet.text.partition('order ')[2]

  if account_registered && !account_blacklisted
    items = order_text.gsub(/^\s+|\s+$/,'').split(/\s+/) #strip leading & trailing spaces and split into items
    sum = 0
    items.each do |item|
      max_id = @db.get_first_value 'SELECT max(id) FROM menu'
      cost = @db.get_first_value('SELECT unitPrice FROM menu WHERE id = ?',item)
      if !cost.nil?
        sum += cost
      end
    end
    sum=sum.round(2)
    new_balance = @db.get_first_value('SELECT balance FROM customer WHERE twitterAcc = ?',twitter_username)-sum

    order_id = @db.execute 'SELECT max(order_id) FROM tweets'
    order_id = order_id[0][0] + 1

    city = @db.get_first_value('SELECT city FROM tweets WHERE sender = ?',twitter_username)
    tweet_arr = [tweet.id,twitter_username,order_text,'Ordered',order_id,sum,city]

    if new_balance<0
      tweet_arr[3] = 'Unpaid'
      tweet_status_change(tweet_arr)
    else
      @db.execute('UPDATE customer SET balance = ? WHERE twitterAcc = ?', [new_balance,twitter_username])
      check_for_offers twitter_username
      $client.favorite(tweet.id)
      $client.update("Hi @#{twitter_username}! Your order with ID:#{order_id} has been accepted. To cancel go to our website!", :in_reply_to_status_id => tweet.id)
    end

    Stats.increment 'orders',city
    @db.execute('INSERT INTO tweets(id,sender,text,status,order_id,sum,city) VALUES (?,?,?,?,?,?,?)',tweet_arr)

  else
    $client.update("Hi @#{twitter_username}! You must have an address registered in our website to process you order.", :in_reply_to_status_id => tweet.id)
  end
end

def check_for_offers twitter_username
  norders = @db.get_first_value('SELECT orders_made FROM customer WHERE twitterAcc = ?',twitter_username)+1
  @db.execute('UPDATE customer SET orders_made = ? WHERE twitterAcc = ?', [norders,twitter_username])
  req_orders = @db.get_first_value('SELECT norders FROM loyalty_offer')

  if (req_orders!=0) && (norders % req_orders ==0)
    reward = @db.get_first_value('SELECT cp FROM loyalty_offer')
    puts "reward = #{reward}"
    new_balance = @db.get_first_value('SELECT balance FROM customer WHERE twitterAcc = ?',twitter_username)+reward
    puts "new_balance = #{new_balance}"
    @db.execute('UPDATE customer SET balance = ? WHERE twitterAcc = ?', [new_balance,twitter_username])
    msg="We thank you for being a loyal customer. To reward you for your loyalty we have added #{reward} CurryPounds to your balance."
    $client.update("@#{twitter_username}: #{msg}")
  end
end

def process_cancellation tweet
  order_id = tweet.text.partition('cancel')[2].to_i
  min_order_id = @db.get_first_value('SELECT min(order_id) FROM tweets')
  max_order_id = @db.get_first_value('SELECT max(order_id) FROM tweets')

  min_order_id = 0 if min_order_id.nil?
  max_order_id = 0 if max_order_id.nil?
  if (order_id!=0) && (order_id<=max_order_id) && (order_id>=min_order_id) #if valid cancel message
    tweet_info = @db.get_first_row('SELECT * FROM tweets WHERE order_id=?',[order_id])
    order_status = tweet_info[3]

    if !((order_status=='Canceled') || (order_status=='Delivering') || (order_status=='Completed'))
      Stats.increment 'cancellations',@db.get_first_value('SELECT city FROM tweets WHERE order_id = ?',order_id)
      @db.execute('UPDATE tweets SET status = "Canceled" WHERE order_id = ?',[order_id])
      tweet_info = @db.get_first_row('SELECT * FROM tweets WHERE order_id=?',[order_id])

      twitter_acc =  tweet.attrs[:user][:screen_name]
      balance = @db.get_first_value('SELECT balance FROM customer WHERE twitterAcc = ?',twitter_acc)
      new_balance = balance + @db.get_first_value('SELECT sum FROM tweets WHERE order_id=?',order_id)

      @db.execute('UPDATE customer SET balance = ? WHERE twitterAcc = ?',[new_balance,twitter_acc])
      tweet_status_change(tweet_info)
    elsif (order_status=='Delivering')
      sender = @db.execute('SELECT sender FROM tweets WHERE order_id=?',[order_id])[0][0]
      $client.update("@#{sender}: Unfortunately we cannot process your cancellation request as you order is already staged for delivery. Order ID:#{order_id}.")
    end
  end
end


def tweet_status_change(tweet_entry)
  msg = case tweet_entry[3]
          when 'Preparing'
            'Your order is now being prepared!'
          when 'Delivering'
            'Your order has been sent for delivery!'
          when 'Completed'
            'Thank for ordering from us!'
          when 'Canceled'
            'Your order has been successfully canceled!'
          when 'Unpaid'
            "We can't process your request since you don't have enough CurryPounds in your balance."
          else
            return
        end
  $client.update("@#{tweet_entry[1]}: #{msg} Order ID:#{tweet_entry[4]}.")#, :in_reply_to_status_id => tweet_entry[0])
end

def cust_offers
  scheduler = Rufus::Scheduler.new

  scheduler.every '14d' do
    code = '00'
    discount_code = code+1
    $client.update("Activate this code on your account and get a discount! CH00#{discount_code}")
  end
end

def ch_twitter
  return $client
end

init