mirror of https://github.com/k3s-io/k3s
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
146 lines
2.1 KiB
146 lines
2.1 KiB
#!/bin/bash
|
|
|
|
request_init() {
|
|
cat > request.go <<EOF
|
|
package protocol
|
|
|
|
// DO NOT EDIT
|
|
//
|
|
// This file was generated by ./schema.sh
|
|
|
|
EOF
|
|
}
|
|
|
|
response_init() {
|
|
cat > response.go <<EOF
|
|
package protocol
|
|
|
|
// DO NOT EDIT
|
|
//
|
|
// This file was generated by ./schema.sh
|
|
|
|
import "fmt"
|
|
|
|
EOF
|
|
}
|
|
|
|
entity=$1
|
|
shift
|
|
|
|
cmd=$1
|
|
shift
|
|
|
|
if [ "$entity" = "--request" ]; then
|
|
if [ "$cmd" = "init" ]; then
|
|
request_init
|
|
exit
|
|
fi
|
|
|
|
args=""
|
|
|
|
for i in "${@}"
|
|
do
|
|
name=$(echo "$i" | cut -f 1 -d :)
|
|
type=$(echo "$i" | cut -f 2 -d :)
|
|
|
|
if [ "$name" = "unused" ]; then
|
|
continue
|
|
fi
|
|
|
|
args=$(echo "${args}, ${name} ${type}")
|
|
done
|
|
|
|
cat >> request.go <<EOF
|
|
|
|
// Encode${cmd} encodes a $cmd request.
|
|
func Encode${cmd}(request *Message${args}) {
|
|
request.reset()
|
|
EOF
|
|
|
|
for i in "${@}"
|
|
do
|
|
name=$(echo "$i" | cut -f 1 -d :)
|
|
type=$(echo "$i" | cut -f 2 -d :)
|
|
|
|
if [ "$name" = "unused" ]; then
|
|
name=$(echo "0")
|
|
fi
|
|
|
|
cat >> request.go <<EOF
|
|
request.put${type^}(${name})
|
|
EOF
|
|
done
|
|
|
|
cat >> request.go <<EOF
|
|
|
|
request.putHeader(Request${cmd})
|
|
}
|
|
EOF
|
|
|
|
fi
|
|
|
|
if [ "$entity" = "--response" ]; then
|
|
if [ "$cmd" = "init" ]; then
|
|
response_init
|
|
exit
|
|
fi
|
|
|
|
returns=""
|
|
|
|
for i in "${@}"
|
|
do
|
|
name=$(echo "$i" | cut -f 1 -d :)
|
|
type=$(echo "$i" | cut -f 2 -d :)
|
|
|
|
if [ "$name" = "unused" ]; then
|
|
continue
|
|
fi
|
|
|
|
returns=$(echo "${returns}${name} ${type}, ")
|
|
done
|
|
|
|
cat >> response.go <<EOF
|
|
|
|
// Decode${cmd} decodes a $cmd response.
|
|
func Decode${cmd}(response *Message) (${returns}err error) {
|
|
mtype, _ := response.getHeader()
|
|
|
|
if mtype == ResponseFailure {
|
|
e := ErrRequest{}
|
|
e.Code = response.getUint64()
|
|
e.Description = response.getString()
|
|
err = e
|
|
return
|
|
}
|
|
|
|
if mtype != Response${cmd} {
|
|
err = fmt.Errorf("unexpected response type %d", mtype)
|
|
return
|
|
}
|
|
|
|
EOF
|
|
|
|
for i in "${@}"
|
|
do
|
|
name=$(echo "$i" | cut -f 1 -d :)
|
|
type=$(echo "$i" | cut -f 2 -d :)
|
|
|
|
assign=$(echo "${name} = ")
|
|
|
|
if [ "$name" = "unused" ]; then
|
|
assign=$(echo "")
|
|
fi
|
|
|
|
cat >> response.go <<EOF
|
|
${assign}response.get${type^}()
|
|
EOF
|
|
done
|
|
|
|
cat >> response.go <<EOF
|
|
|
|
return
|
|
}
|
|
EOF
|
|
|
|
fi
|