initial commit

.gitignore

@@ -0,0 +1 @@
+build

ArqFolder.h

@@ -0,0 +1,43 @@
+/*
+ Copyright (c) 2009, Stefan Reitshamer http://www.haystacksoftware.com
+ 
+ All rights reserved.
+ 
+ Redistribution and use in source and binary forms, with or without
+ modification, are permitted provided that the following conditions are met:
+ 
+ * Redistributions of source code must retain the above copyright
+ notice, this list of conditions and the following disclaimer.
+ 
+ * Redistributions in binary form must reproduce the above copyright
+ notice, this list of conditions and the following disclaimer in the
+ documentation and/or other materials provided with the distribution.
+ 
+ * Neither the names of PhotoMinds LLC or Haystack Software, nor the names of 
+ their contributors may be used to endorse or promote products derived from
+ this software without specific prior written permission.
+ 
+ THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
+ TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
+ PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
+ LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
+ NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+ SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */ 
+
+#import <Cocoa/Cocoa.h>
+#import "DictNode.h"
+
+@interface ArqFolder : NSObject {
+    NSString *s3Path;
+    NSString *localPath;
+}
+- (id)initWithS3Path:(NSString *)theS3Path plist:(DictNode *)plist;
+- (NSString *)s3Path;
+- (NSString *)localPath;
+@end

ArqFolder.m

@@ -0,0 +1,55 @@
+/*
+ Copyright (c) 2009, Stefan Reitshamer http://www.haystacksoftware.com
+ 
+ All rights reserved.
+ 
+ Redistribution and use in source and binary forms, with or without
+ modification, are permitted provided that the following conditions are met:
+ 
+ * Redistributions of source code must retain the above copyright
+ notice, this list of conditions and the following disclaimer.
+ 
+ * Redistributions in binary form must reproduce the above copyright
+ notice, this list of conditions and the following disclaimer in the
+ documentation and/or other materials provided with the distribution.
+ 
+ * Neither the names of PhotoMinds LLC or Haystack Software, nor the names of 
+ their contributors may be used to endorse or promote products derived from
+ this software without specific prior written permission.
+ 
+ THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
+ TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
+ PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
+ LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
+ NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+ SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */ 
+
+#import "ArqFolder.h"
+#import "DictNode.h"
+
+@implementation ArqFolder
+- (id)initWithS3Path:(NSString *)theS3Path plist:(DictNode *)plist {
+    if (self = [super init]) {
+        s3Path = [theS3Path copy];
+        localPath = [[[plist stringNodeForKey:@"LocalPath"] stringValue] copy];
+    }
+    return self;
+}
+- (void)dealloc {
+    [s3Path release];
+    [localPath release];
+    [super dealloc];
+}
+- (NSString *)s3Path {
+    return s3Path;
+}
+- (NSString *)localPath {
+    return localPath;
+}
+@end

ArqRestoreCommand.h

@@ -0,0 +1,44 @@
+/*
+ Copyright (c) 2009, Stefan Reitshamer http://www.haystacksoftware.com
+ 
+ All rights reserved.
+ 
+ Redistribution and use in source and binary forms, with or without
+ modification, are permitted provided that the following conditions are met:
+ 
+ * Redistributions of source code must retain the above copyright
+ notice, this list of conditions and the following disclaimer.
+ 
+ * Redistributions in binary form must reproduce the above copyright
+ notice, this list of conditions and the following disclaimer in the
+ documentation and/or other materials provided with the distribution.
+ 
+ * Neither the names of PhotoMinds LLC or Haystack Software, nor the names of 
+ their contributors may be used to endorse or promote products derived from
+ this software without specific prior written permission.
+ 
+ THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
+ TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
+ PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
+ LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
+ NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+ SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */ 
+
+#import <Cocoa/Cocoa.h>
+@class S3Service;
+
+@interface ArqRestoreCommand : NSObject {
+    NSString *accessKey;
+    NSString *secretKey;
+    NSString *encryptionPassword;
+    S3Service *s3;
+}
+- (BOOL)printArqFolders:(NSError **)error;
+- (BOOL)restorePath:(NSString *)path error:(NSError **)error;
+@end

ArqRestoreCommand.m

@@ -0,0 +1,191 @@
+/*
+ Copyright (c) 2009, Stefan Reitshamer http://www.haystacksoftware.com
+ 
+ All rights reserved.
+ 
+ Redistribution and use in source and binary forms, with or without
+ modification, are permitted provided that the following conditions are met:
+ 
+ * Redistributions of source code must retain the above copyright
+ notice, this list of conditions and the following disclaimer.
+ 
+ * Redistributions in binary form must reproduce the above copyright
+ notice, this list of conditions and the following disclaimer in the
+ documentation and/or other materials provided with the distribution.
+ 
+ * Neither the names of PhotoMinds LLC or Haystack Software, nor the names of 
+ their contributors may be used to endorse or promote products derived from
+ this software without specific prior written permission.
+ 
+ THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
+ TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
+ PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
+ LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
+ NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+ SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */ 
+
+#import "ArqRestoreCommand.h"
+#import "SetNSError.h"
+#import "S3AuthorizationProvider.h"
+#import "S3Service.h"
+#import "RegexKitLite.h"
+#import "DictNode.h"
+#import "ArqFolder.h"
+#import "HTTP.h"
+#import "Restorer.h"
+
+@interface ArqRestoreCommand (internal)
+- (BOOL)validateS3Keys:(NSError **)error;
+- (NSArray *)s3BucketNames:(NSError **)error;
+- (NSString *)s3BucketNameForRegion:(int)s3BucketRegion;
+@end
+
+@implementation ArqRestoreCommand
+- (id)init {
+    if (self = [super init]) {
+        char *theAccessKey = getenv("ARQ_ACCESS_KEY");
+        if (theAccessKey != NULL) {
+            accessKey = [[NSString alloc] initWithUTF8String:theAccessKey];
+        }
+        char *theSecretKey = getenv("ARQ_SECRET_KEY");
+        if (theSecretKey != NULL) {
+            secretKey = [[NSString alloc] initWithUTF8String:theSecretKey];
+        }
+        char *theEncryptionPassword = getenv("ARQ_ENCRYPTION_PASSWORD");
+        if (theEncryptionPassword != NULL) {
+            encryptionPassword = [[NSString alloc] initWithUTF8String:theEncryptionPassword];
+        }
+        if (accessKey != nil && secretKey != nil) {
+            S3AuthorizationProvider *sap = [[S3AuthorizationProvider alloc] initWithAccessKey:accessKey secretKey:secretKey];
+            s3 = [[S3Service alloc] initWithS3AuthorizationProvider:sap useSSL:NO retryOnNetworkError:YES];
+            [sap release];
+        }
+    }
+    return self;
+}
+- (void)dealloc {
+    [accessKey release];
+    [secretKey release];
+    [encryptionPassword release];
+    [s3 release];
+    [super dealloc];
+}
+- (BOOL)printArqFolders:(NSError **)error {
+    if (![self validateS3Keys:error]) {
+        return NO;
+    }
+    NSArray *s3BucketNames = [self s3BucketNames:error];
+    if (s3BucketNames == nil) {
+        return NO;
+    }
+    NSMutableArray *computerUUIDPaths = [NSMutableArray array];
+    for (NSString *s3BucketName in s3BucketNames) {
+        NSString *computerUUIDPrefix = [NSString stringWithFormat:@"/%@/", s3BucketName];
+        NSError *myError = nil;
+        NSArray *computerUUIDs = [s3 commonPrefixesForPathPrefix:computerUUIDPrefix delimiter:@"/" error:&myError];
+        if (computerUUIDs == nil) {
+            if ([[myError domain] isEqualToString:[S3Service serverErrorDomain]] && [myError code] == HTTP_NOT_FOUND) {
+                // Skip.
+            } else {
+                if (error != NULL) {
+                    *error = myError;
+                }
+                return NO;
+            }
+        }
+        for (NSString *computerUUID in computerUUIDs) {
+            [computerUUIDPaths addObject:[computerUUIDPrefix stringByAppendingPathComponent:computerUUID]];
+        }
+    }
+    for (NSString *computerUUIDPath in computerUUIDPaths) {
+        NSString *computerBucketsPrefix = [computerUUIDPath stringByAppendingPathComponent:@"buckets"];
+        NSArray *s3BucketUUIDPaths = [s3 pathsWithPrefix:computerBucketsPrefix error:error];
+        if (s3BucketUUIDPaths == nil) {
+            return NO;
+        }
+        for (NSString *uuidPath in s3BucketUUIDPaths) {
+            NSData *data = [s3 dataAtPath:uuidPath error:error];
+            if (data == nil) {
+                return NO;
+            }
+            DictNode *plist = [DictNode dictNodeWithXMLData:data error:error];
+            if (plist == nil) {
+                return NO;
+            }
+            printf("s3 path=%s\tlocal path=%s\n", [uuidPath UTF8String], [[[plist stringNodeForKey:@"LocalPath"] stringValue] UTF8String]);
+        }
+    }
+    return YES;
+}
+- (BOOL)restorePath:(NSString *)path error:(NSError **)error {
+    if (![self validateS3Keys:error]) {
+        return NO;
+    }
+    if (encryptionPassword == nil) {
+        SETNSERROR(@"ArqErrorDomain", -1, @"missing ARQ_ENCRYPTION_PASSWORD environment variable");
+        return NO;
+    }
+    NSString *pattern = @"^/([^/]+)/([^/]+)/buckets/([^/]+)";
+    NSRange s3BucketNameRange = [path rangeOfRegex:pattern capture:1];
+    NSRange computerUUIDRange = [path rangeOfRegex:pattern capture:2];
+    NSRange bucketUUIDRange = [path rangeOfRegex:pattern capture:3];
+    if (s3BucketNameRange.location == NSNotFound || computerUUIDRange.location == NSNotFound || bucketUUIDRange.location == NSNotFound) {
+        SETNSERROR(@"ArqErrorDomain", -1, @"invalid S3 path");
+        return NO;
+    }
+    NSData *data = [s3 dataAtPath:path error:error];
+    if (data == nil) {
+        return NO;
+    }
+    DictNode *plist = [DictNode dictNodeWithXMLData:data error:error];
+    if (plist == nil) {
+        return NO;
+    }
+    NSString *s3BucketName = [path substringWithRange:s3BucketNameRange];
+    NSString *computerUUID = [path substringWithRange:computerUUIDRange];
+    NSString *bucketUUID = [path substringWithRange:bucketUUIDRange];
+    NSString *bucketName = [[plist stringNodeForKey:@"BucketName"] stringValue];
+    Restorer *restorer = [[[Restorer alloc] initWithS3Service:s3 s3BucketName:s3BucketName computerUUID:computerUUID bucketUUID:bucketUUID bucketName:bucketName encryptionKey:encryptionPassword] autorelease];
+    if (![restorer restore:error]) {
+        return NO;
+    }
+    printf("restored files are in %s\n", [bucketName fileSystemRepresentation]);
+    return YES;
+}
+@end
+
+@implementation ArqRestoreCommand (internal)
+- (BOOL)validateS3Keys:(NSError **)error {
+    if (accessKey == nil) {
+        SETNSERROR(@"ArqErrorDomain", -1, @"missing ARQ_ACCESS_KEY environment variable");
+        return NO;
+    }
+    if (secretKey == nil) {
+        SETNSERROR(@"ArqErrorDomain", -1, @"missing ARQ_SECRET_KEY environment variable");
+        return NO;
+    }
+    return YES;
+}
+- (NSArray *)s3BucketNames:(NSError **)error {
+    return [NSArray arrayWithObjects:
+            [self s3BucketNameForRegion:BUCKET_REGION_US_STANDARD],
+            [self s3BucketNameForRegion:BUCKET_REGION_US_WEST],
+            [self s3BucketNameForRegion:BUCKET_REGION_EU],
+            nil];
+}
+- (NSString *)s3BucketNameForRegion:(int)s3BucketRegion {
+    NSString *regionSuffix = @"";
+    if (s3BucketRegion == BUCKET_REGION_US_WEST) {
+        regionSuffix = @".us-west-1";
+    } else if (s3BucketRegion == BUCKET_REGION_EU) {
+        regionSuffix = @".eu";
+    }
+    return [NSString stringWithFormat:@"%@.com.haystacksoftware.arq%@", [accessKey lowercaseString], regionSuffix];
+}
+@end

ArqUserLibrary.h

@@ -0,0 +1,40 @@
+/*
+ Copyright (c) 2009, Stefan Reitshamer http://www.haystacksoftware.com
+ 
+ All rights reserved.
+ 
+ Redistribution and use in source and binary forms, with or without
+ modification, are permitted provided that the following conditions are met:
+ 
+ * Redistributions of source code must retain the above copyright
+ notice, this list of conditions and the following disclaimer.
+ 
+ * Redistributions in binary form must reproduce the above copyright
+ notice, this list of conditions and the following disclaimer in the
+ documentation and/or other materials provided with the distribution.
+ 
+ * Neither the names of PhotoMinds LLC or Haystack Software, nor the names of 
+ their contributors may be used to endorse or promote products derived from
+ this software without specific prior written permission.
+ 
+ THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
+ TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
+ PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
+ LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
+ NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+ SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */ 
+
+#import <Cocoa/Cocoa.h>
+
+
+@interface ArqUserLibrary : NSObject {
+
+}
++ (NSString *)arqCachesPath;
+@end