xref: /packages/providers/ContactsProvider/test_common/src/com/android/providers/contacts/testutil/DataUtil.java

/*
 * Copyright (C) 2013 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.testutil;

import android.content.ContentResolver;
import android.content.ContentUris;
import android.content.ContentValues;
import android.net.Uri;
import android.provider.ContactsContract;
import android.provider.ContactsContract.CommonDataKinds.StructuredName;
import android.provider.ContactsContract.Data;

/**
 * Convenience methods for operating on the Data table.
 */
public class DataUtil {

    private static final Uri URI = ContactsContract.Data.CONTENT_URI;

    public static void delete(ContentResolver resolver, long dataId) {
        Uri uri = ContentUris.withAppendedId(URI, dataId);
        resolver.delete(uri, null, null);
    }

    public static void update(ContentResolver resolver, long dataId, ContentValues values) {
        Uri uri = ContentUris.withAppendedId(URI, dataId);
        resolver.update(uri, values, null, null);
    }

    public static Uri insertStructuredName(ContentResolver resolver, long rawContactId,
            ContentValues values) {
        values.put(ContactsContract.Data.RAW_CONTACT_ID, rawContactId);
        values.put(ContactsContract.Data.MIMETYPE,
                ContactsContract.CommonDataKinds.StructuredName.CONTENT_ITEM_TYPE);
        Uri resultUri = resolver.insert(ContactsContract.Data.CONTENT_URI, values);
        return resultUri;
    }

    public static Uri insertStructuredName(ContentResolver resolver, long rawContactId,
            String givenName, String familyName) {
        return insertStructuredName(resolver, rawContactId, givenName, familyName,
                /* phonetic given =*/ null);
    }

    public static Uri insertStructuredName(
            ContentResolver resolver, long rawContactId, String givenName, String familyName,
            String phoneticFamily) {
        return insertStructuredName(resolver, rawContactId, givenName, familyName, phoneticFamily,
                /* isSuperPrimary = true */ false);
    }

    public static Uri insertStructuredName(
            ContentResolver resolver, long rawContactId, String givenName, String familyName,
            String phoneticFamily, boolean isSuperPrimary) {
        ContentValues values = new ContentValues();
        StringBuilder sb = new StringBuilder();
        if (givenName != null) {
            sb.append(givenName);
        }
        if (givenName != null && familyName != null) {
            sb.append(" ");
        }
        if (familyName != null) {
            sb.append(familyName);
        }
        if (sb.length() == 0 && phoneticFamily != null) {
            sb.append(phoneticFamily);
        }
        values.put(StructuredName.DISPLAY_NAME, sb.toString());
        values.put(StructuredName.GIVEN_NAME, givenName);
        values.put(StructuredName.FAMILY_NAME, familyName);
        if (phoneticFamily != null) {
            // When creating phonetic names, be careful to use PHONETIC_FAMILY_NAME instead of
            // PHONETIC_GIVEN_NAME, to work around b/19612393.
            values.put(StructuredName.PHONETIC_FAMILY_NAME, phoneticFamily);
        }
        if (isSuperPrimary) {
            values.put(Data.IS_PRIMARY, 1);
            values.put(Data.IS_SUPER_PRIMARY, 1);
        }

        return insertStructuredName(resolver, rawContactId, values);
    }

    public static Uri insertPhoneticName(ContentResolver resolver, long rawContactId,
            String phoneticFamilyName) {
        ContentValues values = new ContentValues();
        // When creating phonetic names, be careful to use PHONETIC_FAMILY_NAME instead of
        // PHONETIC_GIVEN_NAME, to work around b/19612393.
        values.put(StructuredName.PHONETIC_FAMILY_NAME, phoneticFamilyName);
        return insertStructuredName(resolver, rawContactId, values);
    }
}