1/*
2 * Copyright (c) 2015, 2016, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.  Oracle designates this
8 * particular file as subject to the "Classpath" exception as provided
9 * by Oracle in the LICENSE file that accompanied this code.
10 *
11 * This code is distributed in the hope that it will be useful, but WITHOUT
12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
14 * version 2 for more details (a copy is included in the LICENSE file that
15 * accompanied this code).
16 *
17 * You should have received a copy of the GNU General Public License version
18 * 2 along with this work; if not, write to the Free Software Foundation,
19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20 *
21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
22 * or visit www.oracle.com if you need additional information or have any
23 * questions.
24 */
25package jdk.incubator.http.internal.hpack;
26
27import java.nio.ByteBuffer;
28
29abstract class IndexNameValueWriter implements BinaryRepresentationWriter {
30
31    private final int pattern;
32    private final int prefix;
33    private final IntegerWriter intWriter = new IntegerWriter();
34    private final StringWriter nameWriter = new StringWriter();
35    private final StringWriter valueWriter = new StringWriter();
36
37    protected boolean indexedRepresentation;
38
39    private static final int NEW               = 0;
40    private static final int NAME_PART_WRITTEN = 1;
41    private static final int VALUE_WRITTEN     = 2;
42
43    private int state = NEW;
44
45    protected IndexNameValueWriter(int pattern, int prefix) {
46        this.pattern = pattern;
47        this.prefix = prefix;
48    }
49
50    IndexNameValueWriter index(int index) {
51        indexedRepresentation = true;
52        intWriter.configure(index, prefix, pattern);
53        return this;
54    }
55
56    IndexNameValueWriter name(CharSequence name, boolean useHuffman) {
57        indexedRepresentation = false;
58        intWriter.configure(0, prefix, pattern);
59        nameWriter.configure(name, useHuffman);
60        return this;
61    }
62
63    IndexNameValueWriter value(CharSequence value, boolean useHuffman) {
64        valueWriter.configure(value, useHuffman);
65        return this;
66    }
67
68    @Override
69    public boolean write(HeaderTable table, ByteBuffer destination) {
70        if (state < NAME_PART_WRITTEN) {
71            if (indexedRepresentation) {
72                if (!intWriter.write(destination)) {
73                    return false;
74                }
75            } else {
76                if (!intWriter.write(destination) || !nameWriter.write(destination)) {
77                    return false;
78                }
79            }
80            state = NAME_PART_WRITTEN;
81        }
82        if (state < VALUE_WRITTEN) {
83            if (!valueWriter.write(destination)) {
84                return false;
85            }
86            state = VALUE_WRITTEN;
87        }
88        return state == VALUE_WRITTEN;
89    }
90
91    @Override
92    public IndexNameValueWriter reset() {
93        intWriter.reset();
94        if (!indexedRepresentation) {
95            nameWriter.reset();
96        }
97        valueWriter.reset();
98        state = NEW;
99        return this;
100    }
101}
102