# 基本过程
proc greet(name: string) =
  echo "Hello, ", name, "!"

greet("Nim")  # Hello, Nim!

# 带返回值的过程
proc add(a, b: int): int =
  return a + b

echo add(3, 4)  # 7

# 使用 result 变量（隐式返回值）
proc factorial(n: int): int =
  result = 1
  for i in 2..n:
    result *= i

echo factorial(5)  # 120

proc buildString(): string =
  result = ""
  result.add("Hello")
  result.add(", ")
  result.add("World!")

echo buildString()  # Hello, World!

# 也可以显式 return 提前返回
proc findPositive(data: seq[int]): int =
  for x in data:
    if x > 0:
      return x
  result = -1  # 未找到

# 默认参数（只读）
proc printValue(x: int) =
  echo x
  # x = 10  # ❌ 不能修改

# var 参数（可修改原值）
proc doubleIt(x: var int) =
  x *= 2

var n = 10
doubleIt(n)
echo n  # 20

# 多个 var 参数
proc swap(a, b: var int) =
  let temp = a
  a = b
  b = temp

var (x, y) = (1, 2)
swap(x, y)
echo x, y  # 21

proc divmod(a, b: int): (int, int) =
  (a div b, a mod b)

let (quotient, remainder) = divmod(17, 5)
echo &"17 / 5 = {quotient} 余 {remainder}"
# 17 / 5 = 3 余 2

# 忽略不需要的返回值
let (q, _) = divmod(17, 5)
echo q  # 3

proc getUserInfo(id: int): tuple[name: string, age: int, email: string] =
  ("张三", 28, "[email protected]")

let user = getUserInfo(1)
echo user.name   # 张三
echo user.age    # 28
echo user.email  # [email protected]

# 解构
let (name, age, email) = getUserInfo(1)
echo &"{name}, {age}岁, {email}"

# 使用 var 参数作为输出
proc parseTime(seconds: int, outMinutes, outSeconds: var int) =
  outMinutes = seconds div 60
  outSeconds = seconds mod 60

var m, s: int
parseTime(125, m, s)
echo &"{m}分{s}秒"  # 2分5秒

# 默认参数
proc createWindow(title: string, width = 800, height = 600, visible = true) =
  echo &"Window '{title}': {width}x{height}, visible={visible}"

createWindow("My App")
# Window 'My App': 800x600, visible=true

createWindow("My App", 1024, 768)
# Window 'My App': 1024x768, visible=true

# 命名参数（顺序无关）
createWindow("My App", height = 900, width = 1600)
# Window 'My App': 1600x900, visible=true

# 混合使用
createWindow("My App", 1024, visible = false)
# Window 'My App': 1024x600, visible=false

# 基本泛型
proc max[T](a, b: T): T =
  if a > b: a else: b

echo max(10, 20)       # 20
echo max(3.14, 2.71)   # 3.14
echo max("abc", "def") # def

# 多个类型参数
proc pair[A, B](a: A, b: B): (A, B) =
  (a, b)

let p = pair(42, "hello")
echo p  # (42, "hello")

# 泛型约束
proc sum[T: SomeNumber](values: seq[T]): T =
  result = T(0)
  for v in values:
    result += v

echo sum(@[1, 2, 3, 4, 5])       # 15
echo sum(@[1.1, 2.2, 3.3])       # 6.6

# 概念约束（Concepts）
type
  Container[T] = concept c
    c.len is int
    c[int] is T

proc firstElement[T](c: Container[T]): T =
  c[0]

echo firstElement(@[10, 20, 30])    # 10
echo firstElement(@["a", "b", "c"]) # "a"

# 重载基于参数类型
proc process(x: int) =
  echo "Processing int: ", x

proc process(x: float) =
  echo "Processing float: ", x

proc process(x: string) =
  echo "Processing string: ", x

process(42)       # Processing int: 42
process(3.14)     # Processing float: 3.14
process("hello")  # Processing string: hello

# 重载基于参数数量
proc log(msg: string) =
  echo "[INFO] ", msg

proc log(msg: string, level: int) =
  echo "[LEVEL ", level, "] ", msg

log("System started")
log("Critical error", 5)

# 过程类型
type
  IntProc = proc(x: int): int

# 闭包——捕获外部变量
proc makeCounter(): proc(): int =
  var count = 0
  result = proc(): int =
    count += 1
    return count

let counter = makeCounter()
echo counter()  # 1
echo counter()  # 2
echo counter()  # 3

# 闭包作为参数
proc apply(data: seq[int], fn: proc(x: int): int): seq[int] =
  result = newSeq[int](data.len)
  for i, v in data:
    result[i] = fn(v)

let nums = @[1, 2, 3, 4, 5]
let doubled = apply(nums, proc(x: int): int = x * 2)
let squared = apply(nums, proc(x: int): int = x * x)
echo doubled  # @[2, 4, 6, 8, 10]
echo squared  # @[1, 4, 9, 16, 25]

# do 表示法（更简洁的闭包语法）
let tripled = nums.apply() do (x: int) -> int:
  x * 3
echo tripled  # @[3, 6, 9, 12, 15]

type
  Shape = ref object of RootObj
  Circle = ref object of Shape
    radius: float
  Rectangle = ref object of Shape
    width, height: float

method area(s: Shape): float {.base.} =
  raise newException(CatchableError, "Not implemented")

method area(c: Circle): float =
  PI * c.radius * c.radius

method area(r: Rectangle): float =
  r.width * r.height

let shapes: seq[Shape] = @[
  Circle(radius: 5.0),
  Rectangle(width: 4.0, height: 6.0),
]

for s in shapes:
  echo s.area()
# 78.53981633974483
# 24.0

type Vec2 = object
  x, y: float

proc `+`(a, b: Vec2): Vec2 =
  Vec2(x: a.x + b.x, y: a.y + b.y)

proc `-`(a, b: Vec2): Vec2 =
  Vec2(x: a.x - b.x, y: a.y - b.y)

proc `*`(v: Vec2, s: float): Vec2 =
  Vec2(x: v.x * s, y: v.y * s)

proc `$`(v: Vec2): string =
  &"({v.x}, {v.y})"

proc dot(a, b: Vec2): float =
  a.x * b.x + a.y * b.y

let a = Vec2(x: 1, y: 2)
let b = Vec2(x: 3, y: 4)
echo a + b       # (4.0, 6.0)
echo a * 2.0     # (2.0, 4.0)

# func 不允许修改外部状态
func square(x: int): int =
  x * x

func abs(x: int): int =
  if x < 0: -x else: x

echo square(5)   # 25
echo abs(-42)    # 42

# ❌ 编译错误：func 不能有副作用
# func bad(x: int): int =
#   echo x       # echo 是副作用
#   return x

type
  UserProfile* = object
    name*: string
    age*: int

const MaxRetries* = 3

proc createUser*(name: string, age: int): UserProfile =
  UserProfile(name: name, age: age)

proc displayName*(user: UserProfile) =
  echo user.name

type
  DiscountStrategy = proc(price: float): float

proc noDiscount(price: float): float = price
proc percentageDiscount(pct: float): DiscountStrategy =
  result = proc(price: float): float = price * (1 - pct / 100)

proc fixedDiscount(amount: float): DiscountStrategy =
  result = proc(price: float): float = max(0.0, price - amount)

proc calculateTotal(items: seq[float], strategy: DiscountStrategy): float =
  result = 0
  for price in items:
    result += strategy(price)

let cart = @[29.99, 49.99, 9.99, 149.99]
echo &"原价: ${calculateTotal(cart, noDiscount):.2f}"
echo &"八折: ${calculateTotal(cart, percentageDiscount(20)):.2f}"
echo &"减30: ${calculateTotal(cart, fixedDiscount(30)):.2f}"

type
  EventHandler = proc(event: string, data: string)

  EventBus = object
    handlers: seq[tuple[event: string, handler: EventHandler]]

proc newEventBus(): EventBus =
  EventBus(handlers: @[])

proc on(bus: var EventBus, event: string, handler: EventHandler) =
  bus.handlers.add((event, handler))

proc emit(bus: EventBus, event: string, data: string) =
  for (evt, handler) in bus.handlers:
    if evt == event:
      handler(event, data)

var bus = newEventBus()
bus.on("user.login", proc(event, data: string) =
  echo &"用户登录: {data}")
bus.on("user.login", proc(event, data: string) =
  echo &"记录日志: {event} - {data}")
bus.on("order.create", proc(event, data: string) =
  echo &"新订单: {data}")

bus.emit("user.login", "张三")
bus.emit("order.create", "订单#12345")