# Cookies
module Watir
  class Browser
    # Read cookies from Mozilla cookies.txt-style IO stream
    #
    # @param file [IO,String]
    #
    def load_cookies(file)
      now = ::Time.now

      io = case file
      when String
        open(file)
      else
        file
      end

      io.each_line do |line|
        line.chomp!
        line.gsub!(/#.+/, '')
        fields = line.split("\t")

        next if fields.length != 7

        name, value, domain, for_domain, path, secure, version = fields[5], fields[6], 
          fields[0], (fields[1] == "TRUE"), fields[2], (fields[3] == "TRUE"), 0

        expires_seconds = fields[4].to_i
        expires = (expires_seconds == 0) ? nil : ::Time.at(expires_seconds)
        next if expires and (expires < now)

        cookies.add(name, value, domain: domain, path: path, expires: expires, secure: secure)
      end

      io.close if String === file

      self
    end

    # Write cookies to Mozilla cookies.txt-style IO stream
    #
    # @param file [IO,String]
    def dump_cookies(file)
      io = case file
      when String
        open(file, "w")
      else
        file
      end

      cookies.to_a.each do |cookie|
        io.puts([
          cookie[:domain],
          "FALSE", # for_domain
          cookie[:path],
          cookie[:secure] ? "TRUE" : "FALSE",
          cookie[:expires].to_i.to_s,
          cookie[:name],
          cookie[:value]
        ].join("\t"))
      end

      io.close if String === file

      self
    end
  end
end

# Element
module Watir
  class Element
    # quick set value.
    #
    # @example
    #
    #  form = browser.form
    #  form.quick_set("//input[@name='value']", "hello")
    #  form.quick_set("//input[@name='check']", true)
    #  form.quick_set("//select[@name='foo']", "Bar")
    #  form.quick_set("//textarea[@name='foo']", "bar")
    #
    def quick_set(selector, value)
      elem = element(xpath: selector).to_subtype
      case elem.tag_name
      when "input"
        elem.set value
      when "select"
        elem.select value
      when "textarea"
        elem.set value
      end
    end
  end
end