Smile Bank to Ledger

require 'optparse'
require 'mechanize'
require 'nokogiri'
require 'highline/import'

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 :transactions
	attr_reader :balance
	attr_reader :available

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




	end


	def transactions(date_back)
		@date_back = Date.parse(date_back)
		previous_statements
		#previous calls recent (I could change this though so it reads more sensibly here)
		write_ledger
	end


	#All these private
	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
		sortCode = ask("Enter sortcode") { |a| a.echo = "*" }
		accountNumber = ask("Enter Account number") { |a| a.echo = "*" }
		page.form.sortCode = sortCode
		page.form.accountNumber = accountNumber
		page = page.form.submit
	end


	def logon_passcode
		#Requires
		page = logon_start
		#Passcode login
		doc = Nokogiri::HTML(page.body)
		page.form.firstPassCodeDigit = ask(doc.xpath("//label")[0].text) { |a| a.echo = "*" }
		page.form.secondPassCodeDigit = ask(doc.xpath("//label")[1].text) { |a| a.echo = "*" }
		page = page.form.submit
	end


	def logon_other
		#requires
		page = logon_passcode
		#Other logon details
		page.form.fields.each do |field|
			unless field.type == "hidden"
				field.value = ask(field.name) { |a| a.echo = "*" }
			end
		end
		page = page.form.submit
	end


	def logon_bulletin
		#requires
		page = logon_other
		if page.form
			if page.form.action.include?("bulletinBoard")
				page = page.form.submit
			end
		end
		#Need this to be the last thing so it gets returned if still logon_other
		page
	end


	def recent_transactions
		#requires
		page = logon_bulletin
		page = @a.click(page.link_with(:text => "current account"))
		get_transactions(page)
		doc = Nokogiri::HTML(page.body)
		#Any trailiing minus needs to be moved to before the number, but any trailing plus needs to be removed, don't think this can be done in one gsub.
		@balance = doc.xpath("//td[@class='recentTransactionsAccountData']/table/tr[2]/td[2]").text.strip.gsub(/(£)(\d+\.\d+)(\-)/, "\\1\\3\\2").gsub(/\+/, "")
		@available = doc.xpath("//td[@class='recentTransactionsAccountData']/table/tr[3]/td[2]").text.strip
		#Need to return this
		page
	end


	def previous_statements
		#requires
		page = recent_transactions
		#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(page)
			#On previous statement pages most recent transaction is at bottom
			#This is opposite to recent item page
			date = @transactions[-1][:date]
			page = @a.click(page.link_with(:text => "previous statement page"))
		end
	end


	def get_transactions(page)
		doc = Nokogiri::HTML(page.body)
		#Look for summaryTable, sometimes can be two, get the last
		rows = doc.xpath("//table[@class='summaryTable'][position()=last()]//tr")
		#Skip first row (table headers, ths)
		rows = rows[1..-1]
		transactions = []
		rows.each do |row|
			#Skip last row, maybe
			#Also, sometimes bugs or invalid markup and so even though last row does have just one td doesn't get reported that way
			#So use begin/rescue
			begin
				#Might as well parse date here
				date = Date.parse(row.elements[0].text.strip)
				name = row.elements[1].text.strip
				credit = row.elements[2].text.strip
				debit = row.elements[3].text.strip
				unless name.include?("BROUGHT FORWARD")
					transactions << { :date => date, :name => name, :credit => credit, :debit => debit}
				end
			rescue
			end

		end	
		#Sort in order of newest date first so can add on
		@transactions += transactions.sort { | t1, t2 | t2[:date] <=> t1[:date] }

	end


	def write_ledger
		balance_assertion = ""
		File.open("smile.dat", "w") do |file|
			#Reverse for writing out
			@transactions.reverse!
			@transactions.each.with_index do |transaction, idx|
				file << transaction[:date].strftime("%Y/%m/%d")+"\t*\t"+transaction[:name]+"\n"
				if transaction[:debit].include?("£")
					#Quote all commodities because having ledger utf-8 trouble
					amount = transaction[:debit].gsub(/£/, "\"£\"-")
					type = "Expenses:"
				else
					amount = transaction[:credit].gsub(/£/, "\"£\"")
					type = "Income:"
				end
				if idx == @transactions.length-1
					#Include balance assertion at the end.
					balance_assertion = "\t= "+@balance
					balance_assertion = balance_assertion.gsub(/£/, "\"£\"")
				end
				file << "\t"+"Assets:Smile:Current"+"\t"+amount+balance_assertion+"\n"
				file << "\t"+type+"\n"
				file << "\n"
			end
			#Write out account balance as a comment at the end
			file << "#Available: "+@available+"\n"
		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