Extracting DataRowHandler classes to the top level

Change-Id: Ica84da28a01a9c1b3c19967ab8cb154d796812c2

1 files changed, 115 insertions, 0 deletions

diff --git a/src/com/android/providers/contacts/DataRowHandlerForOrganization.java b/src/com/android/providers/contacts/DataRowHandlerForOrganization.java
new file mode 100644
index 0000000..1382f1f
--- /dev/null
+++ b/src/com/android/providers/contacts/DataRowHandlerForOrganization.java
@@ -0,0 +1,115 @@
+/*
+ * Copyright (C) 2010 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not
+ * use this file except in compliance with the License. You may obtain a copy of
+ * the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+ * License for the specific language governing permissions and limitations under
+ * the License
+ */
+package com.android.providers.contacts;
+
+import com.android.providers.contacts.ContactsDatabaseHelper.Tables;
+
+import android.content.ContentValues;
+import android.database.Cursor;
+import android.database.DatabaseUtils;
+import android.database.sqlite.SQLiteDatabase;
+import android.provider.ContactsContract.CommonDataKinds.Organization;
+import android.provider.ContactsContract.Data;
+
+/**
+ * Handler for organization data rows.
+ */
+public class DataRowHandlerForOrganization extends DataRowHandlerForCommonDataKind {
+
+    public DataRowHandlerForOrganization(ContactsDatabaseHelper dbHelper,
+            ContactAggregator aggregator) {
+        super(dbHelper, aggregator,
+                Organization.CONTENT_ITEM_TYPE, Organization.TYPE, Organization.LABEL);
+    }
+
+    @Override
+    public long insert(SQLiteDatabase db, TransactionContext txContext, long rawContactId,
+            ContentValues values) {
+        String company = values.getAsString(Organization.COMPANY);
+        String title = values.getAsString(Organization.TITLE);
+
+        long dataId = super.insert(db, txContext, rawContactId, values);
+
+        fixRawContactDisplayName(db, txContext, rawContactId);
+        mDbHelper.insertNameLookupForOrganization(rawContactId, dataId, company, title);
+        return dataId;
+    }
+
+    @Override
+    public boolean update(SQLiteDatabase db, TransactionContext txContext, ContentValues values,
+            Cursor c, boolean callerIsSyncAdapter) {
+        if (!super.update(db, txContext, values, c, callerIsSyncAdapter)) {
+            return false;
+        }
+
+        boolean containsCompany = values.containsKey(Organization.COMPANY);
+        boolean containsTitle = values.containsKey(Organization.TITLE);
+        if (containsCompany || containsTitle) {
+            long dataId = c.getLong(DataUpdateQuery._ID);
+            long rawContactId = c.getLong(DataUpdateQuery.RAW_CONTACT_ID);
+
+            String company;
+
+            if (containsCompany) {
+                company = values.getAsString(Organization.COMPANY);
+            } else {
+                mSelectionArgs1[0] = String.valueOf(dataId);
+                company = DatabaseUtils.stringForQuery(db,
+                        "SELECT " + Organization.COMPANY +
+                        " FROM " + Tables.DATA +
+                        " WHERE " + Data._ID + "=?", mSelectionArgs1);
+            }
+
+            String title;
+            if (containsTitle) {
+                title = values.getAsString(Organization.TITLE);
+            } else {
+                mSelectionArgs1[0] = String.valueOf(dataId);
+                title = DatabaseUtils.stringForQuery(db,
+                        "SELECT " + Organization.TITLE +
+                        " FROM " + Tables.DATA +
+                        " WHERE " + Data._ID + "=?", mSelectionArgs1);
+            }
+
+            mDbHelper.deleteNameLookup(dataId);
+            mDbHelper.insertNameLookupForOrganization(rawContactId, dataId, company, title);
+
+            fixRawContactDisplayName(db, txContext, rawContactId);
+        }
+        return true;
+    }
+
+    @Override
+    public int delete(SQLiteDatabase db, TransactionContext txContext, Cursor c) {
+        long dataId = c.getLong(DataUpdateQuery._ID);
+        long rawContactId = c.getLong(DataDeleteQuery.RAW_CONTACT_ID);
+
+        int count = super.delete(db, txContext, c);
+        fixRawContactDisplayName(db, txContext, rawContactId);
+        mDbHelper.deleteNameLookup(dataId);
+        return count;
+    }
+
+    @Override
+    protected int getTypeRank(int type) {
+        switch (type) {
+            case Organization.TYPE_WORK: return 0;
+            case Organization.TYPE_CUSTOM: return 1;
+            case Organization.TYPE_OTHER: return 2;
+            default: return 1000;
+        }
+    }
+}