smugmug 0.0.1

data/lib/smugmug/album/info.rb

@@ -0,0 +1,131 @@
+# -*- Mode: ruby; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*-
+# $Hg: info.rb,v 03ae0d184343 2007/08/11 07:05:36 boumenot $
+
+require 'forwardable'
+require 'smugmug/base'
+require 'smugmug/table'
+require 'smugmug/images'
+
+module SmugMug
+    module Album
+        class Info < Base
+            include Enumerable 
+
+            extend Forwardable
+            def_delegator :images, :each, :each
+            def_delegator :images, :any?, :any?
+            def_delegator :images, :size, :size
+
+            attr_reader :doc
+
+            def initialize(smugmug, doc)
+                super(smugmug)
+                @doc = doc
+                self.class.table = MethodTable.new(Album.data(), smugmug.account_type, self)
+            end
+
+            ## overrides
+
+            def method
+                return 'smugmug.albums.getInfo'
+            end
+
+            ## methods
+
+            def <=>(rhs)
+                return self.title <=> rhs.title
+            end
+
+            def delete
+                raise RuntimeError, "method is not yet implemented"
+            end
+
+            def images
+                @images = Images.new(@smugmug, self) if @images.nil?
+                return @images
+            end
+            private :images
+
+            def fetch
+                params = {}
+                params[:AlbumID] = album_id
+                # params[:Password] = XXX
+
+                # NOTE:
+                # The initial document is different from the re-fetched document.
+                # --initial--
+                # <album>...</album>
+                # --re-fetched--
+                # <rsp><albums><album>...</album></albums></rsp>
+                return super(params)
+            end
+
+            def sort_by
+            end
+
+            def stats
+                @stats = Stats.new(@smugmug, self) if @stats.nil?
+                return @stats
+            end
+
+            ## accessors
+
+            ## FIXME: needed to prevent infinite recursion
+            def album_id() return self.class.table.get(@doc, :album_id) ; end
+            def title()    return self.class.table.get(@doc, :title)    ; end
+
+            def Album.data
+                return %q{
+<!-- 
+    Unless specified the following defaults are assumed.  
+    
+    - type="string"
+    - xpath="//<field>"
+    - account="standard"
+-->
+<smugmug>
+    <AlbumID         type="int"     xpath="//Album/attribute::id"/>
+    <HighlightID     type="int"     xpath="//Highlight/attribute::id"/>
+    <CategoryID      type="int"     xpath="//Category/attribute::id"/>
+<!--
+    <CategoryTitle                  xpath="//Category/Title"/>
+-->
+    <SubCategoryID   type="int"     xpath="//SubCategory/attribute::id"/>
+    <Position        type="int"/>
+    <SortMethod/>
+    <SortDirection   type="boolean"/>
+    <ImageCount      type="int"/>
+    <Title/>
+    <Keywords/>
+    <Description/>
+    <LastUpdated/>
+    <CommunityID     type="int"     xpath="//Community/attribute::id"/>
+    <Public          type="boolean"/>
+    <Password/>
+    <PasswordHint/>
+    <CanRank         type="boolean"/>
+    <Printable       type="boolean"/>
+    <Filenames       type="boolean"/>
+    <Comments        type="boolean"/>
+    <External        type="boolean"/>
+    <Originals       type="boolean"/>
+    <FamilyEdit      type="boolean"/>
+    <FriendEdit      type="boolean"/>
+    <HideOwner       type="boolean"/>
+    <EXIF            type="boolean"/>
+    <Share           type="boolean"/>
+    <Header          type="boolean" account="power"/>
+    <TemplateID      type="int"     account="power" xpath="//Template/attribute::id"/>
+    <Larges          type="boolean" account="pro"/>
+    <DefaultColor    type="boolean" account="pro"/>
+    <Clean           type="boolean" account="pro"/>
+    <Protected       type="boolean" account="pro"/>
+    <Watermark       type="boolean" account="pro"/>
+    <SmugSearchable  type="boolean"/>
+    <WorldSearchable type="boolean"/>
+</smugmug>
+}
+            end
+        end
+    end
+end

data/lib/smugmug/album/stats.rb

@@ -0,0 +1,31 @@
+# -*- Mode: ruby; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*-
+# $Hg$
+
+require 'smugmug/base'
+
+module SmugMug
+    module Album
+        class Stats < BaseStats
+            attr_reader  :album, :doc
+
+            def initialize(smugmug, _album)
+                @album = _album
+                super(smugmug)
+                @doc = fetch()
+            end
+
+            ## overrides
+
+            def method
+                return 'smugmug.albums.getStats'
+            end
+
+            def fetch
+                params = {}
+                params[:AlbumID] = album.album_id
+
+                return super(params)
+            end
+        end
+    end
+end

data/lib/smugmug/albums.rb

@@ -0,0 +1,39 @@
+# -*- ruby -*-
+# $Hg$
+
+require 'forwardable'
+require 'smugmug/base'
+require 'smugmug/album/info'
+
+module SmugMug
+    class Albums < Base
+        include Enumerable
+
+        extend Forwardable
+        def_delegator :@albums, :each, :each
+        def_delegator :@albums, :any?, :any?
+        def_delegator :@albums, :size, :size
+
+        def initialize(smugmug)
+            super(smugmug)
+            @albums = []
+
+            @doc = fetch().root
+            @doc.elements.each('Albums/Album') do |x| 
+                @albums << Album::Info.new(@smugmug, REXML::Document.new(x.to_s)) 
+                $log.info('Albums#initialize') { "Added the album #{@albums[-1].title}" }
+            end
+        end
+
+        ## overrides
+
+        def method
+            return 'smugmug.albums.get'
+        end
+
+        def method_missing(sym, *args)
+            raise NoMethodError.new("undefined method `#{method}' for " +
+                                    "#{self.inspect}:#{self.class.name}")
+        end
+    end
+end

data/lib/smugmug/base.rb

@@ -0,0 +1,104 @@
+# -*- ruby -*-
+# $Hg: base.rb,v 1a7959f46f5c 2007/08/11 05:56:18 boumenot $
+
+require 'forwardable'
+require 'logger'
+require 'net/https'
+require 'net/httpz'
+require 'rexml/document'
+require 'uri'
+
+module SmugMug
+    class Base
+        extend Forwardable
+        def_delegator :@smugmug, :url, :url
+
+        attr_reader :smugmug
+
+        @table = nil
+
+        class << self
+            attr_reader :table
+            def table=(val)
+                return unless @table.nil?
+                @table = val
+            end
+        end
+
+        def initialize(smugmug)
+            @smugmug = smugmug
+            @heavy   = false
+        end
+
+        ## accessors
+
+        def data
+            raise SmugMugError, "data() must be defined in the derived class!"
+        end
+
+        def method
+            raise SmugMugError, "method() must be defined in the derived class!"
+        end
+
+        ## methods
+
+        def heavy?
+            return @heavy == true
+        end
+
+        def fetch(params={})
+            params[:method]    = method
+            params[:SessionID] = smugmug.session_id
+
+            @heavy = true
+            return smugmug.fetch(params)
+        end
+
+        def method_missing(sym, *args)
+            if sym.to_s[-1] == '='
+                raise NoMethodError.new("undefined method `#{method}' for " +
+                                    "#{self.inspect}:#{self.class.name}")
+            end
+            @doc = fetch unless heavy?
+            return self.class.table.get(@doc, sym)
+        end
+    end
+
+    class BaseStats < Base
+        def initialize(smugmug)
+            super(smugmug)
+            self.class.table = MethodTable.new(BaseStats.data(), smugmug.account_type, self.class)
+        end
+
+        ## methods
+
+        def fetch(params)
+            params[:Month] = Time.now.month
+            params[:Year]  = Time.now.year
+
+            return super(params)
+        end
+
+        def self.data
+            return %q{
+<!-- 
+    Unless specified the following defaults are assumed.  
+    
+    - type="string"
+    - xpath="//<field>"
+    - account="standard"
+    = readonly="true"
+-->
+<smugmug>
+    <Bytes    type="int"/>
+    <Tiny     type="int"/>
+    <Thumb    type="int"/>
+    <Small    type="int"/>
+    <Medium   type="int"/>
+    <Large    type="int"/>
+    <Original type="float"/>
+</smugmug>
+}
+        end
+    end
+end

data/lib/smugmug/cache.rb

@@ -0,0 +1,33 @@
+# -*- Mode: ruby; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*-
+# $Hg$
+
+module SmugMug
+    class Cache
+        def initialize
+            @cache = {}
+        end
+
+        def [](val)
+            return @cache[val]['data']
+        end
+
+        def []=(val, data)
+            @cache[val] ||= {}
+
+            @cache[val]['data'] = data
+            @cache[val]['time'] = Time.now.httpdate
+        end
+
+        def cached?(val)
+            return @cache.has_key?(val)
+        end
+
+        def last_modified(val)
+            return @cache[val]['time']
+        end
+
+        def invalidate(val)
+            @cache.delete(val)
+        end
+    end
+end

data/lib/smugmug/config.rb

@@ -0,0 +1,48 @@
+# -*- ruby -*-
+# $Hg$
+
+require 'smugmug/util.rb'
+
+module SmugMug
+    class Config
+        attr_accessor :AccountType, :EmailAddress, :PasswordHash, :UserID
+
+        FILENAME = '~/.smugmugrc'
+        VARIABLES = %w{AccountType EmailAddress PasswordHash UserID}
+
+        def initialize
+            @filename = ENV.has_key?('SMUGMUG_RC') ? ENV['SMUGMUG_RC'] : FILENAME
+            restore()
+        end
+
+        ## methods
+
+        def save
+            doc = REXML::Element.new('smugmug')
+            VARIABLES.each do |var|
+                doc << REXML::Element.new(var).add_text(self.send(var.to_sym))
+            end
+            File.write(filename(), doc.to_s)
+        end
+
+        ## accesors
+
+        private
+
+        def filename
+          return File.expand_path(@filename)
+        end
+
+        def restore
+            begin
+              doc = REXML::Document.new(IO.read(filename()))
+              VARIABLES.each do |var|
+                val = (doc.root.elements[var].nil?) ? nil : doc.root.elements[var].text
+                self.instance_variable_set("@#{var}", val)
+              end  
+            rescue
+                $log.info('Config') { "#{filename} does not exist, skipping restore" }
+            end
+        end
+    end
+end

data/lib/smugmug/image/exif.rb

@@ -0,0 +1,72 @@
+# -*- ruby -*-
+# $Hg: exif.rb,v 1f3d53375e7c 2007/08/11 06:35:02 boumenot $
+
+require 'smugmug/base'
+require 'smugmug/table'
+
+module SmugMug
+    module Image
+        class Exif < Base
+            attr_reader :image
+            def initialize(smugmug, image)
+                super(smugmug)
+                @image = image
+                self.class.table = MethodTable.new(Exif.data(), smugmug.account_type, self)
+            end
+
+            ## overrides
+
+            def method
+                return 'smugmug.images.getEXIF'
+            end
+
+            def fetch
+                params = {}
+                # params[:Password] = image.album.password
+                params[:ImageID]  = image.image_id
+
+                return super(params)
+            end
+
+            ## methods
+
+            ## accessors
+
+            def Exif.data
+                return %q{
+<smugmug>
+    <ImageID         type="int"     xpath="//Image/attribute::id"/>
+    <DateTime/>
+    <DateTimeOriginal/>
+    <DateTimeDigitized/>
+    <Make/>
+    <Model/>
+    <ExposureTime/>
+    <Aperture/>
+    <ISO             type="int"/>
+    <FocalLength/>
+    <FocalLength35mmFilm/>
+    <CCDWidth/>
+    <CompressedBitsPerPixel/>
+    <Flash           type="int"/>
+    <Metering        type="int"/>
+    <ExposureProgram type="int"/>
+    <ExposureBiasValue/>
+    <ExposureMode    type="int"/>
+    <LightSource     type="int"/>
+    <WhiteBalance    type="int"/>
+    <DigitalZoomRatio/>
+    <Contrast        type="int"/>
+    <Saturation      type="int"/>
+    <Sharpness       type="int"/>
+    <SubjectDistance/>
+    <SubjectDistanceRange/>
+    <SensingMethod   type="int"/>
+    <ColorSpace/>
+    <Brightness/>
+</smugmug>
+}
+            end
+        end
+    end
+end