aboutsummaryrefslogtreecommitdiff
path: root/internal/reader/readingtime/readingtime.go
blob: 0cfb2b22d055ac3d5e5280b1c7ae0bd79ef3a789 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
// SPDX-FileCopyrightText: Copyright The Miniflux Authors. All rights reserved.
// SPDX-License-Identifier: Apache-2.0

// Package readingtime provides a function to estimate the reading time of an article.
package readingtime

import (
	"math"
	"strings"
	"unicode/utf8"

	"miniflux.app/v2/internal/reader/sanitizer"

	"github.com/abadojack/whatlanggo"
)

// EstimateReadingTime returns the estimated reading time of an article in minute.
func EstimateReadingTime(content string, defaultReadingSpeed, cjkReadingSpeed int) int {
	sanitizedContent := sanitizer.StripTags(content)

	// Litterature on language detection says that around 100 signes is enough, we're safe here.
	truncationPoint := int(math.Min(float64(len(sanitizedContent)), 250))

	// We're only interested in identifying Japanse/Chinese/Korean
	options := whatlanggo.Options{
		Whitelist: map[whatlanggo.Lang]bool{
			whatlanggo.Jpn: true,
			whatlanggo.Cmn: true,
			whatlanggo.Kor: true,
		},
	}
	langInfo := whatlanggo.DetectWithOptions(sanitizedContent[:truncationPoint], options)

	if langInfo.IsReliable() {
		return int(math.Ceil(float64(utf8.RuneCountInString(sanitizedContent)) / float64(cjkReadingSpeed)))
	}
	nbOfWords := len(strings.Fields(sanitizedContent))
	return int(math.Ceil(float64(nbOfWords) / float64(defaultReadingSpeed)))
}