#
# Copyright (C) 2010  Red Hat, Inc.
#
# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements.  See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.  The
# ASF licenses this file to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance with the
# License.  You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
# License for the specific language governing permissions and limitations
# under the License.

#Windows Azure (WAZ) gem at http://github.com/johnnyhalife/waz-storage
require 'waz-blobs'
require 'deltacloud/base_driver'
module Deltacloud
  module Drivers
    module Azure

class AzureDriver < Deltacloud::BaseDriver

  def supported_collections; [:buckets]
  end

#--
# Buckets
#--
  def buckets(credentials, opts)
    buckets = []
    azure_connect(credentials)
    safely do
      WAZ::Blobs::Container.list.each do |waz_container|
        buckets << convert_container(waz_container)
      end
    end
    buckets = filter_on(buckets, :id, opts)
  end

#--
# Create bucket
#--
  def create_bucket(credentials, name, opts)
    bucket = nil
    azure_connect(credentials)
    safely do
      waz_container = WAZ::Blobs::Container.create(name)
      bucket = convert_container(waz_container)
    end
    bucket
  end

#--
# Delete bucket
#--
  def delete_bucket(credentials, name, opts)
    azure_connect(credentials)
    safely do
      WAZ::Blobs::Container.find(name).destroy!
    end
  end

#--
# Blobs
#--
  def blobs(credentials, opts)
    blob_list = []
    azure_connect(credentials)
    safely do
      the_bucket = WAZ::Blobs::Container.find(opts['bucket'])
      the_bucket.blobs.each do |waz_blob|
        blob_list << convert_blob(waz_blob)
      end
    end
    blob_list = filter_on(blob_list, :id, opts)
    blob_list
  end

  def blob_data(credentials, bucket_id, blob_id, opts)
    azure_connect(credentials)
    # WAZ get blob data methods cant accept blocks for 'streaming'... FIXME
      yield WAZ::Blobs::Container.find(bucket_id)[blob_id].value
  end

  private

  def azure_connect(credentials)
    options = {:account_name => credentials.user, :access_key => credentials.password}
    safely do
      WAZ::Storage::Base.establish_connection!(options)
    end
  end

  def convert_container(waz_container)
    blob_list = []
    waz_container.blobs.each do |blob|
      blob_list << blob.name
    end
    Bucket.new({ :id => waz_container.name,
                    :name => waz_container.name,
                    :size => blob_list.size,
                    :blob_list => blob_list
                  })
  end

  def convert_blob(waz_blob)
    url = waz_blob.url.split('/')
    bucket = url[url.length-2] #FIXME
    Blob.new({   :id => waz_blob.name,
                 :bucket => bucket,
                 :content_length => waz_blob.metadata[:content_length],
                 :content_type => waz_blob.metadata[:content_type],
                 :last_modified => waz_blob.metadata[:last_modified]
              })
  end


end

    end #module Azure
  end #module Drivers
end #module Deltacloud