require_relative 'tmpfile'
require_relative 'path'
require_relative 'cmd'

require_relative 'open/stream'
require_relative 'open/util'
require_relative 'open/remote'
require_relative 'open/lock'

module Open
  module NamedStream
    attr_accessor :filename

    def digest_str
      if Path === filename && ! filename.located?
        filename
      else
        Misc.file_md5(filename)
      end
    end
  end

  def self.get_stream(file, mode = 'r', options = {})
    return file if Open.is_stream?(file)
    return file.stream if Open.has_stream?(file)
    file = file.find if Path === file

    return Open.ssh(file, options) if Open.ssh?(file)
    return Open.wget(file, options) if Open.remote?(file)

    File.open(file, mode)
  end

  def self.file_open(file, grep = false, mode = 'r', invert_grep = false, options = {})
    Open.mkdir File.dirname(file) if mode.include? 'w'

    stream = get_stream(file, mode, options)

    if grep
      grep(stream, grep, invert_grep)
    else
      stream
    end
  end

  def self.file_write(file, content, mode = 'w')
    File.open(file, mode) do |f|
      begin
        f.flock(File::LOCK_EX)
        f.write content 
        f.flock(File::LOCK_UN)
      ensure
        f.close unless f.closed?
      end
    end
  end

  def self.open(file, options = {})
    if IO === file || StringIO === file
      if block_given?
        res = yield file, options
        file.close
        return res
      else
        return file
      end
    end

    options = IndiferentHash.add_defaults options, :noz => false, :mode => 'r'

    mode = IndiferentHash.process_options options, :mode

    options[:noz] = true if mode.include? "w"

    io = file_open(file, options[:grep], mode, options[:invert_grep], options)

    io = unzip(io)   if ((String === file and zip?(file))   and not options[:noz]) or options[:zip]
    io = gunzip(io)  if ((String === file and gzip?(file))  and not options[:noz]) or options[:gzip]
    io = bgunzip(io) if ((String === file and bgzip?(file)) and not options[:noz]) or options[:bgzip]

    io.extend NamedStream
    io.filename = file

    if block_given?
      res = nil
      begin
        res = yield(io)
      rescue DontClose
        res = $!.payload
      rescue Exception
        io.abort $! if io.respond_to? :abort
        io.join if io.respond_to? :join
        raise $!
      ensure
        io.close if io.respond_to? :close and not io.closed?
        io.join if io.respond_to? :join
      end
      res
    else
      io
    end
  end

  def self.read(file, options = {}, &block)
    open(file, options) do |f|
      if block_given?
        res = []
        while not f.eof?
          l = f.gets
          l = Misc.fixutf8(l) unless options[:nofix]
          res << yield(l)
        end
        res
      else
        if options[:nofix]
          f.read
        else
          Misc.fixutf8(f.read)
        end
      end
    end
  end

  def self.write(file, content = nil, options = {})
    options = IndiferentHash.add_defaults options, :mode => 'w'

    file = file.find(options[:where]) if Path === file
    mode = IndiferentHash.process_options options, :mode

    FileUtils.mkdir_p File.dirname(file)

    case
    when block_given?
      begin
        f = File.open(file, mode)
        begin
          yield f
        ensure
          f.close unless f.closed?
        end
      rescue Exception
        FileUtils.rm file if File.exist? file
        raise $!
      end
    when content.nil?
      file_write(file, "", mode)
    when String === content
      file_write(file, content, mode)
    when (IO === content || StringIO === content)
      begin
        File.open(file, mode) do |f| 
          f.flock(File::LOCK_EX)
          while block = content.read(Open::BLOCK_SIZE)
            f.write block
          end
          f.flock(File::LOCK_UN)
        end
      rescue Exception
        FileUtils.rm_rf file if File.exist? file
        raise $!
      end
      content.close unless content.closed?
      content.join if content.respond_to? :join
    else
      raise "Content unknown #{Log.fingerprint content}"
    end

    notify_write(file)
  end
end