50 lines
1.2 KiB
Go
50 lines
1.2 KiB
Go
|
package server
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"strconv"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func writeBencoded(w io.Writer, data interface{}) {
|
||
|
|
switch v := data.(type) {
|
||
|
|
case string:
|
||
|
|
str := fmt.Sprintf("%s:%s", strconv.Itoa(len(v)), v)
|
||
|
|
io.WriteString(w, str)
|
||
|
|
case int:
|
||
|
|
str := fmt.Sprintf("i%se", strconv.Itoa(v))
|
||
|
|
io.WriteString(w, str)
|
||
|
|
case uint:
|
||
|
|
str := fmt.Sprintf("i%se", strconv.FormatUint(uint64(v), 10))
|
||
|
|
io.WriteString(w, str)
|
||
|
|
case int64:
|
||
|
|
str := fmt.Sprintf("i%se", strconv.FormatInt(v, 10))
|
||
|
|
io.WriteString(w, str)
|
||
|
|
case uint64:
|
||
|
|
str := fmt.Sprintf("i%se", strconv.FormatUint(v, 10))
|
||
|
|
io.WriteString(w, str)
|
||
|
|
case time.Duration: // Assume seconds
|
||
|
|
str := fmt.Sprintf("i%se", strconv.FormatInt(int64(v/time.Second), 10))
|
||
|
|
io.WriteString(w, str)
|
||
|
|
case map[string]interface{}:
|
||
|
|
io.WriteString(w, "d")
|
||
|
|
for key, val := range v {
|
||
|
|
str := fmt.Sprintf("%s:%s", strconv.Itoa(len(key)), key)
|
||
|
|
io.WriteString(w, str)
|
||
|
|
writeBencoded(w, val)
|
||
|
|
}
|
||
|
|
io.WriteString(w, "e")
|
||
|
|
case []string:
|
||
|
|
io.WriteString(w, "l")
|
||
|
|
for _, val := range v {
|
||
|
|
writeBencoded(w, val)
|
||
|
|
}
|
||
|
|
io.WriteString(w, "e")
|
||
|
|
default:
|
||
|
|
// Although not currently necessary,
|
||
|
|
// should handle []interface{} manually; Go can't do it implicitly
|
||
|
|
panic("Tried to bencode an unsupported type!")
|
||
|
|
}
|
||
|
|
}
|