spotlight 0.0.2 → 0.0.3

data/README.rdoc

@@ -9,15 +9,24 @@ Description goes here.
 === Finds files matching a given query
 
  query = Spotlight::Query.new('kMDItemKind = "JPEG image" && kMDItemPixelHeight > 2000')
+ query.scopes << '/Users/youpy/'
  query.execute.map {|item|
    puts item.get(:kMDItemPath)
  }
 
-=== Gets the metadata attributes for the specified file
+=== Creates query from .savedSearch
+
+ query = Spotlight::Query.from_saved_search('foo.savedSearch')
+
+=== Gets the metadata attribute for the specified file
 
  item = Spotlight::Item.new('/Users/youpy/foo')
  item.get(:kMDItemKind)
 
+=== Gets the metadata attribute names for the specified file
+
+ item.attribute_names
+
 == Note on Patches/Pull Requests
  
 * Fork the project.

data/ext/md_item_native/md_item_native.c

@@ -9,7 +9,73 @@ struct ItemObject {
 CFStringRef CString2CFString(char *str) {
   return CFStringCreateWithCString(kCFAllocatorDefault,
                                    str,
-                                   CFStringGetSystemEncoding());
+                                   kCFStringEncodingUTF8);
+}
+
+VALUE CFString2RString(CFStringRef str) {
+  int stringSize;
+  char *tmpptr;
+  VALUE result;
+
+  stringSize = CFStringGetMaximumSizeForEncoding(CFStringGetLength(str), kCFStringEncodingUTF8) + 1;
+  tmpptr = (char *)malloc(sizeof(char) * stringSize);
+  CFStringGetCString(str, tmpptr, stringSize, kCFStringEncodingUTF8);
+  result = rb_str_new2(tmpptr);
+  free(tmpptr);
+
+  return result;
+}
+
+VALUE CFType2Value(CFTypeRef thing) {
+  int i;
+  double doubleValue;
+  CFNumberType numberType;
+  CFAbsoluteTime timeNow;
+  CFGregorianDate current;
+  VALUE result;
+
+/*   CFStringRef stringValue = CFCopyDescription(thing); */
+/*   rb_p(CFString2RString(stringValue)); */
+/*   CFRelease(stringValue); */
+
+  if(CFGetTypeID(thing) == CFStringGetTypeID()) {
+    result = CFString2RString((CFStringRef)thing);
+  } else if(CFGetTypeID(thing) == CFNumberGetTypeID()) {
+    numberType = CFNumberGetType((CFNumberRef)thing);
+    switch(numberType) {
+    case kCFNumberFloat64Type:
+      CFNumberGetValue((CFNumberRef)thing, numberType, &doubleValue);
+      result = rb_float_new(doubleValue);
+      break;
+    case kCFNumberSInt32Type:
+    case kCFNumberSInt64Type:
+    default:
+      CFNumberGetValue((CFNumberRef)thing, kCFNumberDoubleType, &doubleValue);
+      result = INT2FIX((double)doubleValue);
+      break;
+    }
+  } else if(CFGetTypeID(thing) == CFDateGetTypeID()) {
+    timeNow = CFDateGetAbsoluteTime((CFDateRef)thing);
+    current = CFAbsoluteTimeGetGregorianDate(timeNow, NULL);
+    result = rb_funcall(rb_cTime, rb_intern("gm"), 6,
+                        INT2FIX((unsigned int)current.year),
+                        INT2FIX((unsigned int)current.month),
+                        INT2FIX((unsigned int)current.day),
+                        INT2FIX((unsigned int)current.hour),
+                        INT2FIX((unsigned int)current.minute),
+                        INT2FIX((unsigned int)current.second));
+  } else if(CFGetTypeID(thing) == CFBooleanGetTypeID()) {
+    result = CFBooleanGetValue(thing) ? Qtrue : Qfalse;
+  } else if(CFGetTypeID(thing) == CFArrayGetTypeID()) {
+    result = rb_ary_new();
+    for(i = 0; i < CFArrayGetCount(thing); i ++) {
+      rb_ary_push(result, CFType2Value(CFArrayGetValueAtIndex(thing, i)));
+    }
+  } else {
+    result = Qnil;
+  }
+
+  return result;
 }
 
 MDItemRef getItem(VALUE obj) {
@@ -27,6 +93,7 @@ void cMDItemNative_free(void *ptr) {
   }
 
   free(ptr);
+
 /*   printf("cMDItemNative_free()\n"); */
 }
 
@@ -53,7 +120,7 @@ static VALUE cMDItemNative_new(int argc, VALUE *argv, VALUE klass)
   if(item != NULL) {
     obj = createInstanceFromMDItem(item);
   } else {
-    rb_raise(rb_eArgError, "no such file or directory");
+    rb_raise(rb_eArgError, "no such file or directory: %s", filename);
   }
 
   CFRelease(cfFilename);
@@ -64,11 +131,10 @@ static VALUE cMDItemNative_new(int argc, VALUE *argv, VALUE klass)
 static VALUE cMDItemNative_get(int argc, VALUE *argv, VALUE self)
 {
   MDItemRef item = getItem(self);
-  CFStringRef itemValue, cfAttrName;
+  CFTypeRef itemValue;
+  CFStringRef cfAttrName;
   VALUE attrName, result;
-  char *tmpptr;
-  int stringSize;
-  
+
   rb_scan_args(argc, argv, "1", &attrName);
 
   if(TYPE(attrName) == T_SYMBOL) {
@@ -77,23 +143,41 @@ static VALUE cMDItemNative_get(int argc, VALUE *argv, VALUE self)
     cfAttrName = (CFStringRef)CString2CFString(StringValuePtr(attrName));
   }
 
-  itemValue = (CFStringRef)MDItemCopyAttribute(item, cfAttrName);
+/*   rb_p(CFString2RString(cfAttrName)); */
+
+  itemValue = MDItemCopyAttribute(item, cfAttrName);
+
+  CFRelease(cfAttrName);
 
   if(itemValue != NULL) {
-    stringSize = CFStringGetMaximumSizeForEncoding(CFStringGetLength(itemValue), kCFStringEncodingUTF8) + 1;
-    tmpptr = (char *)malloc(sizeof(char) * stringSize);
+    result = CFType2Value(itemValue);
+    CFRelease(itemValue);
+    return result;
+  }
 
-    CFStringGetCString(itemValue, tmpptr, stringSize, kCFStringEncodingUTF8);
+  return Qnil;
+}
 
-    result = rb_str_new2(tmpptr);
+static VALUE cMDItemNative_attribute_names(int argc, VALUE *argv, VALUE self)
+{
+  MDItemRef item = getItem(self);
+  CFArrayRef cfAttributeNames;
+  VALUE result = rb_ary_new();
+  int numAttributes, i;
 
-    free(tmpptr);
-    CFRelease(cfAttrName);
-    CFRelease(itemValue);
-  } else {
-    return Qnil;
+  rb_scan_args(argc, argv, "0");
+
+  cfAttributeNames = MDItemCopyAttributeNames(item);
+
+  if(cfAttributeNames != NULL) {
+    numAttributes = (int)CFArrayGetCount(cfAttributeNames);
+    for(i = 0; i < numAttributes; i ++) {
+      rb_ary_push(result, CFString2RString((CFStringRef)CFArrayGetValueAtIndex(cfAttributeNames, i)));
+    }
   }
 
+  CFRelease(cfAttributeNames);
+
   return result;
 }
 
@@ -104,4 +188,5 @@ void Init_md_item_native(void){
   rb_cMDItemNative = rb_define_class_under(rb_mSpotlight, "MDItemNative", rb_cObject);
   rb_define_singleton_method(rb_cMDItemNative, "new", cMDItemNative_new, -1);
   rb_define_method(rb_cMDItemNative, "get", cMDItemNative_get, -1);
+  rb_define_method(rb_cMDItemNative, "attribute_names", cMDItemNative_attribute_names, -1);
 }

data/ext/md_item_native/md_item_native.h

@@ -3,6 +3,8 @@
 
 #include <ruby.h>
 #include </System/Library/Frameworks/CoreServices.framework/Versions/A/Frameworks/Metadata.framework/Versions/A/Headers/MDItem.h>
+#include </System/Library/Frameworks/CoreFoundation.framework/Versions/A/Headers/CFDate.h>
+#include </System/Library/Frameworks/CoreFoundation.framework/Versions/A/Headers/CFNumber.h>
 
 VALUE createInstanceFromMDItem(MDItemRef item);
 

data/lib/spotlight/query.rb

@@ -1,8 +1,18 @@
+require 'rubygems'
+require 'plist'
+
 module Spotlight
   class Query
     attr_reader :query_string
     attr_accessor :scopes
 
+    def self.from_saved_search(filename)
+      plist = Plist::parse_xml(filename)
+      query = new(plist['RawQuery'])
+      query.scopes = plist['SearchCriteria']['FXScopeArrayOfPaths']
+      query
+    end
+
     def initialize(query_string)
       @query_string = query_string
       @scopes = []

data/spec/md_item_native_spec.rb

@@ -2,19 +2,30 @@ require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
 
 describe Spotlight::MDItemNative do
   before do
-    @query = Spotlight::MDItemNative.new(__FILE__)
+    @item = Spotlight::MDItemNative.new(__FILE__)
+    @filename = File.basename(__FILE__)
   end
 
   it "should instanciate" do
-    @query.should be_an_instance_of(Spotlight::MDItemNative)
+    @item.should be_an_instance_of(Spotlight::MDItemNative)
   end
 
   it "should get attribute by name" do
-    @query.get('kMDItemKind').should eql('Ruby Source File')
-    @query.get(:kMDItemKind).should eql('Ruby Source File')
+    @item.get('kMDItemKind').should eql('Ruby Source File')
+    @item.get(:kMDItemKind).should eql('Ruby Source File')
+    @item.get(:kMDItemFSName).should eql(@filename)
+    @item.get(:kMDItemFSCreationDate).should be_an_instance_of(Time)
+    @item.get(:kMDItemFSOwnerUserID).should be_an_instance_of(Fixnum)
+    @item.get(:kMDItemFSIsExtensionHidden).should be_an_instance_of(FalseClass)
   end
 
   it "should get nil if attribute is not found" do
-    @query.get('kMDXXXXXX').should be_nil
+    @item.get('kMDXXXXXX').should be_nil
+  end
+
+  it "should get attribute names" do
+    attribute_names = @item.attribute_names
+    attribute_names.should be_an_instance_of(Array)
+    attribute_names.should be_include('kMDItemKind')
   end
 end

data/spec/spec_helper.rb

@@ -4,6 +4,10 @@ require 'spotlight'
 require 'spec'
 require 'spec/autorun'
 
+system('mdimport ' + File.dirname(__FILE__))
+
 Spec::Runner.configure do |config|
-  
+  def fixture(filename)
+    File.dirname(__FILE__) + '/fixtures/' + filename
+  end
 end

data/spec/spotlight_query_spec.rb

@@ -5,6 +5,12 @@ describe Spotlight::Query do
     @query = Spotlight::Query.new('kMDItemDisplayName = "spotlight_query_spec.rb"')
   end
 
+  it "should create query from saved search" do
+    query = Spotlight::Query.from_saved_search(fixture('test.savedSearch'))
+    query.query_string.should eql('((true) && (true)) && ((* = "test*"cdw || kMDItemTextContent = "test*"cdw))')
+    query.scopes.should eql(['kMDQueryScopeComputer'])
+  end
+
   it "should have query string" do
     @query.query_string.should eql('kMDItemDisplayName = "spotlight_query_spec.rb"')
   end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: spotlight
 version: !ruby/object:Gem::Version 
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors: 
 - youpy
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-02-16 00:00:00 +09:00
+date: 2010-02-18 00:00:00 +09:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -32,6 +32,16 @@ dependencies:
       - !ruby/object:Gem::Version 
         version: "0"
     version: 
+- !ruby/object:Gem::Dependency 
+  name: plist
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
 description: Library to use Spotlight from Ruby
 email: youpy@buycheapviagraonlinenow.com
 executables: []
@@ -51,7 +61,7 @@ files:
 - LICENSE
 - README.rdoc
 has_rdoc: true
-homepage: http://github.com/youpy/spotlight
+homepage: http://github.com/youpy/ruby-spotlight
 licenses: []
 
 post_install_message: