lal/pkg/rtmp/metadata.go

// Copyright 2020, Chef.  All rights reserved.
// https://github.com/q191201771/lal
//
// Use of this source code is governed by a MIT-style license
// that can be found in the License file.
//
// Author: Chef (191201771@qq.com)

package rtmp

import (
	"bytes"
	"github.com/q191201771/naza/pkg/nazabytes"
	"github.com/q191201771/naza/pkg/nazalog"

	"github.com/q191201771/lal/pkg/base"
)

func ParseMetadata(b []byte) (ObjectPairArray, error) {
	pos := 0
	v, l, err := Amf0.ReadString(b[pos:])
	if err != nil {
		return nil, err
	}
	pos += l
	if v == "@setDataFrame" {
		_, l, err = Amf0.ReadString(b[pos:])
		if err != nil {
			return nil, err
		}
		pos += l
	}
	opa, _, err := Amf0.ReadObjectOrArray(b[pos:])
	return opa, err
}

// TODO(chef): [test] MetadataEnsureWithSetDataFrame 这两个函数增加单元测试 202207

// MetadataEnsureWithSdf
//
// 确保metadata中包含@setDataFrame
//
// @return 返回的内存块为内部独立申请
func MetadataEnsureWithSdf(b []byte) ([]byte, error) {
	var ret []byte
	v, _, err := Amf0.ReadString(b)
	if err != nil {
		nazalog.Errorf("%+v", err)
		return append(ret, b...), err
	}

	// 已经有了
	if v == "@setDataFrame" {
		return append(ret, b...), nil
	}

	buf := nazabytes.NewBuffer(16 + len(b)) // 16=1+2+13 @setDataFrame
	if err = Amf0.WriteString(buf, "@setDataFrame"); err != nil {
		nazalog.Errorf("%+v", err)
		return append(ret, b...), err
	}
	_, err = buf.Write(b)
	return buf.Bytes(), err
}

// MetadataEnsureWithoutSdf
//
// 确保metadata中不包含@setDataFrame
//
// @return 返回的内存块为内部独立申请
func MetadataEnsureWithoutSdf(b []byte) ([]byte, error) {
	var ret []byte
	v, l, err := Amf0.ReadString(b)
	if err != nil {
		nazalog.Errorf("%+v", err)
		return append(ret, b...), err
	}

	// 本来就不包含
	if v != "@setDataFrame" {
		return append(ret, b...), nil
	}

	return append(ret, b[l:]...), nil
}

// BuildMetadata spec-video_file_format_spec_v10.pdf
// onMetaData
// - duration        DOUBLE, seconds
// - width           DOUBLE
// - height          DOUBLE
// - videodatarate   DOUBLE
// - framerate       DOUBLE
// - videocodecid    DOUBLE
// - audiosamplerate DOUBLE
// - audiosamplesize DOUBLE
// - stereo          BOOL
// - audiocodecid    DOUBLE
// - filesize        DOUBLE, bytes
//
// 目前包含的字段：
// - width
// - height
// - audiocodecid
// - videocodecid
// - version
//
// @param width        如果为-1，则metadata中不写入该字段
// @param height       如果为-1，则metadata中不写入该字段
// @param audiocodecid 如果为-1，则metadata中不写入该字段
//
//	AAC 10
//
// @param videocodecid 如果为-1，则metadata中不写入该字段
//
//	H264 7
//	H265 12
//
// @return 返回的内存块为新申请的独立内存块
func BuildMetadata(width int, height int, audiocodecid int, videocodecid int) ([]byte, error) {
	buf := &bytes.Buffer{}
	if err := Amf0.WriteString(buf, "onMetaData"); err != nil {
		return nil, err
	}

	var opa ObjectPairArray
	if width != -1 {
		opa = append(opa, ObjectPair{
			Key:   "width",
			Value: width,
		})
	}
	if height != -1 {
		opa = append(opa, ObjectPair{
			Key:   "height",
			Value: height,
		})
	}
	if audiocodecid != -1 {
		opa = append(opa, ObjectPair{
			Key:   "audiocodecid",
			Value: audiocodecid,
		})
	}
	if videocodecid != -1 {
		opa = append(opa, ObjectPair{
			Key:   "videocodecid",
			Value: videocodecid,
		})
	}
	opa = append(opa, ObjectPair{
		Key:   "version",
		Value: base.LalRtmpBuildMetadataEncoder,
	})
	opa = append(opa, ObjectPair{
		Key:   "lal",
		Value: base.LalVersionDot,
	})

	if err := Amf0.WriteObject(buf, opa); err != nil {
		return nil, err
	}

	return buf.Bytes(), nil
}