mahmad/stream-server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Refactor command process to separate file

Browse source
This commit is contained in:
Muaz Ahmad 2023-08-11 13:12:12 +05:00
parent 0c054b67c0
commit 3d7e00169b
2 changed files with 31 additions and 32 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
rtmp/amf/command.go Normal file
View file

@ -0,0 +1,31 @@
package amf
import "errors"
func (amf_obj_root AMFObj) ProcessConnect() (string, error) {
err := errors.New("Bad AMF connect command")
if _, ok := amf_obj_root[0]; !ok {
return "", err
} else if command_string, ok := amf_obj_root[0].(string); !ok || command_string != "connect" {
return "", err
}
if _, ok := amf_obj_root[1]; !ok {
return "", err
} else if transac_id_float, ok := amf_obj_root[1].(float64); !ok || transac_id_float != 1.0 {
return "", err
}
if _, ok := amf_obj_root[2]; !ok {
return "", err
} else if _, ok := amf_obj_root[2].(AMFObj); !ok {
return "", err
} else if _, ok := amf_obj_root[2].(AMFObj)["tcUrl"]; !ok {
return "", err
} else if tcUrl, ok := amf_obj_root[2].(AMFObj)["tcUrl"].(string); ok {
return tcUrl, nil
} else {
return "", err
}
}

32
rtmp/amf/decode.go
View file

@ -1,9 +1,5 @@
package amf
import (
"errors"
)
type AMFObj map[interface{}]interface{}
func DecodeAMF(data *[]byte) (AMFObj, error) {
@ -15,31 +11,3 @@ func DecodeAMF(data *[]byte) (AMFObj, error) {
test["tcUrl"] = "rtmp://test-domain.nil/Desktop/tmp"
return obj, nil
}
func (amf_obj_root AMFObj) ProcessConnect() (string, error) {
err := errors.New("Bad AMF connect command")
if _, ok := amf_obj_root[0]; !ok {
return "", err
} else if command_string, ok := amf_obj_root[0].(string); !ok || command_string != "connect" {
return "", err
}
if _, ok := amf_obj_root[1]; !ok {
return "", err
} else if transac_id_float, ok := amf_obj_root[1].(float64); !ok || transac_id_float != 1.0 {
return "", err
}
if _, ok := amf_obj_root[2]; !ok {
return "", err
} else if _, ok := amf_obj_root[2].(AMFObj); !ok {
return "", err
} else if _, ok := amf_obj_root[2].(AMFObj)["tcUrl"]; !ok {
return "", err
} else if tcUrl, ok := amf_obj_root[2].(AMFObj)["tcUrl"].(string); ok {
return tcUrl, nil
} else {
return "", err
}
}