package main

import (
	"fmt"
	"go/token"
)

func main() {
	testPos()
	testToken()
	testFileSet()
	testFile()
	testPosition()
	testTokenPrecedence()
	testTokenKeywords()
	testUtilityFunctions()
}

func testPos() {
	fmt.Println("=== Test Pos ===")
	
	pos1 := token.Pos(100)
	pos2 := token.Pos(200)
	
	if pos1 != 100 {
		panic(fmt.Sprintf("Expected pos1 to be 100, got %d", pos1))
	}
	if pos2 != 200 {
		panic(fmt.Sprintf("Expected pos2 to be 200, got %d", pos2))
	}
	fmt.Printf("Pos1: %d, Pos2: %d\n", pos1, pos2)
	
	if !pos1.IsValid() {
		panic("Expected pos1.IsValid() to be true")
	}
	fmt.Printf("Pos1.IsValid(): %v\n", pos1.IsValid())
	
	noPos := token.NoPos
	if noPos != 0 {
		panic(fmt.Sprintf("Expected NoPos to be 0, got %d", noPos))
	}
	if noPos.IsValid() {
		panic("Expected NoPos.IsValid() to be false")
	}
	fmt.Printf("NoPos: %d, IsValid: %v\n", noPos, noPos.IsValid())
	
	fmt.Println("SUCCESS: Pos operations work correctly\n")
}

func testToken() {
	fmt.Println("\n=== Test Token Types ===")
	
	expectedStrings := map[token.Token]string{
		token.ADD:    "+",
		token.SUB:    "-",
		token.MUL:    "*",
		token.QUO:    "/",
		token.LPAREN: "(",
		token.RPAREN: ")",
		token.EQL:    "==",
		token.NEQ:    "!=",
	}
	
	for tok, expected := range expectedStrings {
		if tok.String() != expected {
			panic(fmt.Sprintf("Expected %v.String() to be %q, got %q", tok, expected, tok.String()))
		}
	}
	
	tokens := []token.Token{
		token.ILLEGAL,
		token.EOF,
		token.COMMENT,
		token.IDENT,
		token.INT,
		token.FLOAT,
		token.IMAG,
		token.CHAR,
		token.STRING,
		token.ADD,
		token.SUB,
		token.MUL,
		token.QUO,
		token.REM,
		token.AND,
		token.OR,
		token.XOR,
		token.SHL,
		token.SHR,
		token.AND_NOT,
		token.ADD_ASSIGN,
		token.SUB_ASSIGN,
		token.MUL_ASSIGN,
		token.QUO_ASSIGN,
		token.REM_ASSIGN,
		token.AND_ASSIGN,
		token.OR_ASSIGN,
		token.XOR_ASSIGN,
		token.SHL_ASSIGN,
		token.SHR_ASSIGN,
		token.AND_NOT_ASSIGN,
		token.LAND,
		token.LOR,
		token.ARROW,
		token.INC,
		token.DEC,
		token.EQL,
		token.LSS,
		token.GTR,
		token.ASSIGN,
		token.NOT,
		token.NEQ,
		token.LEQ,
		token.GEQ,
		token.DEFINE,
		token.ELLIPSIS,
		token.LPAREN,
		token.LBRACK,
		token.LBRACE,
		token.COMMA,
		token.PERIOD,
		token.RPAREN,
		token.RBRACK,
		token.RBRACE,
		token.SEMICOLON,
		token.COLON,
	}
	
	for _, tok := range tokens {
		fmt.Printf("Token: %s (String: %q)\n", tok, tok.String())
	}
	
	fmt.Println("SUCCESS: Token types work correctly\n")
}

func testTokenKeywords() {
	fmt.Println("\n=== Test Keywords ===")
	
	keywords := []token.Token{
		token.BREAK,
		token.CASE,
		token.CHAN,
		token.CONST,
		token.CONTINUE,
		token.DEFAULT,
		token.DEFER,
		token.ELSE,
		token.FALLTHROUGH,
		token.FOR,
		token.FUNC,
		token.GO,
		token.GOTO,
		token.IF,
		token.IMPORT,
		token.INTERFACE,
		token.MAP,
		token.PACKAGE,
		token.RANGE,
		token.RETURN,
		token.SELECT,
		token.STRUCT,
		token.SWITCH,
		token.TYPE,
		token.VAR,
	}
	
	for _, kw := range keywords {
		if !kw.IsKeyword() {
			panic(fmt.Sprintf("Expected %s to be a keyword", kw))
		}
		fmt.Printf("Keyword: %s, IsKeyword: %v\n", kw, kw.IsKeyword())
	}
	
	if token.ADD.IsKeyword() {
		panic("Expected ADD operator to not be a keyword")
	}
	if token.IDENT.IsKeyword() {
		panic("Expected IDENT token to not be a keyword")
	}
	
	fmt.Println("SUCCESS: Keyword checks work correctly\n")
}

func testTokenPrecedence() {
	fmt.Println("\n=== Test Token Precedence ===")
	
	if token.MUL.Precedence() <= token.ADD.Precedence() {
		panic("Expected MUL to have higher precedence than ADD")
	}
	if token.LAND.Precedence() <= token.LOR.Precedence() {
		panic("Expected LAND to have higher precedence than LOR")
	}
	if token.MUL.Precedence() != token.QUO.Precedence() {
		panic("Expected MUL and QUO to have same precedence")
	}
	if token.ADD.Precedence() != token.SUB.Precedence() {
		panic("Expected ADD and SUB to have same precedence")
	}
	
	operators := []token.Token{
		token.ADD,
		token.SUB,
		token.MUL,
		token.QUO,
		token.REM,
		token.LAND,
		token.LOR,
		token.EQL,
		token.LSS,
		token.GTR,
	}
	
	for _, op := range operators {
		fmt.Printf("Operator: %s, Precedence: %d\n", op, op.Precedence())
	}
	
	fmt.Println("SUCCESS: Precedence operations work correctly\n")
}

func testFileSet() {
	fmt.Println("\n=== Test FileSet ===")
	
	fset := token.NewFileSet()
	
	file1 := fset.AddFile("file1.go", -1, 1000)
	file2 := fset.AddFile("file2.go", -1, 2000)
	
	fmt.Printf("Added file1: %s, Base: %d, Size: %d\n", file1.Name(), file1.Base(), file1.Size())
	fmt.Printf("Added file2: %s, Base: %d, Size: %d\n", file2.Name(), file2.Base(), file2.Size())
	
	pos := file1.Pos(100)
	retrievedFile := fset.File(pos)
	if retrievedFile != file1 {
		panic("FileSet.File failed to retrieve correct file")
	}
	
	position := fset.Position(pos)
	fmt.Printf("Position at offset 100: %s\n", position)
	
	fmt.Println("SUCCESS: FileSet operations work correctly\n")
}

func testFile() {
	fmt.Println("\n=== Test File ===")
	
	fset := token.NewFileSet()
	file := fset.AddFile("test.go", -1, 1000)
	
	if file.Name() != "test.go" {
		panic(fmt.Sprintf("Expected file name to be 'test.go', got %q", file.Name()))
	}
	if file.Size() != 1000 {
		panic(fmt.Sprintf("Expected file size to be 1000, got %d", file.Size()))
	}
	
	file.AddLine(0)
	file.AddLine(50)
	file.AddLine(100)
	
	if file.LineCount() != 3 {
		panic(fmt.Sprintf("Expected line count to be 3, got %d", file.LineCount()))
	}
	
	fmt.Printf("File name: %s\n", file.Name())
	fmt.Printf("File base: %d\n", file.Base())
	fmt.Printf("File size: %d\n", file.Size())
	fmt.Printf("File line count: %d\n", file.LineCount())
	
	pos := file.Pos(50)
	fmt.Printf("Pos at offset 50: %d\n", pos)
	
	offset := file.Offset(pos)
	if offset != 50 {
		panic(fmt.Sprintf("Expected offset to be 50, got %d", offset))
	}
	fmt.Printf("Offset of pos: %d\n", offset)
	
	line := file.Line(pos)
	if line != 2 {
		panic(fmt.Sprintf("Expected line to be 2, got %d", line))
	}
	fmt.Printf("Line number at pos: %d\n", line)
	
	lineStart := file.LineStart(2)
	fmt.Printf("Line 2 starts at pos: %d\n", lineStart)
	
	position := file.Position(pos)
	fmt.Printf("Position: %s\n", position)
	
	fmt.Println("SUCCESS: File operations work correctly\n")
}

func testPosition() {
	fmt.Println("\n=== Test Position ===" )
	
	pos := token.Position{
		Filename: "test.go",
		Offset:   100,
		Line:     5,
		Column:   10,
	}
	
	if !pos.IsValid() {
		panic("Expected valid position to be valid")
	}
	if pos.Filename != "test.go" {
		panic(fmt.Sprintf("Expected filename to be 'test.go', got %q", pos.Filename))
	}
	if pos.Line != 5 {
		panic(fmt.Sprintf("Expected line to be 5, got %d", pos.Line))
	}
	if pos.Column != 10 {
		panic(fmt.Sprintf("Expected column to be 10, got %d", pos.Column))
	}
	
	fmt.Printf("Position: %s\n", pos.String())
	fmt.Printf("Filename: %s, Line: %d, Column: %d, Offset: %d\n", 
		pos.Filename, pos.Line, pos.Column, pos.Offset)
	fmt.Printf("IsValid: %v\n", pos.IsValid())
	
	invalidPos := token.Position{}
	if invalidPos.IsValid() {
		panic("Expected empty position to be invalid")
	}
	fmt.Printf("Invalid position IsValid: %v\n", invalidPos.IsValid())
	
	fmt.Println("SUCCESS: Position operations work correctly\n")
}

func testUtilityFunctions() {
	fmt.Println("\n=== Test Utility Functions ===")

	fmt.Printf("IsExported(\"Foo\"): %v\n", token.IsExported("Foo"))
	fmt.Printf("IsExported(\"foo\"): %v\n", token.IsExported("foo"))
	fmt.Printf("IsExported(\"_foo\"): %v\n", token.IsExported("_foo"))

	fmt.Printf("IsIdentifier(\"foo\"): %v\n", token.IsIdentifier("foo"))
	fmt.Printf("IsIdentifier(\"foo123\"): %v\n", token.IsIdentifier("foo123"))
	fmt.Printf("IsIdentifier(\"123foo\"): %v\n", token.IsIdentifier("123foo"))
	fmt.Printf("IsIdentifier(\"foo-bar\"): %v\n", token.IsIdentifier("foo-bar"))

	fmt.Printf("IsKeyword(\"func\"): %v\n", token.IsKeyword("func"))
	fmt.Printf("IsKeyword(\"if\"): %v\n", token.IsKeyword("if"))
	fmt.Printf("IsKeyword(\"foo\"): %v\n", token.IsKeyword("foo"))

	lookupFunc := token.Lookup("func")
	fmt.Printf("Lookup(\"func\"): %s\n", lookupFunc)

	lookupIdent := token.Lookup("myVar")
	fmt.Printf("Lookup(\"myVar\"): %s\n", lookupIdent)

	lookupFor := token.Lookup("for")
	fmt.Printf("Lookup(\"for\"): %s\n", lookupFor)

	fmt.Println("SUCCESS: Utility functions work correctly\n")
}