v/vlib/strings/builder.js.v

131 lines
2.2 KiB
V
Raw Normal View History

2022-01-04 10:21:08 +01:00
// Copyright (c) 2019-2022 Alexander Medvednikov. All rights reserved.
2019-06-23 04:21:30 +02:00
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.
2019-09-14 22:48:30 +02:00
module strings
2019-06-22 20:20:28 +02:00
/*
2019-10-24 11:47:21 +02:00
pub struct Builder {
mut:
2022-04-15 14:35:35 +02:00
buf []u8
2019-12-16 16:54:44 +01:00
pub mut:
2020-12-21 21:00:32 +01:00
len int
initial_size int = 1
}*/
2022-04-15 13:10:11 +02:00
pub type Builder = []u8
2019-06-22 20:20:28 +02:00
pub fn new_builder(initial_size int) Builder {
2022-04-15 13:10:11 +02:00
return []u8{cap: initial_size}
2019-06-22 20:20:28 +02:00
}
2022-04-15 13:58:56 +02:00
[deprecated: 'Use write_u8() instead']
2022-04-15 13:10:11 +02:00
pub fn (mut b Builder) write_b(data u8) {
b << data
2019-12-06 21:02:09 +01:00
}
pub fn (mut b Builder) write_byte(data byte) {
b << data
}
2022-04-15 13:58:56 +02:00
pub fn (mut b Builder) write_u8(data u8) {
b << data
}
2022-04-15 13:10:11 +02:00
pub fn (mut b Builder) write(data []u8) ?int {
2021-08-31 16:10:19 +02:00
if data.len == 0 {
return 0
}
b << data
2021-08-31 16:10:19 +02:00
return data.len
2019-06-22 20:20:28 +02:00
}
2022-04-15 13:10:11 +02:00
pub fn (b &Builder) byte_at(n int) u8 {
unsafe {
return b[n]
}
2019-06-22 20:20:28 +02:00
}
2021-08-31 16:10:19 +02:00
pub fn (mut b Builder) write_string(s string) {
if s.len == 0 {
return
}
for c in s {
b << c
2021-08-31 16:10:19 +02:00
}
2019-06-22 20:20:28 +02:00
}
2021-08-31 16:10:19 +02:00
pub fn (mut b Builder) writeln(s string) {
if s.len > 0 {
b.write_string(s)
}
b << 10
2019-06-22 20:20:28 +02:00
}
2021-08-31 16:10:19 +02:00
pub fn (mut b Builder) str() string {
s := ''
#for (const c of b.val.arr.arr)
2021-08-31 16:10:19 +02:00
#s.str += String.fromCharCode(+c)
b.trim(0)
2021-08-31 16:10:19 +02:00
return s
2019-06-24 22:34:21 +02:00
}
pub fn (mut b Builder) cut_last(n int) string {
cut_pos := b.len - n
x := b[cut_pos..]
res := x.bytestr()
b.trim(cut_pos)
return res
}
pub fn (mut b Builder) go_back_to(pos int) {
b.trim(pos)
}
// go_back discards the last `n` bytes from the buffer
pub fn (mut b Builder) go_back(n int) {
b.trim(b.len - n)
}
// cut_to cuts the string after `pos` and returns it.
// if `pos` is superior to builder length, returns an empty string
// and cancel further operations
pub fn (mut b Builder) cut_to(pos int) string {
if pos > b.len {
return ''
}
return b.cut_last(b.len - pos)
}
pub fn (mut b Builder) write_runes(runes []rune) {
for r in runes {
res := r.str()
#res.str = String.fromCharCode(r.val)
b << res.bytes()
}
}
// after(6) returns 'world'
// buf == 'hello world'
pub fn (mut b Builder) after(n int) string {
if n >= b.len {
return ''
}
x := b.slice(n, b.len)
return x.bytestr()
}
// last_n(5) returns 'world'
// buf == 'hello world'
pub fn (b &Builder) last_n(n int) string {
if n >= b.len {
return ''
}
x := b.slice(b.len - n, b.len)
return x.bytestr()
}