From 27c106863d39d1e45455b3461607571d33859018 Mon Sep 17 00:00:00 2001 From: Michael Mayer Date: Sun, 25 Apr 2021 14:17:34 +0200 Subject: [PATCH] Metadata: Merge existing keywords #1153 --- internal/meta/data.go | 2 +- internal/meta/exif.go | 5 +-- internal/meta/exif_test.go | 14 +++---- internal/meta/json_exiftool.go | 39 ++++++++++++++++-- internal/meta/json_test.go | 12 +++--- internal/meta/keywords.go | 37 ++++++++++------- internal/meta/keywords_test.go | 30 +++++++------- internal/photoprism/index_mediafile.go | 8 ++-- internal/photoprism/index_mediafile_test.go | 24 ++++++----- internal/photoprism/mediafile_meta_test.go | 4 +- .../testdata/flash.jpg} | Bin .../testdata/flash.json} | 4 +- pkg/txt/words.go | 13 +++++- 13 files changed, 121 insertions(+), 71 deletions(-) rename internal/{meta/testdata/exifWithFlashDescriptionProjectionKeywordsSmall.jpg => photoprism/testdata/flash.jpg} (100%) rename internal/{meta/testdata/exifWithFlashDescriptionProjectionKeywordsSmall.json => photoprism/testdata/flash.json} (98%) diff --git a/internal/meta/data.go b/internal/meta/data.go index b10ca35cc..2f89f6930 100644 --- a/internal/meta/data.go +++ b/internal/meta/data.go @@ -20,7 +20,7 @@ type Data struct { Codec string `meta:"CompressorID,Compression,FileType"` Title string `meta:"Title"` Subject string `meta:"Subject,PersonInImage,ObjectName,HierarchicalSubject,CatalogSets"` - Keywords string `meta:"Keywords"` + Keywords Keywords `meta:"Keywords"` Notes string `meta:"-"` Artist string `meta:"Artist,Creator,OwnerName"` Description string `meta:"Description"` diff --git a/internal/meta/exif.go b/internal/meta/exif.go index 5b15a0e13..fe3c87e4e 100644 --- a/internal/meta/exif.go +++ b/internal/meta/exif.go @@ -270,7 +270,7 @@ func (data *Data) Exif(fileName string, fileType fs.FileFormat) (err error) { if value, ok := tags["Flash"]; ok { if i, err := strconv.Atoi(value); err == nil && i&1 == 1 { - data.AddKeyword(KeywordFlash) + data.AddKeywords(KeywordFlash) data.Flash = true } } @@ -281,11 +281,10 @@ func (data *Data) Exif(fileName string, fileType fs.FileFormat) (err error) { } if value, ok := tags["ProjectionType"]; ok { - data.AddKeyword(KeywordPanorama) + data.AddKeywords(KeywordPanorama) data.Projection = SanitizeString(value) } - data.Keywords = SanitizeMeta(data.Keywords) data.Subject = SanitizeMeta(data.Subject) data.Artist = SanitizeMeta(data.Artist) diff --git a/internal/meta/exif_test.go b/internal/meta/exif_test.go index 424fd497c..eac8bf2ee 100644 --- a/internal/meta/exif_test.go +++ b/internal/meta/exif_test.go @@ -51,8 +51,8 @@ func TestExif(t *testing.T) { assert.Equal(t, "Photographer: TMB", data.Artist) assert.Equal(t, "2011-07-10T17:34:28Z", data.TakenAt.Format("2006-01-02T15:04:05Z")) assert.Equal(t, "2011-07-10T19:34:28Z", data.TakenAtLocal.Format("2006-01-02T15:04:05Z")) - assert.Equal(t, "", data.Title) // Should be "Ladybug" - assert.Equal(t, "", data.Keywords) // Should be "Ladybug" + assert.Equal(t, "", data.Title) // Should be "Ladybug" + assert.Equal(t, "", data.Keywords.String()) // Should be "Ladybug" assert.Equal(t, "", data.Description) assert.Equal(t, "", data.Copyright) assert.Equal(t, 540, data.Height) @@ -84,7 +84,7 @@ func TestExif(t *testing.T) { assert.Equal(t, "2017-12-21T05:17:28Z", data.TakenAt.Format("2006-01-02T15:04:05Z")) assert.Equal(t, "2017-12-21T05:17:28Z", data.TakenAtLocal.Format("2006-01-02T15:04:05Z")) assert.Equal(t, "", data.Title) - assert.Equal(t, "", data.Keywords) + assert.Equal(t, "", data.Keywords.String()) assert.Equal(t, "", data.Description) assert.Equal(t, "", data.Copyright) assert.Equal(t, 180, data.Height) @@ -305,7 +305,7 @@ func TestExif(t *testing.T) { assert.Equal(t, "2020-05-24T08:55:21Z", data.TakenAt.Format("2006-01-02T15:04:05Z")) assert.Equal(t, "2020-05-24T11:55:21Z", data.TakenAtLocal.Format("2006-01-02T15:04:05Z")) assert.Equal(t, "", data.Title) - assert.Equal(t, "", data.Keywords) + assert.Equal(t, "", data.Keywords.String()) assert.Equal(t, "", data.Description) assert.Equal(t, "", data.Copyright) assert.Equal(t, 3600, data.Height) @@ -336,7 +336,7 @@ func TestExif(t *testing.T) { assert.Equal(t, "0001-01-01T00:00:00Z", data.TakenAt.Format("2006-01-02T15:04:05Z")) assert.Equal(t, "0001-01-01T00:00:00Z", data.TakenAtLocal.Format("2006-01-02T15:04:05Z")) assert.Equal(t, "", data.Title) - assert.Equal(t, "", data.Keywords) + assert.Equal(t, "", data.Keywords.String()) assert.Equal(t, "", data.Description) assert.Equal(t, "", data.Copyright) assert.Equal(t, 43, data.Height) @@ -367,7 +367,7 @@ func TestExif(t *testing.T) { assert.Equal(t, "2017-04-09T18:33:44Z", data.TakenAt.Format("2006-01-02T15:04:05Z")) assert.Equal(t, "2017-04-09T18:33:44Z", data.TakenAtLocal.Format("2006-01-02T15:04:05Z")) assert.Equal(t, "", data.Title) - assert.Equal(t, "", data.Keywords) + assert.Equal(t, "", data.Keywords.String()) assert.Equal(t, "", data.Description) assert.Equal(t, "", data.Copyright) assert.Equal(t, 2448, data.Height) @@ -400,7 +400,7 @@ func TestExif(t *testing.T) { assert.Equal(t, "2020-10-17T17:48:24Z", data.TakenAtLocal.Format("2006-01-02T15:04:05Z")) assert.Equal(t, "Europe/Berlin", data.TimeZone) assert.Equal(t, "", data.Title) - assert.Equal(t, "", data.Keywords) + assert.Equal(t, "", data.Keywords.String()) assert.Equal(t, "", data.Description) assert.Equal(t, "", data.Copyright) assert.Equal(t, 2736, data.Height) diff --git a/internal/meta/json_exiftool.go b/internal/meta/json_exiftool.go index 3a0796909..84936c38d 100644 --- a/internal/meta/json_exiftool.go +++ b/internal/meta/json_exiftool.go @@ -64,12 +64,16 @@ func (data *Data) Exiftool(jsonData []byte, originalName string) (err error) { } // Skip empty values. - if !jsonValue.Exists() || !fieldValue.IsZero() { + if !jsonValue.Exists() { continue } switch t := fieldValue.Interface().(type) { case time.Time: + if !fieldValue.IsZero() { + continue + } + s := strings.TrimSpace(jsonValue.String()) s = strings.ReplaceAll(s, "/", ":") @@ -79,16 +83,46 @@ func (data *Data) Exiftool(jsonData []byte, originalName string) (err error) { fieldValue.Set(reflect.ValueOf(tv.Round(time.Second))) } case time.Duration: + if !fieldValue.IsZero() { + continue + } + fieldValue.Set(reflect.ValueOf(StringToDuration(jsonValue.String()))) case int, int64: + if !fieldValue.IsZero() { + continue + } + fieldValue.SetInt(jsonValue.Int()) case float32, float64: + if !fieldValue.IsZero() { + continue + } + fieldValue.SetFloat(jsonValue.Float()) case uint, uint64: + if !fieldValue.IsZero() { + continue + } + fieldValue.SetUint(jsonValue.Uint()) + case []string: + existing := fieldValue.Interface().([]string) + fieldValue.Set(reflect.ValueOf(txt.AddToWords(existing, strings.TrimSpace(jsonValue.String())))) + case Keywords: + existing := fieldValue.Interface().(Keywords) + fieldValue.Set(reflect.ValueOf(txt.AddToWords(existing, strings.TrimSpace(jsonValue.String())))) case string: + if !fieldValue.IsZero() { + continue + } + fieldValue.SetString(strings.TrimSpace(jsonValue.String())) case bool: + if !fieldValue.IsZero() { + continue + } + fieldValue.SetBool(jsonValue.Bool()) default: log.Warnf("metadata: can't assign value of type %s to %s (exiftool)", t, tagValue) @@ -190,12 +224,11 @@ func (data *Data) Exiftool(jsonData []byte, originalName string) (err error) { } if data.Projection == "equirectangular" { - data.AddKeyword(KeywordPanorama) + data.AddKeywords(KeywordPanorama) } data.Title = SanitizeTitle(data.Title) data.Description = SanitizeDescription(data.Description) - data.Keywords = SanitizeMeta(data.Keywords) data.Subject = SanitizeMeta(data.Subject) data.Artist = SanitizeMeta(data.Artist) diff --git a/internal/meta/json_test.go b/internal/meta/json_test.go index d10b086e0..b5487ab97 100644 --- a/internal/meta/json_test.go +++ b/internal/meta/json_test.go @@ -446,7 +446,7 @@ func TestJSON(t *testing.T) { assert.Equal(t, "2020-05-24T08:55:21Z", data.TakenAt.Format("2006-01-02T15:04:05Z")) assert.Equal(t, "2020-05-24T11:55:21Z", data.TakenAtLocal.Format("2006-01-02T15:04:05Z")) assert.Equal(t, "", data.Title) - assert.Equal(t, "panorama", data.Keywords) + assert.Equal(t, "panorama", data.Keywords.String()) assert.Equal(t, "", data.Description) assert.Equal(t, "", data.Copyright) assert.Equal(t, 3600, data.Height) @@ -477,7 +477,7 @@ func TestJSON(t *testing.T) { assert.Equal(t, "2018-07-25T11:18:42Z", data.TakenAt.Format("2006-01-02T15:04:05Z")) assert.Equal(t, "2018-07-25T11:18:42Z", data.TakenAtLocal.Format("2006-01-02T15:04:05Z")) assert.Equal(t, "", data.Title) - assert.Equal(t, "", data.Keywords) + assert.Equal(t, "", data.Keywords.String()) assert.Equal(t, "", data.Description) assert.Equal(t, "", data.Copyright) assert.Equal(t, 1080, data.Height) @@ -535,7 +535,7 @@ func TestJSON(t *testing.T) { assert.Equal(t, "2020-10-17T17:48:24Z", data.TakenAtLocal.Format("2006-01-02T15:04:05Z")) assert.Equal(t, "Europe/Berlin", data.TimeZone) assert.Equal(t, "", data.Title) - assert.Equal(t, "Berlin, Shop", data.Keywords) + assert.Equal(t, "berlin, shop", data.Keywords.String()) assert.Equal(t, "", data.Description) assert.Equal(t, "", data.Copyright) assert.Equal(t, 375, data.Height) @@ -693,7 +693,7 @@ func TestJSON(t *testing.T) { assert.Equal(t, "iPhone 6s", data.CameraModel) assert.Equal(t, "iPhone 6s back camera 4.15mm f/2.2", data.LensModel) assert.Equal(t, "holiday", data.Subject) - assert.Equal(t, "holiday", data.Keywords) + assert.Equal(t, "holiday", data.Keywords.String()) }) t.Run("subject-2.json", func(t *testing.T) { @@ -719,7 +719,7 @@ func TestJSON(t *testing.T) { assert.Equal(t, "iPhone 6s", data.CameraModel) assert.Equal(t, "iPhone 6s back camera 4.15mm f/2.2", data.LensModel) assert.Equal(t, "holiday, greetings", data.Subject) - assert.Equal(t, "holiday, greetings", data.Keywords) + assert.Equal(t, "greetings, holiday", data.Keywords.String()) }) t.Run("newline.json", func(t *testing.T) { @@ -735,7 +735,7 @@ func TestJSON(t *testing.T) { assert.Equal(t, "0001-01-01T00:00:00Z", data.TakenAt.Format("2006-01-02T15:04:05Z")) assert.Equal(t, "0001-01-01T00:00:00Z", data.TakenAtLocal.Format("2006-01-02T15:04:05Z")) assert.Equal(t, "This is the title", data.Title) - assert.Equal(t, "", data.Keywords) + assert.Equal(t, "", data.Keywords.String()) assert.Equal(t, "This is a\n\ndescription!", data.Description) assert.Equal(t, "This is the world.", data.Subject) assert.Equal(t, "© 2011 PhotoPrism", data.Copyright) diff --git a/internal/meta/keywords.go b/internal/meta/keywords.go index f10451131..0bda39d16 100644 --- a/internal/meta/keywords.go +++ b/internal/meta/keywords.go @@ -1,6 +1,9 @@ package meta -import "strings" +import ( + "github.com/photoprism/photoprism/pkg/txt" + "strings" +) const ( KeywordFlash = "flash" @@ -10,36 +13,38 @@ const ( KeywordEquirectangular = "equirectangular" ) +// Keywords represents a list of metadata keywords. +type Keywords []string + +// String returns a string containing all keywords. +func (w Keywords) String() string { + return strings.Join(w, ", ") +} + var AutoKeywords = []string{KeywordHdr, KeywordBurst, KeywordPanorama, KeywordEquirectangular} -// AddKeyword appends a keyword if not exists. -func (data *Data) AddKeyword(w string) { - w = strings.ToLower(SanitizeString(w)) +// AddKeywords appends keywords. +func (data *Data) AddKeywords(w string) { + w = strings.ToLower(SanitizeMeta(w)) - if len(w) < 3 { + if len(w) < 1 { return } - if !strings.Contains(data.Keywords, w) { - if data.Keywords == "" { - data.Keywords = w - } else { - data.Keywords += ", " + w - } - } + data.Keywords = txt.AddToWords(data.Keywords, w) } -// AutoAddKeywords automatically adds relevant keywords from a string (e.g. description). +// AutoAddKeywords automatically appends relevant keywords from a string (e.g. description). func (data *Data) AutoAddKeywords(s string) { - s = strings.ToLower(SanitizeString(s)) + s = strings.ToLower(SanitizeMeta(s)) - if len(s) < 3 { + if len(s) < 1 { return } for _, w := range AutoKeywords { if strings.Contains(s, w) { - data.AddKeyword(w) + data.AddKeywords(w) } } } diff --git a/internal/meta/keywords_test.go b/internal/meta/keywords_test.go index 1b90feb95..5c8f11228 100644 --- a/internal/meta/keywords_test.go +++ b/internal/meta/keywords_test.go @@ -6,29 +6,29 @@ import ( "github.com/stretchr/testify/assert" ) -func TestData_AddKeyword(t *testing.T) { +func TestData_AddKeywords(t *testing.T) { t.Run("success", func(t *testing.T) { data := NewData() - assert.Equal(t, "", data.Keywords) + assert.Equal(t, "", data.Keywords.String()) - data.AddKeyword("FooBar") + data.AddKeywords("FooBar") - assert.Equal(t, "foobar", data.Keywords) + assert.Equal(t, "foobar", data.Keywords.String()) - data.AddKeyword("BAZ") + data.AddKeywords("BAZ; pro") - assert.Equal(t, "foobar, baz", data.Keywords) + assert.Equal(t, "foobar, pro", data.Keywords.String()) }) t.Run("ignore", func(t *testing.T) { data := NewData() - assert.Equal(t, "", data.Keywords) + assert.Equal(t, "", data.Keywords.String()) - data.AddKeyword("Fo") + data.AddKeywords("Fo") - assert.Equal(t, "", data.Keywords) + assert.Equal(t, "", data.Keywords.String()) }) } @@ -36,30 +36,30 @@ func TestData_AutoAddKeywords(t *testing.T) { t.Run("success", func(t *testing.T) { data := NewData() - assert.Equal(t, "", data.Keywords) + assert.Equal(t, "", data.Keywords.String()) data.AutoAddKeywords("FooBar burst baz flash") - assert.Equal(t, "burst", data.Keywords) + assert.Equal(t, "burst", data.Keywords.String()) }) t.Run("ignore", func(t *testing.T) { data := NewData() - assert.Equal(t, "", data.Keywords) + assert.Equal(t, "", data.Keywords.String()) data.AutoAddKeywords("FooBar go pro baz banana") - assert.Equal(t, "", data.Keywords) + assert.Equal(t, "", data.Keywords.String()) }) t.Run("ignore because too short", func(t *testing.T) { data := NewData() - assert.Equal(t, "", data.Keywords) + assert.Equal(t, "", data.Keywords.String()) data.AutoAddKeywords("es") - assert.Equal(t, "", data.Keywords) + assert.Equal(t, "", data.Keywords.String()) }) } diff --git a/internal/photoprism/index_mediafile.go b/internal/photoprism/index_mediafile.go index d0e90d1f1..cb0ce7bc5 100644 --- a/internal/photoprism/index_mediafile.go +++ b/internal/photoprism/index_mediafile.go @@ -338,7 +338,7 @@ func (ind *Index) MediaFile(m *MediaFile, o IndexOptions, originalName string) ( photo.SetCoordinates(metaData.Lat, metaData.Lng, metaData.Altitude, entity.SrcXmp) // Update metadata details. - details.SetKeywords(metaData.Keywords, entity.SrcXmp) + details.SetKeywords(metaData.Keywords.String(), entity.SrcXmp) details.SetNotes(metaData.Notes, entity.SrcXmp) details.SetSubject(metaData.Subject, entity.SrcXmp) details.SetArtist(metaData.Artist, entity.SrcXmp) @@ -356,7 +356,7 @@ func (ind *Index) MediaFile(m *MediaFile, o IndexOptions, originalName string) ( photo.SetCameraSerial(metaData.CameraSerial) // Update metadata details. - details.SetKeywords(metaData.Keywords, entity.SrcMeta) + details.SetKeywords(metaData.Keywords.String(), entity.SrcMeta) details.SetNotes(metaData.Notes, entity.SrcMeta) details.SetSubject(metaData.Subject, entity.SrcMeta) details.SetArtist(metaData.Artist, entity.SrcMeta) @@ -405,7 +405,7 @@ func (ind *Index) MediaFile(m *MediaFile, o IndexOptions, originalName string) ( photo.SetCameraSerial(metaData.CameraSerial) // Update metadata details. - details.SetKeywords(metaData.Keywords, entity.SrcMeta) + details.SetKeywords(metaData.Keywords.String(), entity.SrcMeta) details.SetNotes(metaData.Notes, entity.SrcMeta) details.SetSubject(metaData.Subject, entity.SrcMeta) details.SetArtist(metaData.Artist, entity.SrcMeta) @@ -504,7 +504,7 @@ func (ind *Index) MediaFile(m *MediaFile, o IndexOptions, originalName string) ( photo.SetCameraSerial(metaData.CameraSerial) // Update metadata details. - details.SetKeywords(metaData.Keywords, entity.SrcMeta) + details.SetKeywords(metaData.Keywords.String(), entity.SrcMeta) details.SetNotes(metaData.Notes, entity.SrcMeta) details.SetSubject(metaData.Subject, entity.SrcMeta) details.SetArtist(metaData.Artist, entity.SrcMeta) diff --git a/internal/photoprism/index_mediafile_test.go b/internal/photoprism/index_mediafile_test.go index 91a877a7d..2ea367515 100644 --- a/internal/photoprism/index_mediafile_test.go +++ b/internal/photoprism/index_mediafile_test.go @@ -14,8 +14,7 @@ func TestIndex_MediaFile(t *testing.T) { t.Skip("skipping test in short mode.") } - //TODO This test MUST run if https://github.com/photoprism/photoprism/issues/1153 is fixed - /*t.Run("/exifWithFlashDescriptionProjectionKeywordsSmall", func(t *testing.T) { + t.Run("flash.jpg", func(t *testing.T) { conf := config.TestConfig() conf.InitializeTestData(t) @@ -26,24 +25,27 @@ func TestIndex_MediaFile(t *testing.T) { ind := NewIndex(conf, tf, nd, convert, NewFiles(), NewPhotos()) indexOpt := IndexOptionsAll() - mediaFile, err := NewMediaFile("../meta/testdata/exifWithFlashDescriptionProjectionKeywordsSmall.jpg") + mediaFile, err := NewMediaFile("testdata/flash.jpg") if err != nil { t.Fatal(err) } - assert.Equal(t, "", mediaFile.metaData.Keywords) - result := ind.MediaFile(mediaFile, indexOpt, "exifWithFlashDescriptionProjectionKeywordsSmall.jpg") + assert.Equal(t, "", mediaFile.metaData.Keywords.String()) - assert.Contains(t, mediaFile.metaData.Keywords, "Marienkäfer") - assert.Contains(t, mediaFile.metaData.Keywords, "burst") - assert.Contains(t, mediaFile.metaData.Keywords, "flash") - assert.Contains(t, mediaFile.metaData.Keywords, "panorama") + result := ind.MediaFile(mediaFile, indexOpt, "flash.jpg") + + words := mediaFile.metaData.Keywords.String() + + assert.Contains(t, words, "marienkäfer") + assert.Contains(t, words, "burst") + assert.Contains(t, words, "flash") + assert.Contains(t, words, "panorama") assert.Equal(t, "Animal with green eyes on table burst", mediaFile.metaData.Description) assert.Equal(t, IndexStatus("added"), result.Status) - })*/ + }) - t.Run("/blue-go-video.mp4", func(t *testing.T) { + t.Run("blue-go-video.mp4", func(t *testing.T) { conf := config.TestConfig() conf.InitializeTestData(t) diff --git a/internal/photoprism/mediafile_meta_test.go b/internal/photoprism/mediafile_meta_test.go index 79eab07f9..f2eb13a9c 100644 --- a/internal/photoprism/mediafile_meta_test.go +++ b/internal/photoprism/mediafile_meta_test.go @@ -215,7 +215,7 @@ func TestMediaFile_Exif_JPEG(t *testing.T) { assert.Equal(t, "2020-05-24T08:55:21Z", data.TakenAt.Format("2006-01-02T15:04:05Z")) assert.Equal(t, "2020-05-24T11:55:21Z", data.TakenAtLocal.Format("2006-01-02T15:04:05Z")) assert.Equal(t, "", data.Title) - assert.Equal(t, "panorama", data.Keywords) + assert.Equal(t, "panorama", data.Keywords.String()) assert.Equal(t, "", data.Description) assert.Equal(t, "", data.Copyright) assert.Equal(t, 3600, data.Height) @@ -252,7 +252,7 @@ func TestMediaFile_Exif_JPEG(t *testing.T) { assert.Equal(t, "2020-10-17T17:48:24Z", data.TakenAtLocal.Format("2006-01-02T15:04:05Z")) assert.Equal(t, "Europe/Berlin", data.TimeZone) assert.Equal(t, "", data.Title) - assert.Equal(t, "Berlin, Shop", data.Keywords) + assert.Equal(t, "berlin, shop", data.Keywords.String()) assert.Equal(t, "", data.Description) assert.Equal(t, "", data.Copyright) assert.Equal(t, 2736, data.Height) diff --git a/internal/meta/testdata/exifWithFlashDescriptionProjectionKeywordsSmall.jpg b/internal/photoprism/testdata/flash.jpg similarity index 100% rename from internal/meta/testdata/exifWithFlashDescriptionProjectionKeywordsSmall.jpg rename to internal/photoprism/testdata/flash.jpg diff --git a/internal/meta/testdata/exifWithFlashDescriptionProjectionKeywordsSmall.json b/internal/photoprism/testdata/flash.json similarity index 98% rename from internal/meta/testdata/exifWithFlashDescriptionProjectionKeywordsSmall.json rename to internal/photoprism/testdata/flash.json index 02968cc38..00284bee2 100644 --- a/internal/meta/testdata/exifWithFlashDescriptionProjectionKeywordsSmall.json +++ b/internal/photoprism/testdata/flash.json @@ -1,7 +1,7 @@ [{ - "SourceFile": "exifWithFlashDescriptionProjectionKeywordsSmall.jpg", + "SourceFile": "flash.jpg", "ExifToolVersion": 11.88, - "FileName": "exifWithFlashDescriptionProjectionKeywordsSmall.jpg", + "FileName": "flash.jpg", "Directory": ".", "FileSize": "151 kB", "FileModifyDate": "2021:03:25 11:13:55+01:00", diff --git a/pkg/txt/words.go b/pkg/txt/words.go index d74288a95..ecaa0c264 100644 --- a/pkg/txt/words.go +++ b/pkg/txt/words.go @@ -153,6 +153,17 @@ func RemoveFromWords(words []string, remove string) (results []string) { return results } +// AddToWords add words to a string slice and returns the sorted result. +func AddToWords(existing []string, words string) []string { + w := Keywords(words) + + if len(w) < 1 { + return existing + } + + return UniqueWords(append(existing, w...)) +} + // UniqueKeywords returns a slice of unique and sorted keywords without stopwords. func UniqueKeywords(s string) (results []string) { if s == "" { @@ -180,7 +191,7 @@ func UniqueKeywords(s string) (results []string) { return results } -// Sorts string slice case insensitive. +// SortCaseInsensitive performs a case-insensitive slice sort. func SortCaseInsensitive(words []string) { sort.Slice(words, func(i, j int) bool { return strings.ToLower(words[i]) < strings.ToLower(words[j]) }) }