Monotone-Parent: 2fdf0ead0781ef2362dd389811d5db44ca4cc84a

Monotone-Revision: c91eef7edcfd08d414664c25a590b6268a7e58e2

Monotone-Author: wsourdeau@inverse.ca
Monotone-Date: 2009-11-29T04:19:32
Monotone-Branch: ca.inverse.sogo

SoObjects/SOGo/SOGoSQLUserProfile.m

@@ -0,0 +1,242 @@
+/* SOGoSQLUserProfile.m - this file is part of SOGo
+ *
+ * Copyright (C) 2009 Inverse inc.
+ *
+ * Author: Wolfgang Sourdeau <wsourdeau@inverse.ca>
+ *
+ * This file is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2, or (at your option)
+ * any later version.
+ *
+ * This file is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; see the file COPYING.  If not, write to
+ * the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
+ */
+
+#import <NGExtensions/NSObject+Logs.h>
+#import <NGExtensions/NSNull+misc.h>
+
+#import <GDLContentStore/GCSChannelManager.h>
+#import <GDLContentStore/NSURL+GCS.h>
+#import <GDLAccess/EOAdaptorChannel.h>
+#import <GDLAccess/EOAdaptorContext.h>
+#import <GDLAccess/EOAttribute.h>
+
+#import "SOGoSystemDefaults.h"
+
+#import "SOGoSQLUserProfile.h"
+
+static NSURL *tableURL = nil;
+static NSString *uidColumnName = @"c_uid";
+
+@implementation SOGoSQLUserProfile
+
++ (void) initialize
+{
+  NSString *profileURL;
+  SOGoSystemDefaults *sd;
+
+  if (!tableURL)
+    {
+      sd = [SOGoSystemDefaults sharedSystemDefaults];
+      profileURL = [sd profileURL];
+      if (profileURL)
+        tableURL = [[NSURL alloc] initWithString: profileURL];
+    }
+}
+
+- (id) init
+{
+  if ((self = [super init]))
+    {
+      fieldName = nil;
+    }
+
+  return self;
+}
+
+- (void) dealloc
+{
+  [fieldName release];
+  [super dealloc];
+}
+
+- (void) setProfileType: (SOGoUserProfileType) newProfileType
+{
+  if (newProfileType == SOGoUserProfileTypeDefaults)
+    ASSIGN (fieldName, @"c_defaults");
+  else if (newProfileType == SOGoUserProfileTypeSettings)
+    ASSIGN (fieldName, @"c_settings");
+  else
+    [self errorWithFormat: @"unknown profile value: %d", newProfileType];
+
+  [super setProfileType: newProfileType];
+}
+
+- (NSString *) fetchJSONProfileFromDB
+{
+  GCSChannelManager *cm;
+  EOAdaptorChannel *channel;
+  NSDictionary *row;
+  NSException *ex;
+  NSString *sql, *value;
+  NSArray *attrs;
+
+  value = nil;
+
+  cm = [GCSChannelManager defaultChannelManager];
+  channel = [cm acquireOpenChannelForURL: tableURL];
+  if (channel)
+    {
+      /* generate SQL */
+      defFlags.ready = YES;
+      sql = [NSString stringWithFormat: @"SELECT %@ FROM %@ WHERE %@ = '%@'",
+                      fieldName, [tableURL gcsTableName],
+                      uidColumnName, [self uid]];
+      /* run SQL */
+      ex = [channel evaluateExpressionX: sql];
+      if (ex)
+        [self errorWithFormat:@"could not run SQL '%@': %@", sql, ex];
+      else
+        {
+          /* fetch schema */
+          attrs = [channel describeResults: NO /* don't beautify */];
+          
+          /* fetch values */
+          row = [channel fetchAttributes: attrs withZone: NULL];
+          [channel cancelFetch];
+
+          /* the isNew flag depends on the presence of the row in the
+             database rather than on the existence of the value. */
+          defFlags.isNew = (row == nil);
+
+          value = [row objectForKey: fieldName];
+          if ([value isNotNull])
+            /* The following enables the restitution of coded unicode (\U1234)
+               characters with the Oracle adaptor. */
+            value = [value stringByReplacingString: @"\\\\"
+                                        withString: @"\\"];
+          else
+            value = nil; /* we discard any NSNull instance */
+        }
+
+      [cm releaseChannel: channel];
+    }
+  else
+    {
+      defFlags.ready = NO;
+      [self errorWithFormat:@"failed to acquire channel for URL: %@", 
+            tableURL];
+    }
+
+  return value;
+}
+
+- (NSString *) _sqlJsonRepresentation: (NSString *) jsonRepresentation
+{
+  NSMutableString *sql;
+
+  sql = [jsonRepresentation mutableCopy];
+  [sql autorelease];
+  [sql replaceString: @"\\" withString: @"\\\\"];
+  [sql replaceString: @"'" withString: @"''"];
+
+  return sql;
+}
+
+- (NSString *) generateSQLForInsert: (NSString *) jsonRepresentation
+{
+  NSString *sql;
+
+  if ([jsonRepresentation length])
+    sql = [NSString stringWithFormat: (@"INSERT INTO %@"
+                                       @"            (%@, %@)"
+                                       @"     VALUES ('%@', '%@')"),
+                    [tableURL gcsTableName], uidColumnName, fieldName,
+                    [self uid],
+                    [self _sqlJsonRepresentation: jsonRepresentation]];
+  else
+    sql = nil;
+
+  return sql;
+}
+
+- (NSString *) generateSQLForUpdate: (NSString *) jsonRepresentation
+{
+  NSString *sql;
+
+  if ([jsonRepresentation length])
+    sql = [NSString stringWithFormat: (@"UPDATE %@"
+                                       @"     SET %@ = '%@'"
+                                       @"   WHERE %@ = '%@'"),
+                    [tableURL gcsTableName],
+                    fieldName,
+                    [self _sqlJsonRepresentation: jsonRepresentation],
+                    uidColumnName, [self uid]];
+  else
+    sql = nil;
+
+  return sql;
+}
+
+- (BOOL) storeJSONProfileInDB: (NSString *) jsonRepresentation
+{
+  GCSChannelManager *cm;
+  EOAdaptorChannel *channel;
+  EOAdaptorContext *context;
+  NSException *ex;
+  NSString *sql;
+  BOOL rc;
+
+  rc = NO;
+
+  sql = ((defFlags.isNew)
+         ? [self generateSQLForInsert: jsonRepresentation]
+         : [self generateSQLForUpdate: jsonRepresentation]);
+  cm = [GCSChannelManager defaultChannelManager];
+  channel = [cm acquireOpenChannelForURL: tableURL];
+  if (channel)
+    {
+      context = [channel adaptorContext];
+      if ([context beginTransaction])
+        {
+          defFlags.ready = YES;
+          ex = [channel evaluateExpressionX:sql];
+          if (ex)
+            {
+              [self errorWithFormat: @"could not run SQL '%@': %@", sql, ex];
+              [context rollbackTransaction];
+            }
+          else
+            {
+              rc = YES;
+              defFlags.modified = NO;
+              defFlags.isNew = NO;
+              [context commitTransaction];
+            }
+          [cm releaseChannel: channel];
+        }
+      else
+        {
+          defFlags.ready = NO;
+          [cm releaseChannel: channel immediately: YES];
+        }
+    }
+  else
+    {
+      defFlags.ready = NO;
+      [self errorWithFormat: @"failed to acquire channel for URL: %@", 
+            tableURL];
+    }
+
+  return rc;
+}
+
+@end