1
0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00
v/vlib/strings/builder.c.v

93 lines
1.6 KiB
V
Raw Normal View History

2020-02-03 07:00:36 +03:00
// Copyright (c) 2019-2020 Alexander Medvednikov. All rights reserved.
2019-06-23 05:21:30 +03:00
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.
2019-09-14 23:48:30 +03:00
module strings
2019-06-22 21:20:28 +03:00
2019-10-24 12:47:21 +03:00
pub struct Builder {
2019-09-14 23:48:30 +03:00
mut:
2019-12-20 00:29:37 +03:00
buf []byte
pub mut:
2019-12-20 00:29:37 +03:00
len int
initial_size int = 1
2019-06-22 21:20:28 +03:00
}
pub fn new_builder(initial_size int) Builder {
2019-12-20 00:29:37 +03:00
return Builder{
buf: make(0, initial_size, 1)
initial_size: initial_size
2019-06-22 21:20:28 +03:00
}
}
2019-12-11 19:20:46 +03:00
pub fn (b mut Builder) write_bytes(bytes byteptr, howmany int) {
b.buf.push_many(bytes, howmany)
b.len += howmany
}
2019-12-06 23:02:09 +03:00
pub fn (b mut Builder) write_b(data byte) {
b.buf << data
2019-12-08 14:11:19 +03:00
b.len++
2019-12-06 23:02:09 +03:00
}
pub fn (b mut Builder) write(s string) {
2019-12-24 20:54:43 +03:00
if s == '' {
return
}
2019-10-10 01:15:19 +03:00
b.buf.push_many(s.str, s.len)
2019-12-20 00:29:37 +03:00
// for c in s {
// b.buf << c
// }
// b.buf << []byte(s) // TODO
2019-06-22 21:20:28 +03:00
b.len += s.len
}
2020-03-11 01:21:26 +03:00
pub fn (b mut Builder) go_back(n int) {
b.buf.trim(b.buf.len-n)
b.len -= n
}
2020-03-11 01:21:26 +03:00
pub fn (b mut Builder) writeln(s string) {
2019-12-20 00:29:37 +03:00
// for c in s {
// b.buf << c
// }
2019-10-10 01:15:19 +03:00
b.buf.push_many(s.str, s.len)
2019-12-20 00:29:37 +03:00
// b.buf << []byte(s) // TODO
2019-06-22 21:20:28 +03:00
b.buf << `\n`
b.len += s.len + 1
}
2020-03-21 09:01:06 +03:00
// buf == 'hello world'
// last_n(5) returns 'world'
2020-03-21 09:04:53 +03:00
pub fn (b &Builder) last_n(n int) string {
2020-03-21 09:01:06 +03:00
if n > b.len {
return ''
}
buf := b.buf[b.len-n..]
return string(buf.clone())
}
// buf == 'hello world'
// after(6) returns 'world'
2020-03-21 09:04:53 +03:00
pub fn (b &Builder) after(n int) string {
2020-03-21 09:01:06 +03:00
if n >= b.len {
return ''
}
buf := b.buf[n..]
2020-03-21 11:29:16 +03:00
mut copy := buf.clone()
copy << `\0`
return string(copy)
2020-03-21 09:01:06 +03:00
}
2019-11-17 02:22:43 +03:00
pub fn (b mut Builder) str() string {
b.buf << `\0`
2019-12-20 00:29:37 +03:00
return string(b.buf,b.len)
2019-06-22 21:20:28 +03:00
}
pub fn (b mut Builder) free() {
2019-12-20 00:29:37 +03:00
unsafe{
free(b.buf.data)
}
b.buf = make(0, b.initial_size, 1)
b.len = 0
2019-06-24 23:34:21 +03:00
}
2019-12-20 00:29:37 +03:00