mirror of
https://github.com/crazy-max/diun.git
synced 2024-12-22 19:38:28 +00:00
91 lines
2.7 KiB
Go
91 lines
2.7 KiB
Go
package uniseg
|
|
|
|
import "unicode/utf8"
|
|
|
|
// FirstSentence returns the first sentence found in the given byte slice
|
|
// according to the rules of [Unicode Standard Annex #29, Sentence Boundaries].
|
|
// This function can be called continuously to extract all sentences from a byte
|
|
// slice, as illustrated in the example below.
|
|
//
|
|
// If you don't know the current state, for example when calling the function
|
|
// for the first time, you must pass -1. For consecutive calls, pass the state
|
|
// and rest slice returned by the previous call.
|
|
//
|
|
// The "rest" slice is the sub-slice of the original byte slice "b" starting
|
|
// after the last byte of the identified sentence. If the length of the "rest"
|
|
// slice is 0, the entire byte slice "b" has been processed. The "sentence" byte
|
|
// slice is the sub-slice of the input slice containing the identified sentence.
|
|
//
|
|
// Given an empty byte slice "b", the function returns nil values.
|
|
//
|
|
// [Unicode Standard Annex #29, Sentence Boundaries]: http://unicode.org/reports/tr29/#Sentence_Boundaries
|
|
func FirstSentence(b []byte, state int) (sentence, rest []byte, newState int) {
|
|
// An empty byte slice returns nothing.
|
|
if len(b) == 0 {
|
|
return
|
|
}
|
|
|
|
// Extract the first rune.
|
|
r, length := utf8.DecodeRune(b)
|
|
if len(b) <= length { // If we're already past the end, there is nothing else to parse.
|
|
return b, nil, sbAny
|
|
}
|
|
|
|
// If we don't know the state, determine it now.
|
|
if state < 0 {
|
|
state, _ = transitionSentenceBreakState(state, r, b[length:], "")
|
|
}
|
|
|
|
// Transition until we find a boundary.
|
|
var boundary bool
|
|
for {
|
|
r, l := utf8.DecodeRune(b[length:])
|
|
state, boundary = transitionSentenceBreakState(state, r, b[length+l:], "")
|
|
|
|
if boundary {
|
|
return b[:length], b[length:], state
|
|
}
|
|
|
|
length += l
|
|
if len(b) <= length {
|
|
return b, nil, sbAny
|
|
}
|
|
}
|
|
}
|
|
|
|
// FirstSentenceInString is like [FirstSentence] but its input and outputs are
|
|
// strings.
|
|
func FirstSentenceInString(str string, state int) (sentence, rest string, newState int) {
|
|
// An empty byte slice returns nothing.
|
|
if len(str) == 0 {
|
|
return
|
|
}
|
|
|
|
// Extract the first rune.
|
|
r, length := utf8.DecodeRuneInString(str)
|
|
if len(str) <= length { // If we're already past the end, there is nothing else to parse.
|
|
return str, "", sbAny
|
|
}
|
|
|
|
// If we don't know the state, determine it now.
|
|
if state < 0 {
|
|
state, _ = transitionSentenceBreakState(state, r, nil, str[length:])
|
|
}
|
|
|
|
// Transition until we find a boundary.
|
|
var boundary bool
|
|
for {
|
|
r, l := utf8.DecodeRuneInString(str[length:])
|
|
state, boundary = transitionSentenceBreakState(state, r, nil, str[length+l:])
|
|
|
|
if boundary {
|
|
return str[:length], str[length:], state
|
|
}
|
|
|
|
length += l
|
|
if len(str) <= length {
|
|
return str, "", sbAny
|
|
}
|
|
}
|
|
}
|