class StoryKey::Console < Thor
  package_name "StoryKey"

  desc "new [BITSIZE]",
       <<~TEXT
         Create a new key/story (default #{StoryKey::DEFAULT_BITSIZE} bits, max #{StoryKey::MAX_BITSIZE})
       TEXT
  option :image,
         desc: "Whether to generate an image (requires OpenAI key and ImageMagick)",
         type: :boolean,
         aliases: "-i"

  def new(bitsize = StoryKey::DEFAULT_BITSIZE)
    key, story = StoryKey.generate(bitsize: bitsize.to_i)
    puts story_str(key, story)
    print_image_path(story.tokenized, story.phrases) if options[:image]
  rescue StoryKey::KeyTooLarge
    quit "Key too large"
  end

  desc "encode [KEY]",
       "Encode a key passed as an argument or from a file"
  option :file,
         desc: "File containing key",
         aliases: "-f"
  option :format,
         desc: "Format of key",
         enum: %w[base58 hex bin dec],
         default: "base58"
  option :image,
         desc: "Whether to generate an image (requires OpenAI key and ImageMagick)",
         type: :boolean,
         aliases: "-i"

  def encode(key = nil)
    key ||= File.read(options[:file])
    story = StoryKey.encode(key:, format: options[:format])
    puts story_str(key, story)
    print_image_path(story.tokenized, story.phrases) if options[:image]
  rescue StoryKey::InvalidFormat
    quit "Invalid format"
  rescue StoryKey::KeyTooLarge
    quit "Key too large"
  rescue Errno::ENOENT
    quit "Invalid file specified"
  end

  desc "decode [STORY]",
       "Decode a story passed as an argument or from a file"
  option :file,
         desc: "File containing story",
         aliases: "-f"
  option :format,
         desc: "Format of key",
         enum: %w[base58 hex bin dec],
         default: "base58"

  def decode(story = nil)
    story ||= File.read(options[:file])
    format ||= options[:format]
    puts StoryKey.decode(story:, format:)
  rescue StoryKey::InvalidVersion
    puts "Invalid version"
    exit
  rescue StoryKey::InvalidChecksum, StoryKey::InvalidWord
    puts "Invalid story"
    exit
  end

  desc "recover", "Decode a story interactively"
  def recover
    StoryKey.recover
  end

  private

  def print_image_path(seed, phrases)
    puts "Generating image..."
    image_path = StoryKey::ImageGenerator.call(seed:, phrases:)
    puts "No panels generated - check your OpenAI key" if image_path.empty?
    puts "#{titleize('Image')} #{image_path}"
  end

  def quit(msg)
    puts msg
    exit
  end

  def titleize(str)
    "\e[44m#{str}:\e[0m"
  end

  def story_str(key, story)
    [
      titleize("Key"),
      key,
      titleize("Story"),
      story.humanized,
      titleize("Seed Phrase"),
      story.tokenized
    ].join("\n")
  end
end