/*
 * Copyright (C) 2019 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.protolog.tool

import perfetto.protos.PerfettoTrace.ProtoLogLevel
import perfetto.protos.PerfettoTrace.ProtoLogViewerConfig

/**
 * A builder class to construct the viewer configuration (i.e. mappings of protolog hashes to log
 * message information used to decode the protolog messages) encoded as a proto message.
 */
class ViewerConfigProtoBuilder : ProtoLogTool.ProtologViewerConfigBuilder {
    /**
     * @return a byte array of a ProtoLogViewerConfig proto message encoding all the viewer
     * configurations mapping protolog hashes to message information and log group information.
     */
    override fun build(groups: Collection<LogGroup>, statements: Map<ProtoLogTool.LogCall, Long>): ByteArray {
        val configBuilder = ProtoLogViewerConfig.newBuilder()

        // TODO(b/373754057): We are passing all the groups now, because some groups might only be
        //  used by Kotlin code that is not processed, but for group that get enabled to log to
        //  logcat we try and load the viewer configurations for this group, so the group must exist
        //  in the viewer config. Once Kotlin is pre-processed or this logic changes we should only
        //  use the groups that are actually used as an optimization.
        groups.forEach { group ->
            configBuilder.addGroups(ProtoLogViewerConfig.Group.newBuilder()
                    .setId(group.id)
                    .setName(group.name)
                    .setTag(group.tag)
                    .build())
        }

        statements.forEach { (log, key) ->
            configBuilder.addMessages(
                ProtoLogViewerConfig.MessageData.newBuilder()
                        .setMessageId(key)
                        .setMessage(log.messageString)
                        .setLevel(
                            ProtoLogLevel.forNumber(log.logLevel.protoMessageId))
                        .setGroupId(log.logGroup.id)
                        .setLocation("${log.position}:${log.lineNumber}")
            )
        }

        return configBuilder.build().toByteArray()
    }
}