poke.go

package pokesay

import (
	"github.com/pkg/errors"
	"strings"
)

type Poke struct {
	phrase string
	thinking bool
	typ string
}

const imageUrlPrefix = "https://raw.githubusercontent.com/TaigaMikami/Pokesay/master/images/"

func NewPoke(options ...Option) (*Poke, error) {
	poke := &Poke{
		phrase: "",
		typ: imageUrlPrefix + "Pikachu.png",
	}

	for _, o := range options {
		if err := o(poke); err != nil {
			return  nil, err
		}
	}
	return poke, nil
}

type Option func(*Poke) error

func Random() Option {
	return func(poke *Poke) error {
		s := pickPoke()
		s = complementFilePath(s)
		poke.typ = s
		return nil
	}
}

func Phrase(s string) Option {
	return func(poke *Poke) error {
		poke.phrase = s
		return nil
	}
}

func Thinking() Option {
	return func(poke *Poke) error {
		poke.thinking = false
		return nil
	}
}

func Type(s string) Option {
	s = complementFilePath(s)

	return func(poke *Poke) error {
		containPoke, err := containPokes(s)
		if err != nil {
			return err
		}
		if containPoke {
			poke.typ = s
			return nil
		}
		return errors.Errorf("Could not find %s", s)
	}
}

func containPokes(s string) (bool, error) {
	for _, image := range PokemonNames() {
		if strings.Contains(s, image) {
			return true, nil
		}
	}
	return false, nil
}

func complementFilePath(s string) string {
	if s == "" {
		s = imageUrlPrefix + "Pikachu.png"
	}

	if !strings.HasSuffix(s, ".png") {
		s += ".png"
	}
	if !strings.HasPrefix(s, imageUrlPrefix) {
		s = imageUrlPrefix + s
	}

	return s
}

func PokemonNames() []string {
	names := []string{
		"Abomasnow",
		"Abra",
		"Absol",
		"Accelgor",
		"Aegislash",
		"Aerodactyl",
		"Aggron",
		"Aipom",
		"Alakazam",
		"Alomomola",
		"Altaria",
		"Amaura",
		"Ambipom",
		"Amoonguss",
		"Ampharos",
		"Anorith",
		"Araquanid",
		"Arbok",
		"Arcanine",
		"Arceus",
		"Archen",
		"Archeops",
		"Ariados",
		"Armaldo",
		"Aromatisse",
		"Aron",
		"Articuno",
		"Audino",
		"Aurorus",
		"Avalugg",
		"Axew",
		"Azelf",
		"Azumarill",
		"Azurill",
		"Bagon",
		"Baltoy",
		"Banette",
		"Barbaracle",
		"Barboach",
		"Basculin-B",
		"Basculin",
		"Bastiodon",
		"Bayleef",
		"Beartic",
		"Beautifly",
		"Beedrill",
		"Beheeyem",
		"Beldum",
		"Bellossom",
		"Bellsprout",
		"Bergmite",
		"Bewear",
		"Bibarel",
		"Bidoof",
		"Binacle",
		"Bisharp",
		"Blastoise",
		"Blaziken",
		"Blissey",
		"Blitzle",
		"Boldore",
		"Bonsly",
		"Bouffalant",
		"Bounsweet",
		"Braixen",
		"Braviary",
		"Breloom",
		"Brionne",
		"Bronzong",
		"Bronzor",
		"Bruxish",
		"Budew",
		"Buizel",
		"Bulbasaur",
		"Buneary",
		"Bunnelby",
		"Burmy-G",
		"Burmy-S",
		"Burmy",
		"Butterfree",
		"Buzzwole",
		"Cacnea",
		"Cacturne",
		"Camerupt",
		"Carbink",
		"Carnivine",
		"Carracosta",
		"Carvanha",
		"Cascoon",
		"Castform-H",
		"Castform-R",
		"Castform-S",
		"Castform",
		"Caterpie",
		"Celebi",
		"Celesteela",
		"Chandelure",
		"Chansey",
		"Charizard",
		"Charjabug",
		"Charmander",
		"Charmeleon",
		"Chatot",
		"Cherrim",
		"Cherubi",
		"Chesnaught",
		"Chespin",
		"Chikorita",
		"Chimchar",
		"Chimecho",
		"Chinchou",
		"Chingling",
		"Cinccino",
		"Clamperl",
		"Clauncher",
		"Clawitzer",
		"Claydol",
		"Clefable",
		"Clefairy",
		"Cleffa",
		"Cloyster",
		"Cobalion",
		"Cofagrigus",
		"Combee",
		"Combusken",
		"Comfey",
		"Conkeldurr",
		"Corphish",
		"Corsola",
		"Cosmoem",
		"Cosmog",
		"Cottonee",
		"Crabominable",
		"Crabrawler",
		"Cradily",
		"Cranidos",
		"Crawdaunt",
		"Cresselia",
		"Croagunk",
		"Crobat",
		"Croconaw",
		"Crustle",
		"Cryogonal",
		"Cubchoo",
		"Cubone",
		"Cutiefly",
		"Cyndaquil",
		"Darkrai",
		"Darmanitan-Z",
		"Darmanitan",
		"Dartrix",
		"Darumaka",
		"Decidueye",
		"Dedenne",
		"Deerling",
		"Deino",
		"Delcatty",
		"Delibird",
		"Delphox",
		"Deoxys-A",
		"Deoxys-D",
		"Deoxys-S",
		"Deoxys",
		"Dewgong",
		"Dewott",
		"Dewpider",
		"Dhelmise",
		"Dialga",
		"Diancie",
		"Diggersby",
		"Diglett-A",
		"Diglett",
		"Ditto",
		"Dodrio",
		"Doduo",
		"Donphan",
		"Doublade",
		"Dragalge",
		"Dragonair",
		"Dragonite",
		"Drampa",
		"Drapion",
		"Dratini",
		"Drifblim",
		"Drifloon",
		"Drilbur",
		"Drowzee",
		"Druddigon",
		"Ducklett",
		"Dugtrio-A",
		"Dugtrio",
		"Dunsparce",
		"Duosion",
		"Durant",
		"Dusclops",
		"Dusknoir",
		"Duskull",
		"Dustox",
		"Dwebble",
		"Eelektrik",
		"Eelektross",
		"Eevee",
		"Ekans",
		"Electabuzz",
		"Electivire",
		"Electrike",
		"Electrode",
		"Elekid",
		"Elgyem",
		"Emboar",
		"Emolga",
		"Empoleon",
		"Entei",
		"Escavalier",
		"Espeon",
		"Espurr",
		"Excadrill",
		"Exeggcute",
		"Exeggutor-A",
		"Exeggutor",
		"Exploud",
		"Farfetch'd",
		"Fearow",
		"Feebas",
		"Fennekin",
		"Feraligatr",
		"Ferroseed",
		"Ferrothorn",
		"Finneon",
		"Flaaffy",
		"Flabébé",
		"Flareon",
		"Fletchinder",
		"Fletchling",
		"Floatzel",
		"Floette",
		"Florges",
		"Flygon",
		"Fomantis",
		"Foongus",
		"Forretress",
		"Fraxure",
		"Frillish-F",
		"Frillish",
		"Froakie",
		"Frogadier",
		"Froslass",
		"Furfrou",
		"Furret",
		"Gabite",
		"Gallade",
		"Galvantula",
		"Garbodor",
		"Garchomp",
		"Gardevoir",
		"Gastly",
		"Gastrodon-E",
		"Gastrodon",
		"Genesect",
		"Gengar",
		"Geodude-A",
		"Geodude",
		"Gible",
		"Gigalith",
		"Girafarig",
		"Giratina-O",
		"Giratina",
		"Glaceon",
		"Glalie",
		"Glameow",
		"Gligar",
		"Gliscor",
		"Gloom",
		"Gogoat",
		"Golbat",
		"Goldeen",
		"Golduck",
		"Golem-A",
		"Golem",
		"Golett",
		"Golisopod",
		"Golurk",
		"Goodra",
		"Goomy",
		"Gorebyss",
		"Gothita",
		"Gothitelle",
		"Gothorita",
		"Gourgeist",
		"Granbull",
		"Graveler-A",
		"Graveler",
		"Greninja",
		"Grimer-A",
		"Grimer",
		"Grotle",
		"Groudon",
		"Grovyle",
		"Growlithe",
		"Grubbin",
		"Grumpig",
		"Gulpin",
		"Gumshoos",
		"Gurdurr",
		"Guzzlord",
		"Gyarados",
		"Hakamo-o",
		"Happiny",
		"Hariyama",
		"Haunter",
		"Hawlucha",
		"Haxorus",
		"Heatmor",
		"Heatran",
		"Heliolisk",
		"Helioptile",
		"Heracross",
		"Herdier",
		"Hippopotas",
		"Hippowdon",
		"Hitmonchan",
		"Hitmonlee",
		"Hitmontop",
		"Ho-Oh",
		"Honchkrow",
		"Honedge",
		"Hoopa-U",
		"Hoopa",
		"Hoothoot",
		"Hoppip",
		"Horsea",
		"Houndoom",
		"Houndour",
		"Huntail",
		"Hydreigon",
		"Hypno",
		"Igglybuff",
		"Illumise",
		"Incineroar",
		"Infernape",
		"Inkay",
		"Ivysaur",
		"Jangmo-o",
		"Jellicent-F",
		"Jellicent",
		"Jigglypuff",
		"Jirachi",
		"Jolteon",
		"Joltik",
		"Jumpluff",
		"Jynx",
		"Kabuto",
		"Kabutops",
		"Kadabra",
		"Kakuna",
		"Kangaskhan",
		"Karrablast",
		"Kartana",
		"Kecleon",
		"Keldeo",
		"Kingdra",
		"Kingler",
		"Kirlia",
		"Klang",
		"Klefki",
		"Klink",
		"Klinklang",
		"Koffing",
		"Komala",
		"Kommo-o",
		"Krabby",
		"Kricketot",
		"Kricketune",
		"Krokorok",
		"Krookodile",
		"Kyogre",
		"Kyurem",
		"Lairon",
		"Lampent",
		"Landorus",
		"Lanturn",
		"Lapras",
		"Larvesta",
		"Larvitar",
		"Latias",
		"Latios",
		"Leafeon",
		"Leavanny",
		"Ledian",
		"Ledyba",
		"Lickilicky",
		"Lickitung",
		"Liepard",
		"Lileep",
		"Lilligant",
		"Lillipup",
		"Linoone",
		"Litleo",
		"Litten",
		"Litwick",
		"Lombre",
		"Lopunny",
		"Lotad",
		"Loudred",
		"Lucario",
		"Ludicolo",
		"Lugia",
		"Lumineon",
		"Lunala",
		"Lunatone",
		"Lurantis",
		"Luvdisc",
		"Luxio",
		"Luxray",
		"Lycanroc",
		"Machamp",
		"Machoke",
		"Machop",
		"Magby",
		"Magcargo",
		"Magearna",
		"Magikarp",
		"Magmar",
		"Magmortar",
		"Magnemite",
		"Magneton",
		"Magnezone",
		"Makuhita",
		"Malamar",
		"Mamoswine",
		"Manaphy",
		"Mandibuzz",
		"Manectric",
		"Mankey",
		"Mantine",
		"Mantyke",
		"Maractus",
		"Mareanie",
		"Mareep",
		"Marill",
		"Marowak-A",
		"Marowak",
		"Marshadow",
		"Marshtomp",
		"Masquerain",
		"Mawile",
		"Medicham",
		"Meditite",
		"Meganium",
		"Meloetta-P",
		"Meloetta",
		"Meowstic",
		"Meowth-A",
		"Meowth",
		"Mesprit",
		"Metagross",
		"Metang",
		"Metapod",
		"Mew",
		"Mewtwo",
		"Mienfoo",
		"Mienshao",
		"Mightyena",
		"Milotic",
		"Miltank",
		"Mime Jr.",
		"Mimikyu",
		"Minccino",
		"Minior",
		"Minun",
		"Misdreavus",
		"Mismagius",
		"Moltres",
		"Monferno",
		"Morelull",
		"Mothim",
		"Mr. Mime",
		"Mudbray",
		"Mudkip",
		"Mudsdale",
		"Muk-A",
		"Muk",
		"Munchlax",
		"Munna",
		"Murkrow",
		"Musharna",
		"Natu",
		"Necrozma",
		"Nidoking",
		"Nidoqueen",
		"Nidoran♀",
		"Nidoran♂",
		"Nidorina",
		"Nidorino",
		"Nihilego",
		"Nincada",
		"Ninetales-A",
		"Ninetales",
		"Ninjask",
		"Noctowl",
		"Noibat",
		"Noivern",
		"Nosepass",
		"Numel",
		"Nuzleaf",
		"Octillery",
		"Oddish",
		"Omanyte",
		"Omastar",
		"Onix",
		"Oranguru",
		"Oricorio",
		"Oshawott",
		"Pachirisu",
		"Palkia",
		"Palossand",
		"Palpitoad",
		"Pancham",
		"Pangoro",
		"Panpour",
		"Pansage",
		"Pansear",
		"Paras",
		"Parasect",
		"Passimian",
		"Patrat",
		"Pawniard",
		"Pelipper",
		"Persian-A",
		"Persian",
		"Petilil",
		"Phanpy",
		"Phantump",
		"Pheromosa",
		"Phione",
		"Pichu",
		"Pidgeot",
		"Pidgeotto",
		"Pidgey",
		"Pidove",
		"Pignite",
		"Pikachu",
		"Pikipek",
		"Piloswine",
		"Pineco",
		"Pinsir",
		"Piplup",
		"Plusle",
		"Politoed",
		"Poliwag",
		"Poliwhirl",
		"Poliwrath",
		"Ponyta",
		"Poochyena",
		"Popplio",
		"Porygon-Z",
		"Porygon",
		"Porygon2",
		"Primarina",
		"Primeape",
		"Prinplup",
		"Probopass",
		"Psyduck",
		"Pumpkaboo",
		"Pupitar",
		"Purrloin",
		"Purugly",
		"Pyroar",
		"Pyukumuku",
		"Quagsire",
		"Quilava",
		"Quilladin",
		"Qwilfish",
		"Raichu-A",
		"Raichu",
		"Raikou",
		"Ralts",
		"Rampardos",
		"Rapidash",
		"Raticate-A",
		"Raticate",
		"Rattata-A",
		"Rattata",
		"Rayquaza",
		"Regice",
		"Regigigas",
		"Regirock",
		"Registeel",
		"Relicanth",
		"Remoraid",
		"Reshiram",
		"Reuniclus",
		"Rhydon",
		"Rhyhorn",
		"Rhyperior",
		"Ribombee",
		"Riolu",
		"Rockruff",
		"Roggenrola",
		"Roselia",
		"Roserade",
		"Rotom-F",
		"Rotom-L",
		"Rotom-O",
		"Rotom-R",
		"Rotom-W",
		"Rotom",
		"Rowlet",
		"Rufflet",
		"Sableye",
		"Salamence",
		"Salandit",
		"Salazzle",
		"Samurott",
		"Sandile",
		"Sandshrew-A",
		"Sandshrew",
		"Sandslash-A",
		"Sandslash",
		"Sandygast",
		"Sawk",
		"Sawsbuck",
		"Scatterbug",
		"Sceptile",
		"Scizor",
		"Scolipede",
		"Scrafty",
		"Scraggy",
		"Scyther",
		"Seadra",
		"Seaking",
		"Sealeo",
		"Seedot",
		"Seel",
		"Seismitoad",
		"Sentret",
		"Serperior",
		"Servine",
		"Seviper",
		"Sewaddle",
		"Sharpedo",
		"Shaymin-S",
		"Shaymin",
		"Shedinja",
		"Shelgon",
		"Shellder",
		"Shellos-E",
		"Shellos",
		"Shelmet",
		"Shieldon",
		"Shiftry",
		"Shiinotic",
		"Shinx",
		"Shroomish",
		"Shuckle",
		"Shuppet",
		"Sigilyph",
		"Silcoon",
		"Silvally",
		"Simipour",
		"Simisage",
		"Simisear",
		"Skarmory",
		"Skiddo",
		"Skiploom",
		"Skitty",
		"Skorupi",
		"Skrelp",
		"Skuntank",
		"Slaking",
		"Slakoth",
		"Sliggoo",
		"Slowbro",
		"Slowking",
		"Slowpoke",
		"Slugma",
		"Slurpuff",
		"Smeargle",
		"Smoochum",
		"Sneasel",
		"Snivy",
		"Snorlax",
		"Snorunt",
		"Snover",
		"Snubbull",
		"Solgaleo",
		"Solosis",
		"Solrock",
		"Spearow",
		"Spewpa",
		"Spheal",
		"Spinarak",
		"Spinda",
		"Spiritomb",
		"Spoink",
		"Spritzee",
		"Squirtle",
		"Stantler",
		"Staraptor",
		"Staravia",
		"Starly",
		"Starmie",
		"Staryu",
		"Steelix",
		"Steenee",
		"Stoutland",
		"Stufful",
		"Stunfisk",
		"Stunky",
		"Sudowoodo",
		"Suicune",
		"Sunflora",
		"Sunkern",
		"Surskit",
		"Swablu",
		"Swadloon",
		"Swalot",
		"Swampert",
		"Swanna",
		"Swellow",
		"Swinub",
		"Swirlix",
		"Swoobat",
		"Sylveon",
		"Taillow",
		"Talonflame",
		"Tangela",
		"Tangrowth",
		"Tapu Bulu",
		"Tapu Fini",
		"Tapu Koko",
		"Tapu Lele",
		"Tauros",
		"Teddiursa",
		"Tentacool",
		"Tentacruel",
		"Tepig",
		"Terrakion",
		"Throh",
		"Thundurus",
		"Timburr",
		"Tirtouga",
		"Togedemaru",
		"Togekiss",
		"Togepi",
		"Togetic",
		"Torchic",
		"Torkoal",
		"Tornadus",
		"Torracat",
		"Torterra",
		"Totodile",
		"Toucannon",
		"Toxapex",
		"Toxicroak",
		"Tranquill",
		"Trapinch",
		"Treecko",
		"Trevenant",
		"Tropius",
		"Trubbish",
		"Trumbeak",
		"Tsareena",
		"Turtonator",
		"Turtwig",
		"Tympole",
		"Tynamo",
		"Type: Null",
		"Typhlosion",
		"Tyranitar",
		"Tyrantrum",
		"Tyrogue",
		"Tyrunt",
		"Umbreon",
		"Unfezant-F",
		"Unfezant",
		"Unown",
		"Ursaring",
		"Uxie",
		"Vanillish",
		"Vanillite",
		"Vanilluxe",
		"Vaporeon",
		"Venipede",
		"Venomoth",
		"Venonat",
		"Venusaur",
		"Vespiquen",
		"Vibrava",
		"Victini",
		"Victreebel",
		"Vigoroth",
		"Vikavolt",
		"Vileplume",
		"Virizion",
		"Vivillon",
		"Volbeat",
		"Volcanion",
		"Volcarona",
		"Voltorb",
		"Vullaby",
		"Vulpix-A",
		"Vulpix",
		"Wailmer",
		"Wailord",
		"Walrein",
		"Wartortle",
		"Watchog",
		"Weavile",
		"Weedle",
		"Weepinbell",
		"Weezing",
		"Whimsicott",
		"Whirlipede",
		"Whiscash",
		"Whismur",
		"Wigglytuff",
		"Wimpod",
		"Wingull",
		"Wishiwashi",
		"Wobbuffet",
		"Woobat",
		"Wooper",
		"Wormadam-G",
		"Wormadam-S",
		"Wormadam",
		"Wurmple",
		"Wynaut",
		"Xatu",
		"Xerneas",
		"Xurkitree",
		"Yamask",
		"Yanma",
		"Yanmega",
		"Yungoos",
		"Yveltal",
		"Zangoose",
		"Zapdos",
		"Zebstrika",
		"Zekrom",
		"Zigzagoon",
		"Zoroark",
		"Zorua",
		"Zubat",
		"Zweilous",
		"Zygarde",
	}

	return names
}