mirror of
https://github.com/vlang/v.git
synced 2025-09-13 14:32:26 +03:00
64 lines
2 KiB
V
64 lines
2 KiB
V
module iconv
|
|
|
|
// Module iconv provides functions convert between vstring(UTF8) to/from different encodings.
|
|
|
|
#flag openbsd -I/usr/local/include
|
|
|
|
#include <iconv.h>
|
|
|
|
#flag darwin -liconv
|
|
#flag freebsd -L/usr/local/lib -liconv
|
|
#flag openbsd -L/usr/local/lib -liconv
|
|
#flag termux -L/data/data/com.termux/files/usr/lib -liconv
|
|
|
|
fn C.iconv_open(tocode charptr, fromcode charptr) voidptr
|
|
fn C.iconv_close(cd voidptr) int
|
|
fn C.iconv(cd voidptr, inbuf &charptr, inbytesleft &usize, outbuf &charptr, outbytesleft &usize) usize
|
|
|
|
// conv convert `fromcode` encoding string to `tocode` encoding string
|
|
@[direct_array_access]
|
|
fn conv(tocode string, fromcode string, src &u8, src_len int) ![]u8 {
|
|
if src_len < 0 {
|
|
return error('src length error')
|
|
}
|
|
|
|
mut src_encoding := fromcode.to_upper()
|
|
mut dst_encoding := tocode.to_upper()
|
|
|
|
// As macos-12 has no UTF16LE/UTF16BE/UTF32LE/UTF32BE, change them to UTF-16LE/UTF-16BE/UTF-32LE/UTF-32BE
|
|
match src_encoding {
|
|
'UTF16LE' { src_encoding = 'UTF-16LE' }
|
|
'UTF16BE' { src_encoding = 'UTF-16BE' }
|
|
'UTF32LE' { src_encoding = 'UTF-32LE' }
|
|
'UTF32BE' { src_encoding = 'UTF-32BE' }
|
|
else {}
|
|
}
|
|
match dst_encoding {
|
|
'UTF16LE' { dst_encoding = 'UTF-16LE' }
|
|
'UTF16BE' { dst_encoding = 'UTF-16BE' }
|
|
'UTF32LE' { dst_encoding = 'UTF-32LE' }
|
|
'UTF32BE' { dst_encoding = 'UTF-32BE' }
|
|
else {}
|
|
}
|
|
|
|
mut cd := C.iconv_open(charptr(dst_encoding.str), charptr(src_encoding.str))
|
|
if isize(cd) == -1 {
|
|
return error('platform can\'t convert from ${src_encoding} to ${dst_encoding}')
|
|
}
|
|
defer { C.iconv_close(cd) }
|
|
|
|
mut dst := []u8{len: (src_len + 1) * 4} // this should be enough to hold the dst encoding string
|
|
|
|
mut src_ptr := charptr(src)
|
|
mut dst_ptr := charptr(dst.data)
|
|
mut src_left := usize(src_len)
|
|
mut dst_left := usize(dst.len)
|
|
res := C.iconv(cd, &src_ptr, &src_left, &dst_ptr, &dst_left)
|
|
if res == usize(-1) {
|
|
return error('convert encoding string fail, iconv return ${res}')
|
|
}
|
|
|
|
// resize dst buf to real length
|
|
dst.trim(dst.len - int(dst_left))
|
|
return dst
|
|
}
|