001    /*
002     * Copyright 2010-2015 JetBrains s.r.o.
003     *
004     * Licensed under the Apache License, Version 2.0 (the "License");
005     * you may not use this file except in compliance with the License.
006     * You may obtain a copy of the License at
007     *
008     * http://www.apache.org/licenses/LICENSE-2.0
009     *
010     * Unless required by applicable law or agreed to in writing, software
011     * distributed under the License is distributed on an "AS IS" BASIS,
012     * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013     * See the License for the specific language governing permissions and
014     * limitations under the License.
015     */
016    
017    package org.jetbrains.kotlin.serialization;
018    
019    import com.google.protobuf.ExtensionRegistryLite;
020    import org.jetbrains.annotations.NotNull;
021    import org.jetbrains.kotlin.serialization.deserialization.NameResolver;
022    import org.jetbrains.kotlin.utils.UtilsPackage;
023    
024    import java.io.ByteArrayInputStream;
025    import java.io.ByteArrayOutputStream;
026    import java.io.IOException;
027    
028    public final class PackageData {
029        @NotNull
030        public static PackageData read(@NotNull byte[] bytes, @NotNull ExtensionRegistryLite registry) {
031            try {
032                ByteArrayInputStream in = new ByteArrayInputStream(bytes);
033                NameResolver nameResolver = NameSerializationUtil.deserializeNameResolver(in);
034                ProtoBuf.Package packageProto = ProtoBuf.Package.parseFrom(in, registry);
035                return new PackageData(nameResolver, packageProto);
036            }
037            catch (IOException e) {
038                throw UtilsPackage.rethrow(e);
039            }
040        }
041    
042        private final NameResolver nameResolver;
043    
044        private final ProtoBuf.Package packageProto;
045    
046        public PackageData(@NotNull NameResolver nameResolver, @NotNull ProtoBuf.Package packageProto) {
047            this.nameResolver = nameResolver;
048            this.packageProto = packageProto;
049        }
050    
051        @NotNull
052        public NameResolver getNameResolver() {
053            return nameResolver;
054        }
055    
056        @NotNull
057        public ProtoBuf.Package getPackageProto() {
058            return packageProto;
059        }
060    
061        @NotNull
062        public byte[] toBytes() {
063            try {
064                ByteArrayOutputStream result = new ByteArrayOutputStream();
065                NameSerializationUtil.serializeNameResolver(result, nameResolver);
066                packageProto.writeTo(result);
067                return result.toByteArray();
068            }
069            catch (IOException e) {
070                throw UtilsPackage.rethrow(e);
071            }
072        }
073    }