mirror of
https://github.com/ChronosX88/yans.git
synced 2024-11-08 14:50:59 +00:00
89 lines
1.7 KiB
Go
89 lines
1.7 KiB
Go
package protocol
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
type CapabilityType int
|
|
|
|
const (
|
|
VersionCapability CapabilityType = iota
|
|
ReaderCapability
|
|
IHaveCapability
|
|
PostCapability
|
|
NewNewsCapability
|
|
HdrCapability
|
|
OverCapability
|
|
ListCapability
|
|
ImplementationCapability
|
|
ModeReaderCapability
|
|
)
|
|
|
|
func (ct CapabilityType) String() string {
|
|
switch ct {
|
|
case VersionCapability:
|
|
return CapabilityNameVersion
|
|
case ReaderCapability:
|
|
return CapabilityNameReader
|
|
case IHaveCapability:
|
|
return CapabilityNameIHave
|
|
case PostCapability:
|
|
return CapabilityNamePost
|
|
case NewNewsCapability:
|
|
return CapabilityNameNewNews
|
|
case HdrCapability:
|
|
return CapabilityNameHdr
|
|
case OverCapability:
|
|
return CapabilityNameOver
|
|
case ListCapability:
|
|
return CapabilityNameList
|
|
case ImplementationCapability:
|
|
return CapabilityNameImplementation
|
|
case ModeReaderCapability:
|
|
return CapabilityNameModeReader
|
|
default:
|
|
return ""
|
|
}
|
|
}
|
|
|
|
type Capability struct {
|
|
Type CapabilityType
|
|
Params string // optional
|
|
}
|
|
|
|
type Capabilities []Capability
|
|
|
|
func (cs *Capabilities) Add(c Capability) {
|
|
for _, v := range *cs {
|
|
if v.Type == c.Type {
|
|
return // allowed only unique items
|
|
}
|
|
}
|
|
*cs = append(*cs, c)
|
|
}
|
|
|
|
func (cs *Capabilities) Remove(ct CapabilityType) {
|
|
for i, v := range *cs {
|
|
if v.Type == ct {
|
|
*cs = append((*cs)[:i], (*cs)[i+1:]...)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (cs Capabilities) String() string {
|
|
sb := strings.Builder{}
|
|
sb.Write([]byte("101 Capability list:" + CRLF))
|
|
|
|
for _, v := range cs {
|
|
if v.Params != "" {
|
|
sb.Write([]byte(fmt.Sprintf("%s %s%s", v.Type, v.Params, CRLF)))
|
|
} else {
|
|
sb.Write([]byte(fmt.Sprintf("%s%s", v.Type, CRLF)))
|
|
}
|
|
}
|
|
sb.Write([]byte(MultilineEnding))
|
|
|
|
return sb.String()
|
|
}
|