blob: ba70193d5d390c77ef687d34972a4b95a996ee83 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
package main
import (
// Uncomment this line to pass the first stage
// "encoding/json"
"fmt"
"os"
"strconv"
"unicode"
// bencode "github.com/jackpal/bencode-go" // Available if you need it!
)
// Example:
// - 5:hello -> hello
// - 10:hello12345 -> hello12345
func decodeBencode(bencodedString string) (interface{}, error) {
if unicode.IsDigit(rune(bencodedString[0])) {
var firstColonIndex int
for i := 0; i < len(bencodedString); i++ {
if bencodedString[i] == ':' {
firstColonIndex = i
break
}
}
lengthStr := bencodedString[:firstColonIndex]
length, err := strconv.Atoi(lengthStr)
if err != nil {
return "", err
}
return bencodedString[firstColonIndex+1 : firstColonIndex+1+length], nil
} else {
return "", fmt.Errorf("Only strings are supported at the moment")
}
}
func main() {
// You can use print statements as follows for debugging, they'll be visible when running tests.
fmt.Println("Logs from your program will appear here!")
command := os.Args[1]
if command == "decode" {
// Uncomment this block to pass the first stage
//
// bencodedValue := os.Args[2]
//
// decoded, err := decodeBencode(bencodedValue)
// if err != nil {
// fmt.Println(err)
// return
// }
//
// jsonOutput, _ := json.Marshal(decoded)
// fmt.Println(string(jsonOutput))
} else {
fmt.Println("Unknown command: " + command)
os.Exit(1)
}
}
|
