# Copyright, 2014, by Masahiro Sano.
# 
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
# 
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
# 
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.

require 'ffi/clang/lib/token'
require 'ffi/clang/lib/cursor'
require 'ffi/clang/source_location'

module FFI
	module Clang
		class Tokens < AutoPointer
			include Enumerable

			attr_reader :size
			attr_reader :tokens

			def initialize(pointer, token_size, translation_unit)
				ptr = Lib::TokensPointer.new(pointer,token_size, translation_unit)
				super ptr

				@translation_unit = translation_unit
				@size = token_size

				@tokens = []
				cur_ptr = pointer
				token_size.times {
					@tokens << Token.new(cur_ptr, translation_unit)
					cur_ptr += Lib::CXToken.size
				}
			end

			def self.release(pointer)
				Lib.dispose_tokens(pointer, pointer.token_size, pointer.translation_unit)
			end

			def each(&block)
				@tokens.each do |token|
					block.call(token)
				end
			end

			def cursors
				ptr = MemoryPointer.new(Lib::CXCursor, @size)
				Lib.annotate_tokens(@translation_unit, self, @size, ptr)

				cur_ptr = ptr
				arr = []
				@size.times {
					arr << Cursor.new(cur_ptr, @translation_unit)
					cur_ptr += Lib::CXCursor.size
				}
				arr
			end
		end

		class Token
			def initialize(token, translation_unit)
				@token = token
				@translation_unit = translation_unit
			end

			def kind
				Lib.get_token_kind(@token)
			end

			def spelling
				Lib.extract_string Lib.get_token_spelliing(@translation_unit, @token)
			end

			def location
				ExpansionLocation.new Lib.get_token_location(@translation_unit, @token)
			end

			def extent
				SourceRange.new Lib.get_token_extent(@translation_unit, @token)
			end
		end
	end
end