@@ -21,7 +21,7 @@ TODO require 'rubygems' require 'stammer' - stammer = Stammer.new('user@example.com', 'password') + stammer = Stammer::Client.new('user@example.com', 'password') messages = stammer.messages sent_messages = stammer.messages('sent') README.txt @@ -1,9 +1,5 @@ # $Id$ -# Equivalent to a header guard in C/C++ -# Used to prevent the class/module from being loaded more than once -unless defined? Stammer - module Stammer # :stopdoc: @@ -49,8 +45,4 @@ module Stammer end # module Stammer -Stammer.require_all_libs_relative_to __FILE__ - -end # unless defined? - -# EOF +Stammer.require_all_libs_relative_to __FILE__ \ No newline at end of file lib/stammer.rb @@ -4,11 +4,16 @@ require 'net/https' module Stammer class Client - def initialize(user, password, secret = nil, client = nil, format = 'json') + + ACCEPTABLE_SUBSETS = [:all, :sent, :received, :following] + FORMAT = 'json' + HEADERS = {'User-Agent' => 'Stammer v0.0.3', 'Accept' => "text/#{FORMAT}"} + API_URL = "https://yammer.com/api/v1" + + def initialize(user, password, secret = nil, client = nil) @user = user @password = password - @format = format - + unless !secret || !client @secret = secret @client = client @@ -18,8 +23,10 @@ module Stammer end # TODO: pagination via older_than/newer_than - def messages(subset = nil) - MessageList.new(send_request('messages', subset)) + def messages(subset = :all) + raise ArgumentError.new("Subset must be in [#{ACCEPTABLE_SUBSETS.join(', ')}]") unless ACCEPTABLE_SUBSETS.include?(subset) + subset = nil if subset == :all + MessageList.new(send_request('messages', subset.to_s)) end # TODO: pagination via page @@ -30,20 +37,39 @@ module Stammer def tags(id = nil) TagList.new(send_request('tags', id)) end + + def update_status(status) + raise ArgumentError.new("must update with a status") if status.to_s == "" + url = URI.parse(API_URL + "/messages") + conn = create_connection(url) + res = conn.start do |http| + req = Net::HTTP::Post.new(url.path, HEADERS) + req.basic_auth(@user, @password) + req.form_data = {'body' => status} + http.request(req) + end + return res.is_a?(Net::HTTPCreated) + end private + def send_request(*args) - url = URI.parse("https://yammer.com/api/v1/#{args.compact.join('/')}.#{@format}") - - conn = Net::HTTP.new(url.host, url.port) - conn.use_ssl = true - conn.verify_mode = OpenSSL::SSL::VERIFY_NONE - res = conn.start { |http| - req = Net::HTTP::Get.new(url.path, {'User-Agent' => 'Stammer v0.0.1', 'Accept' => 'text/json'}) + url = URI.parse("#{API_URL}/#{args.compact.join('/')}.#{FORMAT}") + puts "grabbing #{url.to_s}" + conn = create_connection(url) + res = conn.start do |http| + req = Net::HTTP::Get.new(url.path, HEADERS) req.basic_auth(@user, @password) http.request(req) - } + end JSON.load(res.body) end + + def create_connection(url) + conn = Net::HTTP.new(url.host, url.port) + conn.use_ssl = true + conn.verify_mode = OpenSSL::SSL::VERIFY_NONE + conn + end end end \ No newline at end of file lib/stammer/client.rb @@ -15,7 +15,7 @@ module Stammer def ytb? !(self.plain =~ /^(today|blocker|yesterday)/).nil? end - + def plain self['body']['plain'] end @@ -23,5 +23,7 @@ module Stammer def created Time.parse(self['created_at']) end + + alias :body :plain end end \ No newline at end of file lib/stammer/message.rb @@ -1,6 +1,6 @@ Gem::Specification.new do |s| s.name = "stammer" - s.version = "0.0.2" + s.version = "0.0.3" s.date = "2008-09-19" s.summary = "Super-simple wrapper for the Yammer API" s.email = "ben.scofield@viget.com" stammer.gemspec 36fe7f447b7f0bf1ecaa52518609575cd4b25c42 Terry Heath theath@kapwnt.local http://github.com/bscofield/stammer/commit/86564830d99e801aacb9e294296b6a189377ff0d 86564830d99e801aacb9e294296b6a189377ff0d 2008-10-01T14:28:12-07:00 2008-10-01T14:28:12-07:00 Added a limiter around supported API messages gets and added a way to update status. 5924fc0feae7ceeb2f7d56d10932fc5632541eda Terry Heath theath@kapwnt.local