#!rake ;#

class CHMDocument < NSDocument
	attr_reader :chm

	#- (void)makeWindowControllers
	def makeWindowControllers
		c = CHMWindowController.alloc.initWithWindowNibName("CHMDocument")
		self.addWindowController(c)
	end

	#- (BOOL)readFromURL:(NSURL *)inAbsoluteURL ofType:(NSString *)inTypeName error:(NSError **)outError
	def readFromURL_ofType_error(url, type, error)
		@chm = Chmlib::Chm.new(url.path.to_s)
		true
	end

	#- (BOOL)writeToURL:(NSURL *)inAbsoluteURL ofType:(NSString *)inTypeName error:(NSError **)outError
	def writeToURL_ofType_error(url, type, error)
		false
	end

	#- (void)windowControllerDidLoadWindowNib:(NSWindowController *)windowController
	def windowControllerDidLoadWindowNib(cont)
		log "wCDLWN", cont
	end

#	def dataRepresentationOfType(aType)
#	end
#
#	def loadDataRepresentation_ofType(data, aType)
#	end

	def displayName
		dc = NSDocumentController.sharedDocumentController
		i = dc.documents.index(self) + 1
		cmd = [8984].pack("U")
		"#{cmd}#{i}| #{@chm.title}"
	end

	def windowControllerWillLoadNib(cont)
		log cont
	end

	def winwowNibName
		"CHMDocument"
	end
end

class MySearchWindow < NSWindow

	def sendEvent(e)
		if e.oc_type == NSKeyDown
			return if delegate.process_keybinds(e)
		end
		super_sendEvent(e)
	end

end


require "uri"
class CHMWindowController < NSWindowController
	ib_outlet :webview
	ib_outlet :list
	ib_outlet :tree
	ib_outlet :drawer
	ib_outlet :search

	def windowDidLoad
		@chm = self.document.chm
		uri  = URI(self.document.fileURL.absoluteString)
		log uri
		browse @chm.home
		@now = @index = @chm.index.to_a.sort_by {|k,v| k} # cache
		@list.setDataSource(self)
		@list.setDoubleAction("clicked_")
		@list.setAction("clicked_")

		@tree.setAction("treeclicked_")

		@search.setDelegate(self)
		@drawer.open
	end

	# OutlineView
	#    * outlineView:child:ofItem:
	#    * outlineView:isItemExpandable:
	#    * outlineView:numberOfChildrenOfItem:
	#    * outlineView:objectValueForTableColumn:byItem:
	#    * outlineView:setObjectValue:forTableColumn:byItem:

	def outlineView_child_ofItem(ov, index, item)
		(item || @topics)[:children][index]
	end

	def outlineView_isItemExpandable(ov, item)
		(item || @topics)[:children].length.nonzero?
	end

	def outlineView_numberOfChildrenOfItem(ov, item)
		(item || @topics)[:children].length
	end

	def outlineView_objectValueForTableColumn_byItem(ov, column, item)
		if item
			item[:name]
		else
			""
		end
	end

	def treeclicked(sender)
		path = sender.itemAtRow(sender.selectedRow)[:local]
		log "Tree Clicked: #{path}"
		browse path unless path.empty?
	end


	# Tableview
	def numberOfRowsInTableView(table)
		@now.length
	end

	def tableView_objectValueForTableColumn_row(table, column, row)
		@now[row][0]
	end

	def tableView_setObjectValue_forTableColumn_row(table, value, column, row)
	end

	def tableView_willDisplayCell_forTableColumn_row(table, cell, column, row)
	end

	def textShouldBeginEditing(text)
		true
	end

	def textShouldEndEditing(text)
		true
	end

	def acceptsFirstResponder
		true
	end

	# TabView
	def tabView_willSelectTabViewItem(sender, item)
		log item.label
		if item.label == "Tree"
			Thread.start do
				@topics = @chm.topics
				# おちやすい
				@tree.setDataSource(self)
				@tree.reloadData
			end
		end
	end

	# general

	def controlTextDidChange(anot)
		filtering @search.stringValue
		@list.selectRowIndexes_byExtendingSelection(NSIndexSet.alloc.initWithIndex(0), false)
	end

	def controlTextDidEndEditing(anot)
		log "end #{@now.first.inspect}"
	end

	def jumpToCurrent(sender)
		clicked(sender)
	end

	def filtering(str)
		str = str.to_s
		if str =~ /[A-Z]/
			r = /^#{str}/
		else
			r = /^#{str}/i
		end
		@now = @index.select {|k,v|
			k =~ r
		}.sort_by {|k,v| k.length }
		@list.reloadData
	end

	def clicked(sender)
		if @now[@list.selectedRow]
			browse @now[@list.selectedRow][1].first
		end
	end

	def browse(path)
		if path
			path = "/#{path}" unless path[0] == ?/
			h = @webview.stringByEvaluatingJavaScriptFromString("location.pathname+location.hash")
			unless path == h
				r = NSURLRequest.requestWithURL CHMInternalURLProtocol.url_for(@chm, path)
				log r
				@webview.mainFrame.loadRequest r
			end
		end
	end

	def completion(sender)
		return if @search.stringValue.empty?
		return if @now.empty?
		common = ""
		keys = @now.map{|k,v| k.split(//)}
		if @search.stringValue.to_s =~ /[A-Z]/
			keys[0].zip(*keys[1..-1]) do |a|
				m = a.first
				if a.all? {|v| m == v}
					common << m
				else
					break
				end
			end
		else
			keys[0].zip(*keys[1..-1]) do |a|
				m = a.first.downcase
				if a.all? {|v| v && (m == v.downcase)}
					common << m
				else
					break
				end
			end
		end
		if common.length > @search.stringValue.length
			@search.stringValue = common
		end
	end

	# from menu
	def searchActivate(sender)
		log "activate"
		@search.window.makeFirstResponder(@search)
	end

	def nextCandidate(sender)
		if @list.selectedRow <= @now.size
			@list.selectRowIndexes_byExtendingSelection(NSIndexSet.alloc.initWithIndex(@list.selectedRow+1), false)
			@list.scrollRowToVisible(@list.selectedRow)
			clicked(nil)
		end
	end

	def prevCandidate(sender)
		if @list.selectedRow > 0
			@list.selectRowIndexes_byExtendingSelection(NSIndexSet.alloc.initWithIndex(@list.selectedRow-1), false)
			@list.scrollRowToVisible(@list.selectedRow)
			clicked(nil)
		end
	end

	def jumpToHome(sender)
		browse @chm.home
	end

	def performFindPanelAction(sender)
		log "performFindPanelAction"
		# @webview.performFindPanelAction(sender) # なぜかうごかない
		text = @search.stringValue
		@webview.objc_send(
			:searchFor, text,
			:direction, true,
			:caseSensitive, false,
			:wrap, false
		)
	end

	# from MySearchWindow

	def process_keybinds(e)
		if NSInputManager.currentInputManager
			return false unless NSInputManager.currentInputManager.markedRange.empty?
		end
		key = key_string(e)
		log "keyDown (#{e.characters}:#{e.charactersIgnoringModifiers}) -> '#{key}'"
		keybinds = {
			"C-j" => self.method(:nextCandidate),
			"C-n" => self.method(:nextCandidate),
			"C-k" => self.method(:prevCandidate),
			"C-p" => self.method(:prevCandidate),
			"\r"  => self.method(:jumpToCurrent),
			"\t"  => self.method(:completion),
			" "   => Proc.new {|s|
				@webview.stringByEvaluatingJavaScriptFromString <<-JS
					window.scrollBy(0, 200);
				JS
			},
			"S- " => Proc.new {|s|
				@webview.stringByEvaluatingJavaScriptFromString <<-JS
					window.scrollBy(0, -200);
				JS
			},
			"C-\r" => Proc.new {|s|
				@now = @chm.search(@search.stringValue).map {|title,url|
					[title, [url]]
				}
				@list.reloadData
			},
			"C-u" => Proc.new {|s|
				@search.stringValue = ""
			},
			"G-[" => Proc.new {|s|
				@webview.goBack
			},
			"G-]" => Proc.new {|s|
				@webview.goForward
			},
			"G-=" => Proc.new {|s|
				@webview.makeTextLarger(self)
			},
			"G--" => Proc.new {|s|
				@webview.makeTextSmaller(self)
			},
		}
		(1..9).each do |i|
			keybinds["G-#{i}"] = Proc.new {|s|
				dc = NSDocumentController.sharedDocumentController
				if dc.documents[i-1]
					log(dc.documents[i-1].windowControllers)
					dc.documents[i-1].windowControllers.first.showWindow(self)
				end
			}
		end
		if keybinds.key?(key)
			keybinds[key].call(self)
			true
		else
			false
		end
	end

	def key_string(e)
		key = ""
		m = e.modifierFlags
		key << "S-" if m & NSShiftKeyMask > 0
		key << "C-" if m & NSControlKeyMask > 0
		key << "M-" if m & NSAlternateKeyMask > 0
		key << "G-" if m & NSCommandKeyMask > 0 # TODO
		key << e.charactersIgnoringModifiers.to_s
		key
	end

	# webview policyDelegate
	def webView_decidePolicyForNavigationAction_request_frame_decisionListener(
		sender,
		actionInformation,
		request,
		frame,
		listener
	)

		if CHMInternalURLProtocol.canHandleURL(request.URL)
			listener.use
		else
			NSWorkspace.sharedWorkspace.openURL(request.URL)
			listener.ignore
		end
	end

	def webView_decidePolicyForNewWindowAction_request_newFrameName_decisionListener(
		sender,
		actionInformation,
		request,
		frameName,
		listener
	)
		if CHMInternalURLProtocol.canHandleURL(request.URL)
			listener.use
		else
			NSWorkspace.sharedWorkspace.openURL(request.URL)
			listener.ignore
		end
	end

	# webview loading delegate
	def webView_resource_didFinishLoadingFromDataSource(sender, id, datasource)
#		log "loaded"
	end


end