ayadn 0.6.4 → 1.0.0

data/lib/ayadn/authorize.rb

@@ -1,29 +1,128 @@
-#!/usr/bin/env ruby
 # encoding: utf-8
-class AyaDN
-	def ayadnAuthorize(action)
-		$files.makedir($tools.ayadn_configuration[:authorization_path])
-		if action == "reset"
-			$files.reset_credentials
-		end
-		auth_token = $files.auth_read
-		if auth_token == nil
-			url = @api.makeAuthorizeURL
-			case $tools.ayadn_configuration[:platform]
-			when $tools.winplatforms
-				puts $status.launchAuthorization("win")
-			when /linux/
-				puts $status.launchAuthorization("linux")
-			else
-				puts $status.launchAuthorization("osx")
-				$tools.startBrowser(url)
-			end
-			auth_token = STDIN.gets.chomp()
-			$files.auth_write(auth_token)
-			puts $status.authorized
-			sleep 3
-			puts $tools.helpScreen
-			puts "Enjoy!\n".cyan
-		end
-	end
-end
+module Ayadn
+  class Authorize
+
+    def authorize
+      puts "\e[H\e[2J"
+      try_remove_old_ayadn
+      show_link
+      token = get_token
+      check_token(token)
+      puts "\n\nThanks! Contacting App.net...\n".color(:green)
+      user = create_user_data(token, Dir.home + "/ayadn")
+      prepare(user)
+      puts "Creating configuration...\n".color(:green)
+      Settings.load_config
+      Logs.create_logger
+      install
+      puts Status.done
+      Errors.info "Done!"
+      puts "\nThank you for using Ayadn. Enjoy!\n\n".color(:yellow)
+    end
+
+    private
+
+    def prepare(user)
+      puts "Ok! Creating Ayadn folders...\n".color(:green)
+      create_config_folders(user)
+      puts "Saving user token...\n".color(:green)
+      create_token_file(user)
+      puts "Creating user account for #{user.handle}...\n".color(:green)
+      accounts_db = Daybreak::DB.new("#{user.home_path}/accounts.db")
+      create_account(user, accounts_db)
+    end
+
+    def install
+      puts "Creating api and config files...\n".color(:green)
+      Errors.info "Creating api, version and config files..."
+      Errors.info "Creating version file..."
+      Settings.init_config
+    end
+
+    def create_account(user, accounts_db)
+      accounts_db[user.username] = {username: user.username, id: user.id, handle: user.handle, path: user.user_path}
+      accounts_db['ACTIVE'] = user.username
+      accounts_db.flush
+      accounts_db.close
+    end
+
+    def create_token_file(user)
+      File.write("#{user.user_path}/auth/token", user.token)
+    end
+
+    def create_config_folders(user)
+      begin
+        FileUtils.mkdir_p(user.user_path)
+        %w{log db pagination config auth downloads backup posts messages lists}.each do |target|
+          Dir.mkdir("#{user.user_path}/#{target}")
+        end
+      rescue => e
+        puts "\nError creating Ayadn #{user.handle} account folders.\n\n"
+        puts "Error: #{e}"
+        exit
+      end
+    end
+
+    def show_link
+      puts "\nPlease click this URL, or open a browser then copy/paste it:\n".color(:cyan)
+      puts Endpoints.new.authorize_url
+      puts "\n"
+      puts "On this page, log in with your App.net account to authorize Ayadn.\n".color(:cyan)
+      puts "You will then be redirected to a page showing a 'user token' (your secret code).\n".color(:cyan)
+      puts "Copy it then paste it here:\n".color(:yellow)
+      print "> "
+    end
+
+    def get_user(token)
+      JSON.parse(RestClient.get("https://alpha-api.app.net/stream/0/users/me?access_token=#{token}", :verify_ssl => OpenSSL::SSL::VERIFY_NONE) {|response, request, result| response })
+    end
+
+    def get_token
+      begin
+        STDIN.gets.chomp()
+      rescue Interrupt
+        puts Status.canceled
+        exit
+      end
+    end
+
+    def check_token(token)
+      if token.empty? || token.nil?
+        puts "\n\nOops, something went wrong, I couldn't get the token. Please try again.\n\n".color(:red)
+        exit
+      end
+    end
+
+    def try_remove_old_ayadn
+      if FileOps.old_ayadn?
+        answer = ask_del_old_ayadn
+        unless answer.downcase == "y"
+          puts Status.canceled
+          exit
+        end
+        puts "\nDeleting old version...\n".color(:green)
+        begin
+          old_dir = Dir.home + "/ayadn"
+          FileUtils.remove_dir(old_dir)
+        rescue => e
+          puts "Unable to remove folder: #{old_dir}\n\n".color(:red)
+          raise e
+        end
+      end
+    end
+
+    def ask_del_old_ayadn
+      puts "\nAn obsolete version of Ayadn has been detected and will be deleted. Install and authorize the new version? [y/N]\n".color(:red)
+      print "> "
+      STDIN.getch
+    end
+
+    def create_user_data(token, home_path)
+      resp = get_user(token)
+      model = Struct.new(:resp, :username, :id, :handle, :home_path, :user_path, :token)
+      username = resp['data']['username']
+      model.new(resp, username, resp['data']['id'], "@" + username, home_path, home_path + "/#{username}", token)
+    end
+
+  end
+end

data/lib/ayadn/blacklist.rb

@@ -0,0 +1,116 @@
+# encoding: utf-8
+module Ayadn
+  class Blacklist < Thor
+    desc "blacklist add TYPE TARGET", "Adds a mention, hashtag or client to your blacklist"
+    long_desc Descriptions.blacklist_add
+    def add(*args)
+      if args.length != 2
+        puts Status.type_and_target_missing
+      end
+      blacklist = BlacklistWorkers.new
+      blacklist.add(args)
+      puts Status.done
+    end
+
+    desc "blacklist remove TYPE TARGET", "Removes a mention, hashtag or client from your blacklist"
+    long_desc Descriptions.blacklist_remove
+    def remove(*args)
+      if args.length != 2
+        puts Status.type_and_target_missing
+      end
+      blacklist = BlacklistWorkers.new
+      blacklist.remove(args)
+      puts Status.done
+    end
+
+    desc "blacklist list", "List the content of your blacklist"
+    long_desc Descriptions.blacklist_list
+    def list
+      blacklist = BlacklistWorkers.new
+      blacklist.list
+    end
+
+    desc "blacklist import DATABASE", "Imports a blacklist database from another Ayadn account"
+    long_desc Descriptions.blacklist_import
+    def import(database)
+      blacklist = BlacklistWorkers.new
+      blacklist.import(database)
+      puts Status.done
+    end
+  end
+
+  class BlacklistWorkers
+    def initialize
+      Settings.load_config
+      Settings.get_token
+      Settings.init_config
+      Logs.create_logger
+      Databases.open_databases
+    end
+    def import(database)
+      begin
+        new_db = File.realpath(database)
+        if File.exist?(new_db)
+          Databases.import_blacklist(new_db)
+          Logs.rec.info "Imported '#{new_db}' values in blacklist database."
+        else
+          puts "\nFile '#{new_db}' doesn't exist.\n\n".color(:red)
+          Logs.rec.warn "File '#{new_db}' doesn't exist."
+        end
+      ensure
+        Databases.close_all
+      end
+    end
+    def add(args)
+      begin
+        type, target = args[0], args[1]
+        case type
+        when 'mention', 'mentions'
+          target = Workers.add_arobase_if_missing([target])
+          Databases.add_mention_to_blacklist(target)
+          Logs.rec.info "Added '#{target}' to blacklist of mentions."
+        when 'client', 'source'
+          Databases.add_client_to_blacklist(target)
+          Logs.rec.info "Added '#{target}' to blacklist of clients."
+        when 'hashtag', 'tag'
+          Databases.add_hashtag_to_blacklist(target)
+          Logs.rec.info "Added '#{target}' to blacklist of hashtags."
+        else
+          puts Status.wrong_arguments
+        end
+      ensure
+        Databases.close_all
+      end
+    end
+    def remove(args)
+      begin
+        type, target = args[0], args[1]
+        case type
+        when 'mention', 'mentions'
+          target = Workers.add_arobase_if_missing([target])
+          Databases.remove_from_blacklist(target)
+          Logs.rec.info "Removed '#{target}' from blacklist of mentions."
+        when 'client', 'source', 'hashtag', 'tag'
+          Databases.remove_from_blacklist(target)
+          Logs.rec.info "Removed '#{type}:#{target}' from blacklist."
+        else
+          puts Status.wrong_arguments
+        end
+      ensure
+        Databases.close_all
+      end
+    end
+    def list
+      begin
+        list = Databases.blacklist
+        unless list.empty?
+          puts Workers.new.build_blacklist_list(list)
+        else
+          abort(Status.empty_list)
+        end
+      ensure
+        Databases.close_all
+      end
+    end
+  end
+end

data/lib/ayadn/cnx.rb

@@ -0,0 +1,105 @@
+# encoding: utf-8
+module Ayadn
+  class CNX
+
+    def self.get_response_from(url)
+      begin
+        RestClient.get(url) do |response, request, result| #, :verify_ssl => OpenSSL::SSL::VERIFY_NONE
+          check(response)
+        end
+      rescue SocketError => e
+        puts "\nConnection error.".color(:red)
+        Errors.global_error("cnx.rb/get", url, e)
+      rescue SystemCallError => e
+        puts "\nConnection error.".color(:red)
+        Errors.global_error("cnx.rb/get", url, e)
+      rescue => e
+        Errors.global_error("cnx.rb/get", url, e)
+      end
+    end
+
+    def self.check(response)
+      case response.code
+      when 200
+        response
+      when 204
+        puts "\n'No content'".color(:red)
+        Errors.global_error("cnx.rb", response.headers, "NO CONTENT")
+      when 400
+        puts "\n'Bad request'".color(:red)
+        Errors.global_error("cnx.rb", response.headers, "BAD REQUEST")
+      when 401
+        puts "\n'Unauthorized'".color(:red)
+        Errors.global_error("cnx.rb", response.headers, "UNAUTHORIZED")
+      when 403
+        puts "\n'Forbidden'".color(:red)
+        Errors.global_error("cnx.rb", response.headers, "FORBIDDEN")
+      when 405
+        puts "\n'Method not allowed'".color(:red)
+        Errors.global_error("cnx.rb", response.headers, "METHOD NOT ALLOWED")
+      when 429
+        puts "\n'Too many requests'".color(:red)
+        puts "\n\nAyadn made too many requests to the App.net API. You should wait at least ".color(:cyan) + "#{response.headers[:retry_after]} ".color(:red) + "seconds before trying again. Maybe you launched a lot of Ayadn instances at the same time? That's no problem, but in this case you should increase the value of the scroll timer (do `ayadn set scroll timer 2.5` for example).".color(:cyan)
+        Errors.global_error("cnx.rb", response.headers, "TOO MANY REQUESTS")
+      when 500
+        puts "\n'App.net server error'".color(:red)
+        Errors.global_error("cnx.rb", response.headers, "APP.NET SERVER ERROR")
+      when 507
+        puts "\n'Insufficient storage'".color(:red)
+        Errors.global_error("cnx.rb", response.headers, "INSUFFICIENT STORAGE")
+      else
+        response
+      end
+    end
+
+    def self.delete(url)
+      begin
+        #RestClient::Resource.new(url).delete
+        RestClient.delete(url) do |response, request, result|
+          check(response)
+        end
+      rescue SocketError => e
+        puts "\nConnection error.".color(:red)
+        Errors.global_error("cnx.rb/delete", url, e)
+      rescue SystemCallError => e
+        puts "\nConnection error.".color(:red)
+        Errors.global_error("cnx.rb/delete", url, e)
+      rescue => e
+        Errors.global_error("cnx.rb/delete", url, e)
+      end
+    end
+
+    def self.post(url, payload = nil)
+      begin
+        RestClient.post(url, payload.to_json, :content_type => :json, :accept => :json) do |response, request, result|
+          check(response)
+        end
+      rescue SocketError => e
+        puts "\nConnection error.".color(:red)
+        Errors.global_error("cnx.rb/post", url, e)
+      rescue SystemCallError => e
+        puts "\nConnection error.".color(:red)
+        Errors.global_error("cnx.rb/post", url, e)
+      rescue => e
+        Errors.global_error("cnx.rb/post", [url, payload], e)
+      end
+    end
+
+    def self.put(url, payload)
+      begin
+        RestClient.put(url, payload.to_json, :content_type => :json, :accept => :json) do |response, request, result|
+          check(response)
+        end
+      rescue SocketError => e
+        puts "\nConnection error.".color(:red)
+        Errors.global_error("cnx.rb/put", url, e)
+      rescue SystemCallError => e
+        puts "\nConnection error.".color(:red)
+        Errors.global_error("cnx.rb/put", url, e)
+      rescue => e
+        Errors.global_error("cnx.rb/put", [url, payload], e)
+      end
+    end
+
+  end
+end

data/lib/ayadn/databases.rb

@@ -0,0 +1,110 @@
+# encoding: utf-8
+module Ayadn
+
+  class Databases
+
+    class << self
+      attr_accessor :users, :index, :pagination, :aliases, :blacklist
+    end
+
+    def self.open_databases
+      @users = Daybreak::DB.new "#{Settings.config[:paths][:db]}/users.db"
+      @index = Daybreak::DB.new "#{Settings.config[:paths][:pagination]}/index.db"
+      @pagination = Daybreak::DB.new "#{Settings.config[:paths][:pagination]}/pagination.db"
+      @aliases = Daybreak::DB.new "#{Settings.config[:paths][:db]}/aliases.db"
+      @blacklist = Daybreak::DB.new "#{Settings.config[:paths][:db]}/blacklist.db"
+    end
+
+    def self.close_all
+      [@users, @index, @pagination, @aliases, @blacklist].each do |db|
+        db.flush
+        db.close
+      end
+    end
+
+    def self.add_mention_to_blacklist(target)
+      @blacklist[target] = :mention
+    end
+    def self.add_client_to_blacklist(target)
+      @blacklist[target] = :client
+    end
+    def self.add_hashtag_to_blacklist(target)
+      @blacklist[target] = :hashtag
+    end
+    def self.remove_from_blacklist(target)
+      @blacklist.delete(target)
+    end
+    def self.import_blacklist(blacklist)
+      new_list = Daybreak::DB.new blacklist
+      new_list.each {|name,type| @blacklist[name] = type}
+      new_list.close
+    end
+
+    def self.save_max_id(stream)
+      @pagination[stream['meta']['marker']['name']] = stream['meta']['max_id']
+    end
+
+    def self.create_alias(channel_id, channel_alias)
+      @aliases[channel_alias] = channel_id
+    end
+
+    def self.delete_alias(channel_alias)
+      @aliases.delete(channel_alias)
+    end
+
+    def self.get_channel_id(channel_alias)
+      @aliases[channel_alias]
+    end
+
+    def self.import_aliases(aliases)
+      new_aliases = Daybreak::DB.new aliases
+      new_aliases.each {|al,id| @aliases[al] = id}
+      new_aliases.close
+    end
+
+    def self.get_alias_from_id(channel_id)
+      @aliases.each do |al, id|
+        return al if id == channel_id
+      end
+      nil
+    end
+
+    def self.save_indexed_posts(posts)
+      @index.clear
+      posts.each do |id, hash|
+        @index[id] = hash
+      end
+    end
+
+    def self.get_index_length
+      @index.length
+    end
+
+    def self.get_post_from_index(number)
+      unless number > @index.length || number <= 0
+        @index.to_h.each do |id, values|
+          return values if values[:count] == number
+        end
+      else
+        puts "\nNumber must be in the range of the indexed posts.\n".color(:red)
+        Errors.global_error("databases/get_post_from_index", number, "out of range")
+      end
+    end
+
+    def self.add_to_users_db_from_list(list)
+      list.each do |id, content_array|
+        @users[id] = {content_array[0] => content_array[1]}
+      end
+    end
+
+    def self.add_to_users_db(id, username, name)
+      @users[id] = {username => name}
+    end
+
+    def self.has_new?(stream, title)
+      stream['meta']['max_id'].to_i > @pagination[title].to_i
+    end
+
+  end
+
+end