Smile Bank to Ledger

require 'mechanize'
require 'nokogiri'












a = Mechanize.new





















#Fetch logon page
page = a.get("https://banking.smile.co.uk/SmileWeb/start.do")

#Since dealing with sensitive logon data don't want to get it from anywhere apart from keyboard input

#Sortcode and Account number login
puts "Enter sortcode"
sortCode = gets.strip
puts "Enter Account number"
accountNumber = gets.strip
page.form.sortCode = sortCode
page.form.accountNumber = accountNumber
page = page.form.submit


#TODO: Need to check for timeouts


#Passcode login
doc = Nokogiri::HTML(page.body)
puts doc.xpath("//label")[0].text
page.form.firstPassCodeDigit = gets.strip
puts doc.xpath("//label")[1].text
page.form.secondPassCodeDigit = gets.strip
page = page.form.submit




#Other logon details
#TODO: Have yet to figure out date of birth
puts page.form.fields[1].name


page.form.fields[1].value = gets.strip


page = page.form.submit




if page.form.action.include?("bulletinBoard")
	page = page.form.submit
end


#Logged on!

#Current account

page = a.click(page.link_with(:text => "current account"))



#recent items
















doc = Nokogiri::HTML(page.body)
#Look for summaryTable
rows = doc.xpath("//table[@class='summaryTable']//tr")
#Do something with that!
#array of hashes?
transactions = []
#Skip first row (ths)
rows = rows[1..-1]
rows.each do |row|
	#Skip last row, maybe
	unless row.elements.length == 1
		#TODO: Need to skip BROUGHT FORWARD as well for previous item pages
		date = row.elements[0].text
		name = row.elements[1].text
		credit = row.elements[2].text
		debit = row.elements[3].text

		transactions << { :date => date, :name => name, :credit => credit, :debit => debit}
	end
end	



#TODO: Need to figure out a way to get input for number of previous pages required. Well, a good way, perhaps limit by a date?
puts "Enter date back to fetch transactions to"
#Assume ISO thingy format?
date_limit = gets.strip #.convert_to_date

#Previous statements
page = a.click(page.link_with(:text => "previous statements"))
#Get first page of previous statements
page = a.click(page.links_with(:href => %r{getDomestic} )[0])
#TODO: get transactions in the same way

#then click back
#TODO: This needs to be in a loop
page = a.click(page.link_with(:text => "previous statement page"))
#Need someway of stopping if I call the above in a loop. So probably check dates.

#Write transactions out 
#TODO: Might need to sort by date


File.open("smile.dat", "w") do |file|
	transactions.each do |transaction|
		file << Date.parse(transaction[:date]).strftime("%Y/%m/%d")+"\t"+transaction[:name]+"\n"
		if transaction[:credit].include?("£")
			amount = transaction[:credit].gsub(/£/, "£-")
		else
			amount = transaction[:debit]
		end
		file << "\t"+"Assets:Smile:Current"+"\t"+amount+"\n"
		file << "\n"
	end
end

require 'optparse'
require 'mechanize'
require 'nokogiri'

optparse = OptionParser.new do |opts|
	opts.on('-d', '--date DATE', "Date") { |d| Date_back = d }
end
optparse.parse!

class Smile
	attr_writer :date_back 
	attr_reader :page
	attr_reader :transactions

	def initialize
		@a = Mechanize.new
		@transactions = []
	end


	def transactions(date_back)
		@date_back = Date.parse(date_back)
		logon_start
		logon_passcode
		logon_other
		logon_bulletin
		recent_transactions
		previous_statements
		write_ledger
	end


	#All these private
	#Note sure how valid it is breaking these out into separate methods
	#Might make more sense in a rakefile as separate tasks with dependencies,
	#but these methods can't be used alone
	#Also, really not sure about using instance variables instead of passing args
	def logon_start
		@page = @a.get("https://banking.smile.co.uk/SmileWeb/start.do")

		#Since dealing with sensitive logon data don't want to get it from anywhere apart from keyboard input

		#Sortcode and Account number login
		puts "Enter sortcode"
		sortCode = gets.strip
		puts "Enter Account number"
		accountNumber = gets.strip
		@page.form.sortCode = sortCode
		@page.form.accountNumber = accountNumber
		@page = @page.form.submit
	end



	def logon_passcode
		#Passcode login
		doc = Nokogiri::HTML(@page.body)
		puts doc.xpath("//label")[0].text
		@page.form.firstPassCodeDigit = gets.strip
		puts doc.xpath("//label")[1].text
		@page.form.secondPassCodeDigit = gets.strip
		@page = @page.form.submit
	end


	def logon_other
		#Other logon details

		@page.form.fields.each do |field|
			unless field.type == "hidden"
				puts field.name
				field.value = gets.strip
			end
		end
		@page = @page.form.submit
	end


	def logon_bulletin
		if @page.form.action.include?("bulletinBoard")
			@page = @page.form.submit
		end
	end




	def recent_transactions
		@page = @a.click(@page.link_with(:text => "current account"))
		get_transactions
	end



	def previous_statements
		#Previous statements
		@page = @a.click(@page.link_with(:text => "previous statements"))
		#Get first page of previous statements
		@page = @a.click(@page.links_with(:href => %r{getDomestic} )[0])
		date = Date.today
		until date < @date_back
			get_transactions
			date = Date.parse(@transactions[-1][:date])
			@page = @a.click(@page.link_with(:text => "previous statement page"))
		end
	end


	def get_transactions
		doc = Nokogiri::HTML(@page.body)
		#Look for summaryTable
		rows = doc.xpath("//table[@class='summaryTable']//tr")



		#Skip first row (table headers, ths)
		rows = rows[1..-1]
		rows.each do |row|
			#Skip last row, maybe
			unless row.elements.length == 1
				#TODO: Need to skip BROUGHT FORWARD as well for previous item pages
				date = row.elements[0].text
				name = row.elements[1].text
				credit = row.elements[2].text
				debit = row.elements[3].text
				unless name.include?("BROUGHT FORWARD")
					@transactions << { :date => date, :name => name, :credit => credit, :debit => debit}
				end
			end
		end	
	end


















	#TODO: Might need to sort by date

	def write_ledger
		File.open("smile.dat", "w") do |file|
			@transactions.each do |transaction|
				file << Date.parse(transaction[:date]).strftime("%Y/%m/%d")+"\t"+transaction[:name]+"\n"
				if transaction[:debit].include?("£")
					amount = transaction[:debit].gsub(/£/, "£-")
				else
					amount = transaction[:credit]
				end
				file << "\t"+"Assets:Smile:Current"+"\t"+amount+"\n"
				file << "\n"
			end
		end
	end

	private :logon_start
	private :logon_passcode
	private :logon_other
	private :logon_bulletin
	private :recent_transactions
	private :previous_statements
	private :write_ledger
end


if defined?(Date_back)
	smile = Smile.new
	smile.transactions(Date_back)
end