Fix Truncate

TruncateWordsToWholeSentence knows if the summary is truncated, so let "him" decide.

Fixes #880
This commit is contained in:
bep 2015-02-05 21:44:15 +01:00 committed by Anthony Fok
parent 2bee4a1570
commit 48b6777ea2
3 changed files with 39 additions and 10 deletions

View file

@ -271,10 +271,11 @@ func TruncateWords(s string, max int) string {
} }
// TruncateWordsToWholeSentence takes content and an int // TruncateWordsToWholeSentence takes content and an int
// and returns entire sentences from content, delimited by the int. // and returns entire sentences from content, delimited by the int
func TruncateWordsToWholeSentence(words []string, max int) string { // and whether it's truncated or not.
if max > len(words) { func TruncateWordsToWholeSentence(words []string, max int) (string, bool) {
return strings.Join(words, " ") if max >= len(words) {
return strings.Join(words, " "), false
} }
for counter, word := range words[max:] { for counter, word := range words[max:] {
@ -282,11 +283,12 @@ func TruncateWordsToWholeSentence(words []string, max int) string {
strings.HasSuffix(word, "?") || strings.HasSuffix(word, "?") ||
strings.HasSuffix(word, ".\"") || strings.HasSuffix(word, ".\"") ||
strings.HasSuffix(word, "!") { strings.HasSuffix(word, "!") {
return strings.Join(words[:max+counter+1], " ") upper := max + counter + 1
return strings.Join(words[:upper], " "), (upper < len(words))
} }
} }
return strings.Join(words[:max], " ") return strings.Join(words[:max], " "), true
} }
// GetRstContent calls the Python script rst2html as an external helper // GetRstContent calls the Python script rst2html as an external helper

View file

@ -3,6 +3,7 @@ package helpers
import ( import (
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
"html/template" "html/template"
"strings"
"testing" "testing"
) )
@ -33,3 +34,30 @@ func TestStripEmptyNav(t *testing.T) {
func TestBytesToHTML(t *testing.T) { func TestBytesToHTML(t *testing.T) {
assert.Equal(t, template.HTML("dobedobedo"), BytesToHTML([]byte("dobedobedo"))) assert.Equal(t, template.HTML("dobedobedo"), BytesToHTML([]byte("dobedobedo")))
} }
func TestTruncateWordsToWholeSentence(t *testing.T) {
type test struct {
input, expected string
max int
truncated bool
}
data := []test{
{"a b c", "a b c", 12, false},
{"a b c", "a b c", 3, false},
{"a", "a", 1, false},
{"This is a sentence.", "This is a sentence.", 5, false},
{"This is also a sentence!", "This is also a sentence!", 1, false},
{"To be. Or not to be. That's the question.", "To be.", 1, true},
{" \nThis is not a sentence\n ", "This is not a", 4, true},
}
for i, d := range data {
output, truncated := TruncateWordsToWholeSentence(strings.Fields(d.input), d.max)
if d.expected != output {
t.Errorf("Test %d failed. Expected %q got %q", i, d.expected, output)
}
if d.truncated != truncated {
t.Errorf("Test %d failed. Expected truncated=%t got %t", i, d.truncated, truncated)
}
}
}

View file

@ -190,10 +190,9 @@ func (p *Page) setSummary() {
} else { } else {
// If hugo defines split: // If hugo defines split:
// render, strip html, then split // render, strip html, then split
p.Summary = helpers.BytesToHTML([]byte(helpers.TruncateWordsToWholeSentence(p.PlainWords(), helpers.SummaryLength))) summary, truncated := helpers.TruncateWordsToWholeSentence(p.PlainWords(), helpers.SummaryLength)
p.Summary = helpers.BytesToHTML([]byte(summary))
// todo bep - check if the Plain() can be trimmed earlier p.Truncated = truncated
p.Truncated = len(p.Summary) != len(strings.Trim(p.Plain(), "\n\r "))
} }
} }