m7s/m7s.live/plugin/gb28181/v4@v4.4.4/inviteoption.go

package gb28181

import (
	"errors"
	"fmt"
	"math/rand"
	"strconv"
)

type InviteOptions struct {
	Start       int
	End         int
	dump        string
	ssrc        string
	SSRC        uint32
	MediaPort   uint16
	StreamPath  string
	recyclePort func(p uint16) (err error)
}

func (o InviteOptions) IsLive() bool {
	return o.Start == 0 || o.End == 0
}

func (o InviteOptions) Record() bool {
	return !o.IsLive()
}

func (o *InviteOptions) Validate(start, end string) error {
	if start != "" {
		sint, err1 := strconv.ParseInt(start, 10, 0)
		if err1 != nil {
			return err1
		}
		o.Start = int(sint)
	}
	if end != "" {
		eint, err2 := strconv.ParseInt(end, 10, 0)
		if err2 != nil {
			return err2
		}
		o.End = int(eint)
	}
	if o.Start >= o.End {
		return errors.New("start < end")
	}
	return nil
}

func (o InviteOptions) String() string {
	return fmt.Sprintf("t=%d %d", o.Start, o.End)
}

func (o *InviteOptions) CreateSSRC() {
	ssrc := make([]byte, 10)
	if o.IsLive() {
		ssrc[0] = '0'
	} else {
		ssrc[0] = '1'
	}
	copy(ssrc[1:6], conf.Serial[3:8])
	randNum := 1000 + rand.Intn(8999)
	copy(ssrc[6:], strconv.Itoa(randNum))
	o.ssrc = string(ssrc)
	_ssrc, _ := strconv.ParseInt(o.ssrc, 10, 0)
	o.SSRC = uint32(_ssrc)
}