import t
import stdio
import string
import testcheck

def test_strlen():
    testcheck.section("Test 1: strlen")
    s: t.CChar | t.CPtr = "Hello, World!"
    length: t.CUInt64T = string.strlen(s)
    testcheck.check(length == 13, "strlen (len=13)", "strlen (expect 13)")

def test_strcpy():
    testcheck.section("Test 2: strcpy")
    buf: t.CChar | t.CPtr = w32.win32memory.VirtualAlloc(t.CVoid(0, t.CPtr), 64, 12288, 4)
    if t.CUInt64T(buf) == 0:
        stdio.printf("SKIP: VirtualAlloc returned NULL\n")
        return
    string.memset(buf, 0, 64)
    string.strcpy(buf, "test string")
    cmp_result: t.CInt = string.strcmp(buf, "test string")
    testcheck.check(cmp_result == 0, "strcpy (result=test string)", "strcpy (cmp mismatch)")

def test_strcat():
    testcheck.section("Test 3: strcat (via memcpy)")
    buf: t.CChar | t.CPtr = w32.win32memory.VirtualAlloc(t.CVoid(0, t.CPtr), 64, 12288, 4)
    if t.CUInt64T(buf) == 0:
        stdio.printf("SKIP: VirtualAlloc returned NULL\n")
        return
    string.memset(buf, 0, 64)
    # 手动拼接: 先拷贝 "Hello"，再拷贝 " World"
    hello: t.CChar | t.CPtr = "Hello"
    world: t.CChar | t.CPtr = " World"
    string.memcpy(buf, hello, 5)
    string.memcpy(t.CVoid(t.CUInt64T(buf) + 5, t.CPtr), world, 6)
    cmp_result: t.CInt = string.strcmp(buf, "Hello World")
    testcheck.check(cmp_result == 0, "strcat via memcpy", "strcat via memcpy (cmp mismatch)")

def test_strcmp():
    testcheck.section("Test 4: strcmp")
    r1: t.CInt = string.strcmp("abc", "abc")
    r2: t.CInt = string.strcmp("abc", "abd")
    r3: t.CInt = string.strcmp("abd", "abc")
    testcheck.check(r1 == 0 and r2 < 0 and r3 > 0, "strcmp (eq=0 lt<0 gt>0)", "strcmp (expect eq=0 lt<0 gt>0)")

def test_strncpy():
    testcheck.section("Test 5: strncpy")
    buf: t.CChar | t.CPtr = w32.win32memory.VirtualAlloc(t.CVoid(0, t.CPtr), 64, 12288, 4)
    if t.CUInt64T(buf) == 0:
        stdio.printf("SKIP: VirtualAlloc returned NULL\n")
        return
    string.memset(buf, 0, 64)
    string.strncpy(buf, "ABCDEFGHIJ", 5)
    cmp_result: t.CInt = string.strcmp(buf, "ABCDE")
    testcheck.check(cmp_result == 0, "strncpy (result=ABCDE)", "strncpy (cmp mismatch)")

def test_char_access():
    testcheck.section("Test 6: char access")
    s: t.CChar | t.CPtr = "ABCDE"
    val: t.CUInt8T = s[2]  # 'C' = 67
    testcheck.check(val == 67, "char access (s[2]=67='C')", "char access (s[2] expect 67)")

def test_snprintf():
    testcheck.section("Test 7: snprintf")
    buf: t.CChar | t.CPtr = w32.win32memory.VirtualAlloc(t.CVoid(0, t.CPtr), 128, 12288, 4)
    if t.CUInt64T(buf) == 0:
        stdio.printf("SKIP: VirtualAlloc returned NULL\n")
        return
    string.memset(buf, 0, 128)
    stdio.snprintf(buf, 128, "value=%d hex=0x%x str=%s", 42, 255, "test")
    cmp_result: t.CInt = string.strcmp(buf, "value=42 hex=0xff str=test")
    testcheck.check(cmp_result == 0, "snprintf (result=value=42 hex=0xff str=test)", "snprintf (result mismatch)")

import w32.win32memory

def main() -> t.CInt:
    testcheck.begin("StringTest: 字符串操作基准测试")
    test_strlen()
    test_strcpy()
    test_strcat()
    test_strcmp()
    test_strncpy()
    test_char_access()
    test_snprintf()
    return testcheck.end()
