cgimenes/pocket_importer.rb

## pocket_importer.rb
require 'httparty'
require 'socket'
require 'cgi'
require 'csv'
require 'json'
require 'thor'

class PocketAPI
    def initialize(consumer_key)
        @consumer_key = consumer_key
        @redirect_uri = 'http://localhost:5678'
        authenticate
    end

    def wait_authorization()
        puts 'Waiting for authorization'

        server = TCPServer.new 5678
        session = server.accept
        request = session.gets
        session.print "HTTP/1.1 200\r\n" # 1
        session.print "Content-Type: text/html\r\n" # 2
        session.print "\r\n" # 3
        session.print "<html><head></head><body><h2>Now, you can close this tab</h2></body></html>" #4
        session.close

        puts 'Authorized'
    end

    def get_code()
        response = HTTParty::post 'https://getpocket.com/v3/oauth/request', query: { consumer_key: @consumer_key, redirect_uri: @redirect_uri }
        CGI::parse(response.body)['code'][0]
    end

    def get_access_token(code)
        response = HTTParty::post 'https://getpocket.com/v3/oauth/authorize', query: { consumer_key: @consumer_key, code: code }
        CGI::parse(response.body)['access_token'][0]
    end

    def authenticate()
        code = get_code
        puts "https://getpocket.com/auth/authorize?request_token=#{code}&redirect_uri=#{@redirect_uri}"
        wait_authorization
        @access_token = get_access_token code
    end

    def get()
        HTTParty::post "https://getpocket.com/v3/get", query: { consumer_key: @consumer_key, access_token: @access_token }
    end

    def add(url)
        HTTParty::post "https://getpocket.com/v3/add", query: { url: url, consumer_key: @consumer_key, access_token: @access_token }
    end

    def search(text)
        HTTParty::post "https://getpocket.com/v3/get", query: { search: text, consumer_key: @consumer_key, access_token: @access_token }
    end

    def exists?(url)
        results = JSON.parse(search(url).body)
        !results['list'].empty?
    end
end

class PocketImporter < Thor
    desc "import_file FILE CONSUMER_KEY", "import lines at FILE"
    def import_file(file, consumer_key)
        File.foreach(file) do |line|
            api(consumer_key).add line
        end
    end

    desc "import_csv CSV_FILE CONSUMER_KEY", "import column 'URL' for each rows at CSV_FILE"
    def import_csv(file, consumer_key)
        CSV.foreach(file, :headers => true) do |r|
            api(consumer_key).add r['URL']
        end
    end

    desc "check_file FILE CONSUMER_KEY", "check if lines at FILE were added"
    def check_file(file, consumer_key)
        File.foreach(file) do |line|
            unless api(consumer_key).exists? line
                puts line
            end
        end
    end

    desc "check_csv FILE CONSUMER_KEY", "check if CSV rows at FILE were added"
    def check_csv(file, consumer_key)
        CSV.foreach(file, :headers => true) do |r|
            unless api(consumer_key).exists? r['URL']
                puts r['URL']
            end
        end
    end

    private

    def api(consumer_key)
        if @api.nil?
            @api = PocketAPI.new consumer_key
        end
        @api
    end
end

PocketImporter.start(ARGV)
	require 'httparty'
	require 'socket'
	require 'cgi'
	require 'csv'
	require 'json'
	require 'thor'

	class PocketAPI
	def initialize(consumer_key)
	@consumer_key = consumer_key
	@redirect_uri = 'http://localhost:5678'
	authenticate
	end

	def wait_authorization()
	puts 'Waiting for authorization'

	server = TCPServer.new 5678
	session = server.accept
	request = session.gets
	session.print "HTTP/1.1 200\r\n" # 1
	session.print "Content-Type: text/html\r\n" # 2
	session.print "\r\n" # 3
	session.print "<html><head></head><body><h2>Now, you can close this tab</h2></body></html>" #4
	session.close

	puts 'Authorized'
	end

	def get_code()
	response = HTTParty::post 'https://getpocket.com/v3/oauth/request', query: { consumer_key: @consumer_key, redirect_uri: @redirect_uri }
	CGI::parse(response.body)['code'][0]
	end

	def get_access_token(code)
	response = HTTParty::post 'https://getpocket.com/v3/oauth/authorize', query: { consumer_key: @consumer_key, code: code }
	CGI::parse(response.body)['access_token'][0]
	end

	def authenticate()
	code = get_code
	puts "https://getpocket.com/auth/authorize?request_token=#{code}&redirect_uri=#{@redirect_uri}"
	wait_authorization
	@access_token = get_access_token code
	end

	def get()
	HTTParty::post "https://getpocket.com/v3/get", query: { consumer_key: @consumer_key, access_token: @access_token }
	end

	def add(url)
	HTTParty::post "https://getpocket.com/v3/add", query: { url: url, consumer_key: @consumer_key, access_token: @access_token }
	end

	def search(text)
	HTTParty::post "https://getpocket.com/v3/get", query: { search: text, consumer_key: @consumer_key, access_token: @access_token }
	end

	def exists?(url)
	results = JSON.parse(search(url).body)
	!results['list'].empty?
	end
	end

	class PocketImporter < Thor
	desc "import_file FILE CONSUMER_KEY", "import lines at FILE"
	def import_file(file, consumer_key)
	File.foreach(file) do \|line\|
	api(consumer_key).add line
	end
	end

	desc "import_csv CSV_FILE CONSUMER_KEY", "import column 'URL' for each rows at CSV_FILE"
	def import_csv(file, consumer_key)
	CSV.foreach(file, :headers => true) do \|r\|
	api(consumer_key).add r['URL']
	end
	end

	desc "check_file FILE CONSUMER_KEY", "check if lines at FILE were added"
	def check_file(file, consumer_key)
	File.foreach(file) do \|line\|
	unless api(consumer_key).exists? line
	puts line
	end
	end
	end

	desc "check_csv FILE CONSUMER_KEY", "check if CSV rows at FILE were added"
	def check_csv(file, consumer_key)
	CSV.foreach(file, :headers => true) do \|r\|
	unless api(consumer_key).exists? r['URL']
	puts r['URL']
	end
	end
	end

	private

	def api(consumer_key)
	if @api.nil?
	@api = PocketAPI.new consumer_key
	end
	@api
	end
	end

	PocketImporter.start(ARGV)