path: root/ui/search.go

/*
macli - Unofficial CLI-Based MyAnimeList Client
Copyright © 2022-2024 Vidhu Kant Sharma <vidhukant@vidhukant.com>

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/

package ui

import (
  "strings"
  "fmt"
  "os"
  p "github.com/manifoldco/promptui"
  "vidhukant.com/macli/mal"
  "vidhukant.com/mg"
)

var AnimeSearchFields []string = []string {
  "num_episodes", "alternative_titles",
  "start_date", "end_date", "mean",
  "start_season", "rank",
  "media_type", "status",
  "average_episode_duration",
  "rating", "studios", "genres",
}

// only search animes probably only now
func AnimeSearch(label, searchString string) mg.Anime {
  animes := mal.SearchAnime(searchString, AnimeSearchFields)
  // don't show selection prompt if --auto-select is passed
  if mal.AutoSel > 0 {
    if len(animes) > 0 {
      return animes[0]
    } else {
      fmt.Println("Error: Empty response from MyAnimeList while searching for anime.")
      os.Exit(1)
    }
  }

  for i, anime := range animes {
    animes[i].DurationSeconds = anime.DurationSeconds / 60

    /* I cant find a way to add functions to the details template
     * So I am formatting the studios as one string
     * and setting as the first studio name. pretty hacky. */
    if len(anime.Studios) > 0 {
      var studiosFormatted string
      for j, studio := range anime.Studios {
        studiosFormatted = studiosFormatted + studio.Name
        // setting other studio names as ""
        animes[i].Studios[j].Name = ""
        if j != len(anime.Studios) - 1 {
          studiosFormatted = studiosFormatted + ", "
        }
      }
      animes[i].Studios[0].Name = studiosFormatted
    }

    // same with genres
    if len(anime.Genres) > 0 {
      var genresFormatted string
      for j, genre := range anime.Genres {
        genresFormatted = genresFormatted + genre.Name
        // setting other genre names as ""
        animes[i].Genres[j].Name = ""
        if j != len(anime.Genres) - 1 {
          genresFormatted = genresFormatted + ", "
        }
      }
      animes[i].Genres[0].Name = genresFormatted
    }

    var ratingFormatted string
    switch anime.ParentalRating {
      case "g":
        ratingFormatted = "G - All Ages"
      case "pg":
        ratingFormatted = "PG - Children"
      case "pg_13":
        ratingFormatted = "PG13 - Teens 13 and Older"
      case "r":
        ratingFormatted = "R - 17+ (violence & profanity)"
      case "r+":
        ratingFormatted = "R+ - Profanity & Mild Nudity"
      case "rx":
        ratingFormatted = "Rx - Hentai"
      default:
        ratingFormatted = anime.ParentalRating
    }
    animes[i].ParentalRating = ratingFormatted
  }

  template := &p.SelectTemplates {
    Label: "{{ . }}",
    Active: "{{ .Title | magenta }}",
    Inactive: "{{ .Title }}",
    Selected: "{{ .Title | blue }}",
    Details: `
--------- {{ .Title }} ----------
{{ "Number of Episodes:" | blue | bold }} {{ if .NumEpisodes }}{{ .NumEpisodes }}{{ else }}{{ "unknown" | faint }}{{ end }}
{{ "English Title:" | blue | bold }} {{ if .AltTitles.EnglishTitle }}{{ .AltTitles.EnglishTitle }}{{ else }}{{ "none" | faint }}{{ end }}
{{ "Japanese Title:" | blue | bold }} {{ if .AltTitles.JapaneseTitle }}{{ .AltTitles.JapaneseTitle }}{{ else }}{{ "none" | faint }}{{ end }}
{{ "Original Run:" | blue | bold }} {{ if .StartDate }}{{ .StartDate | cyan }}{{ else }}{{ "unknown" | faint }}{{ end }} - {{ if .EndDate }}{{ .EndDate | yellow }}{{ else }}{{ "unknown" | faint }}{{end}} {{ if .Season.Year }}({{ .Season.Name }} {{ .Season.Year }}){{ else }}{{ end }}
{{ "Mean Score:" | blue | bold }} {{ if .MeanScore }}{{ .MeanScore }}{{ else }}{{ "unknown" | faint }}{{ end }}
{{ "Rank:" | blue | bold }} {{ if .Rank }}{{ .Rank }}{{ else }}{{ "unknown" | faint }}{{ end }}
{{ "Type:" | blue | bold }} {{ .MediaType }}
{{ "Status:" | blue | bold }} {{ .Status }}
{{ "Average Duration:" | blue | bold }} {{ if .DurationSeconds }}{{ .DurationSeconds }} minutes{{ else }}{{ "unknown" | faint }}{{ end }}
{{ "Parental Rating:" | blue | bold }} {{ if .ParentalRating }}{{ .ParentalRating }}{{ else }}{{ "unknown" | faint }}{{ end }}
{{ "Studios:" | blue | bold }} {{ if .Studios }}{{ range .Studios }}{{ .Name }}{{ end }}{{ else }}{{ "unknown" | faint }}{{ end }}
{{ "Genres:" | blue | bold }} {{ if .Genres }}{{ range .Genres }}{{ .Name }}{{ end }}{{ else }}{{ "unknown" | faint }}{{ end }}
`,
  }

  // returns true if input == anime title
  searcher := func(input string, index int) bool {
    title := strings.Replace(strings.ToLower(animes[index].Title), " ", "", -1)
    input = strings.Replace(strings.ToLower(input), " ", "", -1)
    return strings.Contains(title, input)
  }

  prompt := p.Select {
    Label: label,
    Items: animes,
    Templates: template,
    Searcher: searcher,
    Size: PromptLength,
  }

  animeIndex, _, err := prompt.Run()
  if err != nil {
    fmt.Println("Error running search menu.", err.Error())
    os.Exit(1)
  }

  return animes[animeIndex]
}

var MangaSearchFields []string = []string {
  "num_chapters", "num_volumes",
  "alternative_titles", "start_date",
  "end_date", "mean", "rank", "genres",
  "media_type", "status", "authors",
}

func MangaSearch(label, searchString string) mg.Manga {
  mangas := mal.SearchManga(searchString, MangaSearchFields)
  // don't show selection prompt if --auto-select is passed
  if mal.AutoSel > 0 {
    if len(mangas) > 0 {
      return mangas[0]
    } else {
      fmt.Println("Error: Empty response from MyAnimeList while searching for manga.")
      os.Exit(1)
    }
  }

  for i, manga := range mangas {
    /* I cant find a way to add functions to the details template
     * So I am formatting the authors as one string
     * and setting as the first author name. pretty hacky. */
    if len(manga.Authors) > 0 {
      var authorsFormatted string
      for j, author := range manga.Authors {
        authorsFormatted = authorsFormatted + author.Details.FirstName
				if author.Details.LastName != "" {
					authorsFormatted = authorsFormatted + " " + author.Details.LastName
				}

				if author.Role != "" {
					authorsFormatted = authorsFormatted + " (" + author.Role + ")"
				}

        // setting other author names as ""
        mangas[i].Authors[j].Details.FirstName = ""
        mangas[i].Authors[j].Details.LastName = ""
        if j != len(manga.Authors) - 1 {
          authorsFormatted = authorsFormatted + ", "
        }
      }
      mangas[i].Authors[0].Details.FirstName = authorsFormatted
    }

    // same with genres
    if len(manga.Genres) > 0 {
      var genresFormatted string
      for j, genre := range manga.Genres {
        genresFormatted = genresFormatted + genre.Name
        // setting other genre names as ""
        mangas[i].Genres[j].Name = ""
        if j != len(manga.Genres) - 1 {
          genresFormatted = genresFormatted + ", "
        }
      }
      mangas[i].Genres[0].Name = genresFormatted
    }
  }

  template := &p.SelectTemplates {
    Label: "{{ . }}",
    Active: "{{ .Title | magenta }}",
    Inactive: "{{ .Title }}",
    Selected: "{{ .Title | blue }}",
    Details: `
--------- {{ .Title }} ----------
{{ "Number of Volumes:" | blue | bold }} {{ if .NumVolumes }}{{ .NumVolumes }}{{ else }}{{ "unknown" | faint }}{{ end }}
{{ "Number of Chapters:" | blue | bold }} {{ if .NumChapters }}{{ .NumChapters }}{{ else }}{{ "unknown" | faint }}{{ end }}
{{ "English Title:" | blue | bold }} {{ if .AltTitles.EnglishTitle }}{{ .AltTitles.EnglishTitle }}{{ else }}{{ "none" | faint }}{{ end }}
{{ "Japanese Title:" | blue | bold }} {{ if .AltTitles.JapaneseTitle }}{{ .AltTitles.JapaneseTitle }}{{ else }}{{ "none" | faint }}{{ end }}
{{ "Original Run:" | blue | bold }} {{ if .StartDate }}{{ .StartDate | cyan }}{{ else }}{{ "unknown" | faint }}{{ end }} - {{ if .EndDate }}{{ .EndDate | yellow }}{{ else }}{{ "unknown" | faint }}{{end}}
{{ "Mean Score:" | blue | bold }} {{ if .MeanScore }}{{ .MeanScore }}{{ else }}{{ "unknown" | faint }}{{ end }}
{{ "Rank:" | blue | bold }} {{ if .Rank }}{{ .Rank }}{{ else }}{{ "unknown" | faint }}{{ end }}
{{ "Type:" | blue | bold }} {{ .MediaType }}
{{ "Status:" | blue | bold }} {{ .Status }}
{{ "Authors:" | blue | bold }} {{ if .Authors }}{{ range .Authors }}{{ .Details.FirstName }}{{ end }}{{ else }}{{ "unknown" | faint }}{{ end }}
{{ "Genres:" | blue | bold }} {{ if .Genres }}{{ range .Genres }}{{ .Name }}{{ end }}{{ else }}{{ "unknown" | faint }}{{ end }}
`,
  }

  // returns true if input == anime title
  searcher := func(input string, index int) bool {
    title := strings.Replace(strings.ToLower(mangas[index].Title), " ", "", -1)
    input = strings.Replace(strings.ToLower(input), " ", "", -1)
    return strings.Contains(title, input)
  }

  prompt := p.Select {
    Label: label,
    Items: mangas,
    Templates: template,
    Searcher: searcher,
    Size: PromptLength,
  }

  mangaIndex, _, err := prompt.Run()
  if err != nil {
    fmt.Println("Error running search menu.", err.Error())
    os.Exit(1)
  }

  return mangas[mangaIndex]
}