16a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen/* 26a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen * Copyright (C) 2015 The Android Open Source Project 36a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen * 46a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen * Licensed under the Apache License, Version 2.0 (the "License"); 56a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen * you may not use this file except in compliance with the License. 66a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen * You may obtain a copy of the License at 76a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen * 86a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen * http://www.apache.org/licenses/LICENSE-2.0 96a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen * 106a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen * Unless required by applicable law or agreed to in writing, software 116a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen * distributed under the License is distributed on an "AS IS" BASIS, 126a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 136a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen * See the License for the specific language governing permissions and 146a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen * limitations under the License. 156a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen */ 166a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen 176a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wenpackage com.android.statementservice.retriever; 186a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen 196a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wenimport android.util.JsonWriter; 206a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen 216a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wenimport java.io.IOException; 226a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wenimport java.io.StringWriter; 236a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wenimport java.util.List; 246a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wenimport java.util.Locale; 256a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen 266a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen/** 276a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen * Creates a Json string where the order of the fields can be specified. 286a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen */ 296a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen/* package private */ final class AssetJsonWriter { 306a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen 316a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen private StringWriter mStringWriter = new StringWriter(); 326a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen private JsonWriter mWriter; 336a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen private boolean mClosed = false; 346a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen 356a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen public AssetJsonWriter() { 366a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen mWriter = new JsonWriter(mStringWriter); 376a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen try { 386a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen mWriter.beginObject(); 396a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen } catch (IOException e) { 406a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen throw new AssertionError("Unreachable exception."); 416a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen } 426a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen } 436a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen 446a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen /** 456a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen * Appends a field to the output, putting both the key and value in lowercase. Null values are 466a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen * not written. 476a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen */ 486a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen public void writeFieldLower(String key, String value) { 496a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen if (mClosed) { 506a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen throw new IllegalArgumentException( 516a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen "Cannot write to an object that has already been closed."); 526a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen } 536a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen 546a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen if (value != null) { 556a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen try { 566a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen mWriter.name(key.toLowerCase(Locale.US)); 576a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen mWriter.value(value.toLowerCase(Locale.US)); 586a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen } catch (IOException e) { 596a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen throw new AssertionError("Unreachable exception."); 606a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen } 616a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen } 626a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen } 636a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen 646a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen /** 656a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen * Appends an array to the output, putting both the key and values in lowercase. If {@code 666a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen * values} is null, this field will not be written. Individual values in the list must not be 676a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen * null. 686a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen */ 696a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen public void writeArrayUpper(String key, List<String> values) { 706a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen if (mClosed) { 716a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen throw new IllegalArgumentException( 726a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen "Cannot write to an object that has already been closed."); 736a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen } 746a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen 756a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen if (values != null) { 766a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen try { 776a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen mWriter.name(key.toLowerCase(Locale.US)); 786a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen mWriter.beginArray(); 796a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen for (String value : values) { 806a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen mWriter.value(value.toUpperCase(Locale.US)); 816a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen } 826a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen mWriter.endArray(); 836a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen } catch (IOException e) { 846a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen throw new AssertionError("Unreachable exception."); 856a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen } 866a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen } 876a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen } 886a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen 896a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen /** 906a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen * Returns the string representation of the constructed json. After calling this method, {@link 916a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen * #writeFieldLower} can no longer be called. 926a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen */ 936a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen public String closeAndGetString() { 946a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen if (!mClosed) { 956a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen try { 966a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen mWriter.endObject(); 976a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen } catch (IOException e) { 986a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen throw new AssertionError("Unreachable exception."); 996a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen } 1006a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen mClosed = true; 1016a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen } 1026a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen return mStringWriter.toString(); 1036a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen } 1046a34bb2d6a6cbc7a70bdf0c53d238dc28e0b1d58Joseph Wen} 105