vpim2 0.0.1

data/lib/vpim/icalendar.rb

@@ -0,0 +1,381 @@
+=begin
+  Copyright (C) 2008 Sam Roberts
+
+  This library is free software; you can redistribute it and/or modify it
+  under the same terms as the ruby language itself, see the file COPYING for
+  details.
+=end
+
+require "enumerator"
+
+require 'vpim/rfc2425'
+require 'vpim/dirinfo'
+require 'vpim/rrule'
+require 'vpim/vevent'
+require 'vpim/vtodo'
+require 'vpim/vjournal'
+require 'vpim/vpim'
+
+module Vpim
+  # An iCalendar.
+  #
+  # A Calendar is some meta-information followed by a sequence of components.
+  #
+  # Defined components are Event, Todo, Freebusy, Journal, and Timezone, each
+  # of which are represented by their own class, though they share many
+  # properties in common. For example, Event and Todo may both contain
+  # multiple Alarm components.
+  #
+  # = Reference
+  #
+  # The iCalendar format is specified by a series of IETF documents:
+  #
+  # - link:rfc2445.txt: Internet Calendaring and Scheduling Core Object Specification
+  # - link:rfc2446.txt: iCalendar Transport-Independent Interoperability Protocol
+  #   (iTIP) Scheduling Events, BusyTime, To-dos and Journal Entries
+  # - link:rfc2447.txt: iCalendar Message-Based Interoperability Protocol
+  #
+  # = iCalendar and vCalendar
+  #
+  # iCalendar files have VERSION:2.0 and vCalendar have VERSION:1.0.  iCalendar
+  # (RFC 2445) is based on vCalendar, but but is not very compatible.  While
+  # much appears to be similar, the recurrence rule syntax is completely
+  # different.
+  #
+  # iCalendars are usually transmitted in files with <code>.ics</code>
+  # extensions.
+  class Icalendar
+    # FIXME do NOT do this!
+    include Vpim
+
+    # Regular expression strings for the EBNF of RFC 2445
+    module Bnf #:nodoc:
+      # dur-value = ["+" / "-"] "P" [ 1*DIGIT "W" ] [ 1*DIGIT "D" ] [ "T" [ 1*DIGIT "H" ]  [ 1*DIGIT "M" ] [ 1*DIGIT "S" ] ]
+      DURATION = '([-+])?P(\d+W)?(\d+D)?T?(\d+H)?(\d+M)?(\d+S)?'
+    end
+
+    private_class_method :new
+
+    # Create a new Icalendar object from +fields+, an array of
+    # DirectoryInfo::Field objects.
+    #
+    # When decoding Calendar data, you would usually use Icalendar.decode(),
+    # which decodes the data into the field arrays, and calls this method
+    # for each Calendar it finds.
+    def initialize(fields) #:nodoc:
+      # seperate into the outer-level fields, and the arrays of component
+      # fields
+      outer, inner = Vpim.outer_inner(fields)
+
+      # Make a dirinfo out of outer, and check its an iCalendar
+      @properties = DirectoryInfo.create(outer)
+      @properties.check_begin_end('VCALENDAR')
+
+      @components = []
+
+      # could use #constants instead of this
+      factory = {
+        'VEVENT' => Vevent,
+        'VTODO' => Vtodo,
+        'VJOURNAL' => Vjournal,
+        # TODO - VTIMEZONE
+      }
+
+      inner.each do |component|
+        name = component.first.value
+
+        if klass = factory[name]
+          @components << klass.new(component)
+        end
+      end
+    end
+
+    # Add an event to this calendar.
+    #
+    # Yields an event maker, Icalendar::Vevent::Maker.
+    def add_event(&block) #:yield:event
+      push Vevent::Maker.make( &block )
+    end
+
+    # TODO add_todo, add_journal
+
+=begin
+TODO
+    # Allows customization of calendar creation.
+    class Maker
+      def initialize #:nodoc:
+        @prodid = Vpim::PRODID
+      end
+
+      attr :prodid
+    end
+=end
+
+    # The producer ID defaults to Vpim::PRODID but you can set it to something
+    # specific to your application.
+    def Icalendar.create2(producer = Vpim::PRODID) #:yield: self
+      # FIXME - make the primary API
+      di = DirectoryInfo.create( [ DirectoryInfo::Field.create('VERSION', '2.0') ], 'VCALENDAR' )
+
+      di.push_unique DirectoryInfo::Field.create('PRODID', producer.to_str)
+      di.push_unique DirectoryInfo::Field.create('CALSCALE', "Gregorian")
+
+      cal = new(di.to_a)
+
+      if block_given?
+        yield cal
+      end
+
+      cal
+    end
+
+    # Create a new Icalendar object with the minimal set of fields for a valid
+    # Calendar. If specified, +fields+ must be an array of
+    # DirectoryInfo::Field objects to add. They can override the the default
+    # Calendar fields, so, for example, this can be used to set a custom PRODID field.
+    def Icalendar.create(fields=[])
+      di = DirectoryInfo.create( [ DirectoryInfo::Field.create('VERSION', '2.0') ], 'VCALENDAR' )
+
+      DirectoryInfo::Field.create_array(fields).each { |f| di.push_unique f }
+
+      di.push_unique DirectoryInfo::Field.create('PRODID',   Vpim::PRODID)
+      di.push_unique DirectoryInfo::Field.create('CALSCALE', "Gregorian")
+
+      new(di.to_a)
+    end
+
+    # Create a new Icalendar object with a protocol method of REPLY.
+    #
+    # Meeting requests, and such, are Calendar containers with a protocol
+    # method of REQUEST, and contains some number of Events, Todos, etc.,
+    # that may need replying to. In order to reply to any of these components
+    # of a request, you must first build a Calendar object to hold your reply
+    # components.
+    #
+    # This method builds the reply Calendar, you then will add to it replies
+    # to the specific components of the request Calendar that you are replying
+    # to. If you have any particular fields that you want to be in the
+    # Calendar, other than the defaults, then can be supplied as +fields+, an
+    # array of Field objects.
+    def Icalendar.create_reply(fields=[])
+      fields << DirectoryInfo::Field.create('METHOD', 'REPLY')
+
+      Icalendar.create(fields)
+    end
+
+    # Used during encoding.
+    def fields # :nodoc:
+      f = @properties.to_a
+      last = f.pop
+      # Use of #each means we won't encode components in our View, but also
+      # that we won't encode timezones... but we don't decode/support timezones
+      # anyhow, so fix later.
+      each { |c| f << c.fields }
+      f.push last
+    end
+
+    # Encode the Calendar as a string. The width is the maximum width of the
+    # encoded lines, it can be specified, but is better left to the default.
+    def encode(width=nil)
+      # We concatenate the fields of all objects, create a DirInfo, then
+      # encode it.
+      di = DirectoryInfo.create(self.fields.flatten)
+      di.encode(width)
+    end
+
+    alias to_s encode
+
+    # Push a calendar component onto the calendar.
+    def push(component)
+      case component
+        when Vevent, Vtodo, Vjournal
+          @components << component
+        else
+          raise ArgumentError, "can't add a #{component.type} to a calendar"
+      end
+      self
+    end
+
+    alias :<< :push
+
+    # Check if the protocol method is +method+
+    def protocol?(method)
+      Vpim::Methods.casecmp?(protocol, method)
+    end
+
+    def Icalendar.decode_duration(str) #:nodoc:
+      unless match = %r{\s*#{Bnf::DURATION}\s*}.match(str)
+        raise InvalidEncodingError, "duration not valid (#{str})"
+      end
+      dur = 0
+
+      # Remember: match[0] is the whole match string, match[1] is $1, etc.
+
+      # Week
+      if match[2]
+        dur = match[2].to_i
+      end
+      # Days
+      dur *= 7
+      if match[3]
+        dur += match[3].to_i
+      end
+      # Hours
+      dur *= 24
+      if match[4]
+        dur += match[4].to_i
+      end
+      # Minutes
+      dur *= 60
+      if match[5]
+        dur += match[5].to_i
+      end
+      # Seconds
+      dur *= 60
+      if match[6]
+        dur += match[6].to_i
+      end
+
+      if match[1] && match[1] == '-'
+        dur = -dur
+      end
+
+      dur
+    end
+
+    # Decode iCalendar data into an array of Icalendar objects.
+    #
+    # Since iCalendars are self-delimited (by a BEGIN:VCALENDAR and an
+    # END:VCALENDAR), multiple iCalendars can be concatenated into a single
+    # file.
+    #
+    # cal must be String or IO, or implement #each by returning
+    # each line in the input as those classes do.
+    def Icalendar.decode(cal, e = nil)
+      entities = Vpim.expand(Vpim.decode(cal))
+
+      # Since all iCalendars must have a begin/end, the top-level should
+      # consist entirely of entities/arrays, even if its a single iCalendar.
+      if entities.detect { |e| ! e.kind_of? Array }
+        raise "Not a valid iCalendar"
+      end
+
+      calendars = []
+
+      entities.each do |e|
+        calendars << new(e)
+      end
+
+      calendars
+    end
+
+    # The iCalendar version multiplied by 10 as an Integer. iCalendar must have
+    # a version of 20, and vCalendar must have a version of 10.
+    def version
+      v = @properties['VERSION']
+
+      unless v
+        raise InvalidEncodingError, "Invalid calendar, no version field!"
+      end
+
+      v = v.to_f * 10
+      v = v.to_i
+    end
+
+    # The value of the PRODID field, an unstructured string meant to
+    # identify the software which encoded the Calendar data.
+    def producer
+      #f = @properties.field('PRODID')
+      #f && f.to_text
+      @properties.text('PRODID').first
+    end
+
+    # The value of the METHOD field. Protocol methods are used when iCalendars
+    # are exchanged in a calendar messaging system, such as iTIP or iMIP. When
+    # METHOD is not specified, the Calendar object is merely being used to
+    # transport a snapshot of some calendar information; without the intention
+    # of conveying a scheduling semantic.
+    #
+    # Note that this method can't be called +method+, thats already a method of
+    # Object.
+    def protocol
+      m = @properties['METHOD']
+      m ? m.upcase : m
+    end
+
+    # The value of the CALSCALE: property, or "GREGORIAN" if CALSCALE: is not
+    # present.
+    #
+    # This is of academic interest only. There aren't any other calendar scales
+    # defined, and given that its hard enough just dealing with Gregorian
+    # calendars, there probably won't be.
+    def calscale
+      (@properties['CALSCALE'] || 'GREGORIAN').upcase
+    end
+
+    # The array of all supported calendar components. If a class is provided,
+    # return only the components of that class.
+    #
+    # If a block is provided, yield the components instead of returning them.
+    #
+    # Examples:
+    #   calendar.components(Vpim::Icalendar::Vevent)
+    #   => array of all calendar components
+    #
+    #   calendar.components(Vpim::Icalendar::Vtodo) {|c| c... }
+    #   => yield all todo components
+    #
+    #   calendar.components {|c| c... }
+    #   => yield all components
+    #
+    # Note - use of this is mildly deprecated in favour of #each, #events,
+    # #todos, #journals because those won't return timezones, and will return
+    # Enumerators if called without a block.
+    def components(klass=Object) #:yields:component
+      klass ||= Object
+
+      unless block_given?
+        return @components.select{|c| klass === c}.freeze
+      end
+
+      @components.each do |c|
+        if klass === c
+          yield c
+        end
+      end
+      self
+    end
+
+    include Enumerable
+
+    # Enumerate the top-level calendar components. Yields them if a block
+    # is provided, otherwise returns an Enumerator.
+    #
+    # This skips components that are only internally meaningful to iCalendar,
+    # such as timezone definitions.
+    def each(klass=nil, &block) # :yield: component
+      unless block
+        return Enumerable::Enumerator.new(self, :each, klass)
+      end
+      components(klass, &block)
+    end
+
+    # Short-hand for #each(Icalendar::Vevent).
+    def events(&block) #:yield: Vevent
+      each(Icalendar::Vevent, &block)
+    end
+
+    # Short-hand for #each(Icalendar::Vtodo).
+    def todos(&block) #:yield: Vtodo
+      each(Icalendar::Vtodo, &block)
+    end
+
+    # Short-hand for #each(Icalendar::Vjournal).
+    def journals(&block) #:yield: Vjournal
+      each(Icalendar::Vjournal, &block)
+    end
+
+  end
+
+end
+

data/lib/vpim/maker/vcard.rb

@@ -0,0 +1,16 @@
+=begin
+  Copyright (C) 2008 Sam Roberts
+
+  This library is free software; you can redistribute it and/or modify it
+  under the same terms as the ruby language itself, see the file COPYING for
+  details.
+=end
+
+require 'vpim/vcard'
+
+module Vpim
+  module Maker #:nodoc:backwards compat
+    Vcard = Vpim::Vcard::Maker #:nodoc:backwards compat
+  end
+end
+

data/lib/vpim/property/base.rb

@@ -0,0 +1,193 @@
+=begin
+  Copyright (C) 2008 Sam Roberts
+
+  This library is free software; you can redistribute it and/or modify it
+  under the same terms as the ruby language itself, see the file COPYING for
+  details.
+=end
+
+module Vpim
+  class Icalendar
+    module Set #:nodoc:
+      module Util #:nodoc:
+        # TODO - rename module to Private?
+
+        def rm_all(name)
+          rm = @comp.properties.select { |f| f.name? name }
+          rm.each { |f| @comp.properties.delete(f) }
+        end
+
+        def set_token(name, allowed, default, value) #:nodoc:
+          value = value.to_str
+          unless allowed.include?(value)
+            raise Vpim::Unencodeable, "Invalid #{name} value '#{value}'"
+          end
+          rm_all(name)
+          unless value == default
+            @comp.properties.push Vpim::DirectoryInfo::Field.create(name, value)
+          end
+        end
+
+        def field_create(name, value, default_value_type = nil, value_type = nil, params = {})
+          if value_type && value_type != default_value_type
+            params['VALUE'] = value_type
+          end
+          Vpim::DirectoryInfo::Field.create(name, value, params)
+        end
+
+        def set_date_or_datetime(name, default, value)
+          f = nil
+          case value
+          when Date
+            f = field_create(name, Vpim.encode_date(value), default, 'DATE')
+          when Time
+            f = field_create(name, Vpim.encode_date_time(value), default, 'DATE-TIME')
+          else
+            raise Vpim::Unencodeable, "Invalid #{name} value #{value.inspect}"
+          end
+          rm_all(name)
+          @comp.properties.push(f)
+        end
+
+        def set_datetime(name, value)
+          f = field_create(name, Vpim.encode_date_time(value))
+          rm_all(name)
+          @comp.properties.push(f)
+        end
+
+        def set_text(name, value)
+          f = field_create(name, Vpim.encode_text(value))
+          rm_all(name)
+          @comp.properties.push(f)
+        end
+
+        def set_text_list(name, value)
+          f = field_create(name, Vpim.encode_text_list(value))
+          rm_all(name)
+          @comp.properties.push(f)
+        end
+
+        def set_integer(name, value)
+          value = value.to_int.to_s
+          f = field_create(name, value)
+          rm_all(name)
+          @comp.properties.push(f)
+        end
+
+        def add_address(name, value)
+          f = value.encode(name)
+          @comp.properties.push(f)
+        end
+
+        def set_address(name, value)
+          rm_all(name)
+          add_address(name, value)
+        end
+
+      end
+    end
+
+    module Property #:nodoc:
+
+      # FIXME - rename Base to Util
+      module Base #:nodoc:
+        # Value of first property with name +name+
+        def propvalue(name) #:nodoc:
+          prop = @properties.detect { |f| f.name? name }
+          if prop
+            prop = prop.value
+          end
+          prop
+        end
+
+        # Array of values of all properties with name +name+
+        def propvaluearray(name) #:nodoc:
+          @properties.select{ |f| f.name? name }.map{ |p| p.value }
+        end
+
+
+        def propinteger(name) #:nodoc:
+          prop = @properties.detect { |f| f.name? name }
+          if prop
+            prop = Vpim.decode_integer(prop.value)
+          end
+          prop
+        end
+
+        def proptoken(name, allowed, default_token = nil) #:nodoc:
+          prop = propvalue(name)
+
+          if prop
+            prop = prop.to_str.upcase
+            unless allowed.include?(prop)
+              raise Vpim::InvalidEncodingError, "Invalid #{name} value '#{prop}'"
+            end
+          else
+            prop = default_token
+          end
+
+          prop
+        end
+
+        # Value as DATE-TIME or DATE of object of first property with name +name+
+        def proptime(name) #:nodoc:
+          prop = @properties.detect { |f| f.name? name }
+          if prop
+            prop = prop.to_time.first
+          end
+          prop
+        end
+
+        # Value as TEXT of first property with name +name+
+        def proptext(name) #:nodoc:
+          prop = @properties.detect { |f| f.name? name }
+          if prop
+            prop = prop.to_text
+          end
+          prop
+        end
+
+        # Array of values as TEXT of all properties with name +name+
+        def proptextarray(name) #:nodoc:
+          @properties.select{ |f| f.name? name }.map{ |p| p.to_text }
+        end
+
+        # Array of values as TEXT list of all properties with name +name+
+        def proptextlistarray(name) #:nodoc:
+          @properties.select{ |f| f.name? name }.map{ |p| Vpim.decode_text_list(p.value_raw) }.flatten
+        end
+
+        # Duration has "almost" the same definition for Event and Todo
+        def propduration(endfield)
+          dur = @properties.field 'DURATION'
+          dte = @properties.field endfield
+
+          if !dur
+            return nil unless dte
+
+            b = dtstart
+            e = send(endfield.downcase.to_sym)
+
+            return (e - b).to_i
+          end
+
+          Icalendar.decode_duration(dur.value_raw)
+        end
+
+        def propend(endfield)
+          dte = @properties.field endfield.to_s.upcase
+          if dte
+            dte.to_time.first
+          elsif duration
+            dtstart + duration
+          else
+            nil
+          end
+        end
+
+
+      end
+    end
+  end
+end
+