From bfe5550739759bf0333947d0f27ce17ed4d5ddf9 Mon Sep 17 00:00:00 2001 From: Erik Winter Date: Thu, 23 May 2024 10:59:13 +0200 Subject: [PATCH] first version themoviedb export --- tmdb-export/.gitignore | 1 + tmdb-export/go.mod | 34 ++++++++++++ tmdb-export/go.sum | 71 +++++++++++++++++++++++++ tmdb-export/main.go | 112 ++++++++++++++++++++++++++++++++++++++++ tmdb-export/markdown.go | 74 ++++++++++++++++++++++++++ tmdb-export/tmdb.go | 74 ++++++++++++++++++++++++++ 6 files changed, 366 insertions(+) create mode 100644 tmdb-export/.gitignore create mode 100644 tmdb-export/go.mod create mode 100644 tmdb-export/go.sum create mode 100644 tmdb-export/main.go create mode 100644 tmdb-export/markdown.go create mode 100644 tmdb-export/tmdb.go diff --git a/tmdb-export/.gitignore b/tmdb-export/.gitignore new file mode 100644 index 0000000..2e1fa2d --- /dev/null +++ b/tmdb-export/.gitignore @@ -0,0 +1 @@ +*.md \ No newline at end of file diff --git a/tmdb-export/go.mod b/tmdb-export/go.mod new file mode 100644 index 0000000..a6c7e78 --- /dev/null +++ b/tmdb-export/go.mod @@ -0,0 +1,34 @@ +module codeberg.org/ewintr/scripts/tmdb-export + +go 1.21.5 + +require ( + github.com/cyruzin/golang-tmdb v1.6.1 + github.com/erikgeiser/promptkit v0.9.0 + github.com/muesli/termenv v0.15.2 +) + +require ( + code.ewintr.nl/go-kit v0.0.0-20240308074309-a1328c3c44c6 // indirect + github.com/atotto/clipboard v0.1.4 // indirect + github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect + github.com/charmbracelet/bubbles v0.16.1 // indirect + github.com/charmbracelet/bubbletea v0.24.2 // indirect + github.com/charmbracelet/lipgloss v0.7.1 // indirect + github.com/containerd/console v1.0.4-0.20230313162750-1ae8d489ac81 // indirect + github.com/json-iterator/go v1.1.12 // indirect + github.com/lucasb-eyer/go-colorful v1.2.0 // indirect + github.com/mattn/go-isatty v0.0.19 // indirect + github.com/mattn/go-localereader v0.0.1 // indirect + github.com/mattn/go-runewidth v0.0.15 // indirect + github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 // indirect + github.com/modern-go/reflect2 v1.0.2 // indirect + github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 // indirect + github.com/muesli/cancelreader v0.2.2 // indirect + github.com/muesli/reflow v0.3.0 // indirect + github.com/rivo/uniseg v0.4.4 // indirect + golang.org/x/sync v0.3.0 // indirect + golang.org/x/sys v0.11.0 // indirect + golang.org/x/term v0.11.0 // indirect + golang.org/x/text v0.14.0 // indirect +) diff --git a/tmdb-export/go.sum b/tmdb-export/go.sum new file mode 100644 index 0000000..82a7e2c --- /dev/null +++ b/tmdb-export/go.sum @@ -0,0 +1,71 @@ +code.ewintr.nl/go-kit v0.0.0-20240308074309-a1328c3c44c6 h1:qZwAicZOd18o9qWCU3pSj/QnrSNLCAMAOPJvyLgBQ0U= +code.ewintr.nl/go-kit v0.0.0-20240308074309-a1328c3c44c6/go.mod h1:Yk8Mdn1f4/L9tcymurtItqevVkac6P9ljh2Sd3T+FS8= +github.com/atotto/clipboard v0.1.4 h1:EH0zSVneZPSuFR11BlR9YppQTVDbh5+16AmcJi4g1z4= +github.com/atotto/clipboard v0.1.4/go.mod h1:ZY9tmq7sm5xIbd9bOK4onWV4S6X0u6GY7Vn0Yu86PYI= +github.com/aymanbagabas/go-osc52/v2 v2.0.1 h1:HwpRHbFMcZLEVr42D4p7XBqjyuxQH5SMiErDT4WkJ2k= +github.com/aymanbagabas/go-osc52/v2 v2.0.1/go.mod h1:uYgXzlJ7ZpABp8OJ+exZzJJhRNQ2ASbcXHWsFqH8hp8= +github.com/charmbracelet/bubbles v0.16.1 h1:6uzpAAaT9ZqKssntbvZMlksWHruQLNxg49H5WdeuYSY= +github.com/charmbracelet/bubbles v0.16.1/go.mod h1:2QCp9LFlEsBQMvIYERr7Ww2H2bA7xen1idUDIzm/+Xc= +github.com/charmbracelet/bubbletea v0.24.2 h1:uaQIKx9Ai6Gdh5zpTbGiWpytMU+CfsPp06RaW2cx/SY= +github.com/charmbracelet/bubbletea v0.24.2/go.mod h1:XdrNrV4J8GiyshTtx3DNuYkR1FDaJmO3l2nejekbsgg= +github.com/charmbracelet/lipgloss v0.7.1 h1:17WMwi7N1b1rVWOjMT+rCh7sQkvDU75B2hbZpc5Kc1E= +github.com/charmbracelet/lipgloss v0.7.1/go.mod h1:yG0k3giv8Qj8edTCbbg6AlQ5e8KNWpFujkNawKNhE2c= +github.com/containerd/console v1.0.4-0.20230313162750-1ae8d489ac81 h1:q2hJAaP1k2wIvVRd/hEHD7lacgqrCPS+k8g1MndzfWY= +github.com/containerd/console v1.0.4-0.20230313162750-1ae8d489ac81/go.mod h1:YynlIjWYF8myEu6sdkwKIvGQq+cOckRm6So2avqoYAk= +github.com/cyruzin/golang-tmdb v1.6.1 h1:LH0MDPYDb2TEZH3dfNjQGgH6mIlWt2fzpjnnTk+jifA= +github.com/cyruzin/golang-tmdb v1.6.1/go.mod h1:ZSryJLCcY+9TiKU+LbouXKns++YBrM8Tizannr05c+I= +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/erikgeiser/promptkit v0.9.0 h1:3qL1mS/ntCrXdb8sTP/ka82CJ9kEQaGuYXNrYJkWYBc= +github.com/erikgeiser/promptkit v0.9.0/go.mod h1:pU9dtogSe3Jlc2AY77EP7R4WFP/vgD4v+iImC83KsCo= +github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg= +github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM= +github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo= +github.com/lucasb-eyer/go-colorful v1.2.0 h1:1nnpGOrhyZZuNyfu1QjKiUICQ74+3FNCN69Aj6K7nkY= +github.com/lucasb-eyer/go-colorful v1.2.0/go.mod h1:R4dSotOR9KMtayYi1e77YzuveK+i7ruzyGqttikkLy0= +github.com/mattn/go-isatty v0.0.19 h1:JITubQf0MOLdlGRuRq+jtsDlekdYPia9ZFsB8h/APPA= +github.com/mattn/go-isatty v0.0.19/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= +github.com/mattn/go-localereader v0.0.1 h1:ygSAOl7ZXTx4RdPYinUpg6W99U8jWvWi9Ye2JC/oIi4= +github.com/mattn/go-localereader v0.0.1/go.mod h1:8fBrzywKY7BI3czFoHkuzRoWE9C+EiG4R1k4Cjx5p88= +github.com/mattn/go-runewidth v0.0.12/go.mod h1:RAqKPSqVFrSLVXbA8x7dzmKdmGzieGRCM46jaSJTDAk= +github.com/mattn/go-runewidth v0.0.15 h1:UNAjwbU9l54TA3KzvqLGxwWjHmMgBUVhBiTjelZgg3U= +github.com/mattn/go-runewidth v0.0.15/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w= +github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 h1:ZqeYNhU3OHLH3mGKHDcjJRFFRrJa6eAM5H+CtDdOsPc= +github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= +github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M= +github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk= +github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 h1:ZK8zHtRHOkbHy6Mmr5D264iyp3TiX5OmNcI5cIARiQI= +github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6/go.mod h1:CJlz5H+gyd6CUWT45Oy4q24RdLyn7Md9Vj2/ldJBSIo= +github.com/muesli/cancelreader v0.2.2 h1:3I4Kt4BQjOR54NavqnDogx/MIoWBFa0StPA8ELUXHmA= +github.com/muesli/cancelreader v0.2.2/go.mod h1:3XuTXfFS2VjM+HTLZY9Ak0l6eUKfijIfMUZ4EgX0QYo= +github.com/muesli/reflow v0.3.0 h1:IFsN6K9NfGtjeggFP+68I4chLZV2yIKsXJFNZ+eWh6s= +github.com/muesli/reflow v0.3.0/go.mod h1:pbwTDkVPibjO2kyvBQRBxTWEEGDGq0FlB1BIKtnHY/8= +github.com/muesli/termenv v0.15.2 h1:GohcuySI0QmI3wN8Ok9PtKGkgkFIk7y6Vpb5PvrY+Wo= +github.com/muesli/termenv v0.15.2/go.mod h1:Epx+iuz8sNs7mNKhxzH4fWXGNpZwUaJKRS1noLXviQ8= +github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/rivo/uniseg v0.1.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc= +github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc= +github.com/rivo/uniseg v0.4.4 h1:8TfxU8dW6PdqD27gjM8MVNuicgxIjxpm4K7x4jp8sis= +github.com/rivo/uniseg v0.4.4/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88= +github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= +github.com/stretchr/testify v1.7.1 h1:5TQK59W5E3v0r2duFAb7P95B6hEeOyEnHRa8MjYSMTY= +github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +golang.org/x/sync v0.3.0 h1:ftCYgMx6zT/asHUrPw8BLLscYtGznsLAnjq5RH9P66E= +golang.org/x/sync v0.3.0/go.mod h1:FU7BRWz2tNW+3quACPkgCx/L+uEAv1htQ0V83Z9Rj+Y= +golang.org/x/sys v0.1.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.11.0 h1:eG7RXZHdqOJ1i+0lgLgCpSXAp6M3LYlAo6osgSi0xOM= +golang.org/x/sys v0.11.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/term v0.11.0 h1:F9tnn/DA/Im8nCwm+fX+1/eBwi4qFjRT++MhtVC4ZX0= +golang.org/x/term v0.11.0/go.mod h1:zC9APTIj3jG3FdV/Ons+XE1riIZXG4aZ4GTHiPZJPIU= +golang.org/x/text v0.12.0 h1:k+n5B8goJNdU7hSvEtMUz3d1Q6D/XW4COJSJR6fN0mc= +golang.org/x/text v0.12.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE= +golang.org/x/text v0.14.0 h1:ScX5w1eTa3QqT8oi6+ziP7dTV1S2+ALU0bI+0zXKWiQ= +golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= +gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= diff --git a/tmdb-export/main.go b/tmdb-export/main.go new file mode 100644 index 0000000..fc0f2a8 --- /dev/null +++ b/tmdb-export/main.go @@ -0,0 +1,112 @@ +package main + +import ( + "fmt" + "os" + "strings" + + "github.com/erikgeiser/promptkit/selection" + "github.com/erikgeiser/promptkit/textinput" + "github.com/muesli/termenv" +) + +const ( + promptTemplate = ` +{{- if .Prompt -}} + {{ Bold .Prompt }} +{{ end -}} +{{ if .IsFiltered }} + {{- print .FilterPrompt " " .FilterInput }} +{{ end }} + +{{- range $i, $choice := .Choices }} + {{- if IsScrollUpHintPosition $i }} + {{- print "⇡ " -}} + {{- else if IsScrollDownHintPosition $i -}} + {{- print "⇣ " -}} + {{- else -}} + {{- print " " -}} + {{- end -}} + + {{- if eq $.SelectedIndex $i }} + {{- print "[" (Foreground "32" (Bold "x")) "] " (Selected $choice) "\n" }} + {{- else }} + {{- print "[ ] " (Unselected $choice) "\n" }} + {{- end }} +{{- end}} +` + resultTemplate = ` + {{- print .Prompt " " (Foreground "32" (name .FinalChoice)) "\n" -}} + ` +) + +type Movie struct { + TMDBID string + IMDBID string + Title string + EnglishTitle string + Year int + Directors []string + Summary string +} + +func main() { + tmdb, err := NewTMDB(os.Getenv("TMDB_API_KEY")) + if err != nil { + fmt.Println(err) + os.Exit(1) + } + + input := textinput.New("TheMovieDB search term:") + //input.InitialValue = os.Getenv("USER") + //input.Placeholder = "Your name cannot be empty" + + searchTerm, err := input.RunPrompt() + if err != nil { + fmt.Printf("Error: %v\n", err) + + os.Exit(1) + } + + movies, err := tmdb.Search(searchTerm) + if err != nil { + fmt.Println(err) + os.Exit(1) + } + if len(movies) == 0 { + fmt.Println("No results found") + os.Exit(1) + } + + blue := termenv.String().Foreground(termenv.ANSI256Color(32)) //nolint:gomnd + + sp := selection.New("Choose a movie!", movies) + sp.FilterPrompt = "Filter by ID:" + sp.FilterPlaceholder = "Type to filter" + sp.PageSize = 3 + sp.LoopCursor = true + sp.Filter = func(filter string, choice *selection.Choice[Movie]) bool { + return strings.HasPrefix(choice.Value.TMDBID, filter) + } + sp.Template = promptTemplate + sp.ResultTemplate = resultTemplate + sp.SelectedChoiceStyle = func(c *selection.Choice[Movie]) string { + return fmt.Sprintf("%s (%s)", blue.Bold().Styled(c.Value.Title), termenv.String(fmt.Sprintf("%d", c.Value.Year)).Faint().String()) + } + sp.UnselectedChoiceStyle = func(c *selection.Choice[Movie]) string { + return fmt.Sprintf("%s (%s)", c.Value.Title, termenv.String(fmt.Sprintf("%d", c.Value.Year)).Faint().String()) + } + sp.ExtendedTemplateFuncs = map[string]interface{}{ + "name": func(c *selection.Choice[Movie]) string { return c.Value.Title }, + } + + choice, err := sp.RunPrompt() + if err != nil { + fmt.Printf("could not select: %v\n", err) + os.Exit(1) + } + + if err := Export(choice); err != nil { + fmt.Printf("could not export to markdown: %v\n", err) + } +} diff --git a/tmdb-export/markdown.go b/tmdb-export/markdown.go new file mode 100644 index 0000000..d13cbd8 --- /dev/null +++ b/tmdb-export/markdown.go @@ -0,0 +1,74 @@ +package main + +import ( + "fmt" + "os" + "strings" + "text/template" +) + +const ( + markdownTemplate = `--- +tmdb: {{ .TMDBID }} +emdb: {{ .IMDBID }} +englishTitle: {{ .EnglishTitle }} +title: {{ .Title }} +year: {{ .Year }} +directors: {{ .DirectorsYAML }} +inCollection: no +watchedOn: +rating: +--- + +# {{ .EnglishTitle }} ({{ .Year }}) + +Director(s): {{ .Directors }} + +{{ .Summary }} + +## Comment +` +) + +func Export(movie Movie) error { + tpl, err := template.New("page").Parse(markdownTemplate) + if err != nil { + return err + } + + filename := fmt.Sprintf("%s (%d).md", movie.EnglishTitle, movie.Year) + + f, err := os.Create(filename) + if err != nil { + return err + } + + data := struct { + TMDBID string + IMDBID string + EnglishTitle string + Title string + Year int + DirectorsYAML string + Directors string + Summary string + }{ + TMDBID: movie.TMDBID, + IMDBID: movie.IMDBID, + EnglishTitle: movie.EnglishTitle, + Title: movie.Title, + Year: movie.Year, + DirectorsYAML: strings.Join(movie.Directors, ", "), + Directors: fmt.Sprintf("[[%s]]", strings.Join(movie.Directors, "]], [[")), + Summary: movie.Summary, + } + + if err := tpl.Execute(f, data); err != nil { + return err + } + if err := f.Close(); err != nil { + return err + } + + return nil +} diff --git a/tmdb-export/tmdb.go b/tmdb-export/tmdb.go new file mode 100644 index 0000000..ecb0440 --- /dev/null +++ b/tmdb-export/tmdb.go @@ -0,0 +1,74 @@ +package main + +import ( + "fmt" + "time" + + tmdb "github.com/cyruzin/golang-tmdb" +) + +type TMDB struct { + c *tmdb.Client +} + +func NewTMDB(apikey string) (*TMDB, error) { + tmdbClient, err := tmdb.Init(apikey) + if err != nil { + return nil, err + } + tmdbClient.SetClientAutoRetry() + tmdbClient.SetAlternateBaseURL() + + return &TMDB{ + c: tmdbClient, + }, nil +} + +func (t TMDB) Search(query string) ([]Movie, error) { + results, err := t.c.GetSearchMovies(query, nil) + if err != nil { + return nil, err + } + + movies := make([]Movie, len(results.Results)) + for i, result := range results.Results { + movies[i], err = t.GetMovie(result.ID) + if err != nil { + return nil, err + } + } + + return movies, nil +} + +func (t TMDB) GetMovie(id int64) (Movie, error) { + result, err := t.c.GetMovieDetails(int(id), map[string]string{ + "append_to_response": "credits", + }) + if err != nil { + return Movie{}, err + } + + var year int + if release, err := time.Parse("2006-01-02", result.ReleaseDate); err == nil { + year = release.Year() + } + + directors := make([]string, 0) + for crew := range result.Credits.Crew { + if result.Credits.Crew[crew].Job == "Director" { + directors = append(directors, result.Credits.Crew[crew].Name) + } + } + + return Movie{ + Title: result.OriginalTitle, + EnglishTitle: result.Title, + TMDBID: fmt.Sprintf("%d", result.ID), + IMDBID: result.IMDbID, + Year: year, + Directors: directors, + Summary: result.Overview, + }, nil + +}