001    /*
002     * Copyright 2010-2013 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.jet.descriptors.serialization;
018    
019    import com.google.protobuf.ExtensionRegistryLite;
020    import org.jetbrains.annotations.NotNull;
021    import org.jetbrains.jet.utils.UtilsPackage;
022    
023    import java.io.ByteArrayInputStream;
024    import java.io.ByteArrayOutputStream;
025    import java.io.IOException;
026    
027    public final class PackageData {
028        @NotNull
029        public static PackageData read(@NotNull byte[] bytes, @NotNull ExtensionRegistryLite registry) {
030            try {
031                ByteArrayInputStream in = new ByteArrayInputStream(bytes);
032                NameResolver nameResolver = NameSerializationUtil.deserializeNameResolver(in);
033                ProtoBuf.Package packageProto = ProtoBuf.Package.parseFrom(in, registry);
034                return new PackageData(nameResolver, packageProto);
035            }
036            catch (IOException e) {
037                throw UtilsPackage.rethrow(e);
038            }
039        }
040    
041        private final NameResolver nameResolver;
042    
043        private final ProtoBuf.Package packageProto;
044    
045        public PackageData(@NotNull NameResolver nameResolver, @NotNull ProtoBuf.Package packageProto) {
046            this.nameResolver = nameResolver;
047            this.packageProto = packageProto;
048        }
049    
050        @NotNull
051        public NameResolver getNameResolver() {
052            return nameResolver;
053        }
054    
055        @NotNull
056        public ProtoBuf.Package getPackageProto() {
057            return packageProto;
058        }
059    
060        @NotNull
061        public byte[] toBytes() {
062            try {
063                ByteArrayOutputStream result = new ByteArrayOutputStream();
064                NameSerializationUtil.serializeNameResolver(result, nameResolver);
065                packageProto.writeTo(result);
066                return result.toByteArray();
067            }
068            catch (IOException e) {
069                throw UtilsPackage.rethrow(e);
070            }
071        }
072    }