-
Notifications
You must be signed in to change notification settings - Fork 1.2k
/
debugging.go
113 lines (103 loc) · 3.47 KB
/
debugging.go
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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
// Copyright 2015 Keybase, Inc. All rights reserved. Use of
// this source code is governed by the included BSD license.
package service
import (
"fmt"
"strconv"
"time"
"github.com/keybase/client/go/libkb"
gregor1 "github.com/keybase/client/go/protocol/gregor1"
keybase1 "github.com/keybase/client/go/protocol/keybase1"
"github.com/keybase/go-framed-msgpack-rpc/rpc"
"golang.org/x/net/context"
)
type DebuggingHandler struct {
libkb.Contextified
*BaseHandler
userHandler *UserHandler
walletHandler *walletHandler
}
func NewDebuggingHandler(xp rpc.Transporter, g *libkb.GlobalContext, userHandler *UserHandler, walletHandler *walletHandler) *DebuggingHandler {
return &DebuggingHandler{
Contextified: libkb.NewContextified(g),
BaseHandler: NewBaseHandler(g, xp),
userHandler: userHandler,
walletHandler: walletHandler,
}
}
// See debugging_devel.go for additional scripts.
func (t *DebuggingHandler) Script(ctx context.Context, arg keybase1.ScriptArg) (res string, err error) {
ctx = libkb.WithLogTag(ctx, "DG")
m := libkb.NewMetaContext(ctx, t.G())
defer m.Trace(fmt.Sprintf("Script(%s)", arg.Script), &err)()
args := arg.Args
log := func(format string, args ...interface{}) {
t.G().Log.CInfof(ctx, format, args...)
}
defer time.Sleep(100 * time.Millisecond) // Without this CInfof often doesn't reach the CLI
switch arg.Script {
case "journeycard":
log("journeycard-fastforward [days]")
log("journeycard-resetall")
log("journeycard-state <team-id>")
return "", nil
case "journeycard-fastforward":
uidGregor := gregor1.UID(m.G().ActiveDevice.UID().ToBytes())
advance := 24 * time.Hour
if len(args) >= 1 {
days, err := strconv.Atoi(args[0])
if err != nil {
return "", err
}
advance = time.Duration(days) * 24 * time.Hour
}
nTeams, nConvs, err := t.G().ChatHelper.JourneycardTimeTravel(m.Ctx(), uidGregor, advance)
if err != nil {
return "", err
}
log("time advanced by %v for %v teams and %v convs", advance, nTeams, nConvs)
return "", err
case "journeycard-resetall":
uidGregor := gregor1.UID(m.G().ActiveDevice.UID().ToBytes())
err = t.G().ChatHelper.JourneycardResetAllConvs(m.Ctx(), uidGregor)
if err != nil {
return "", err
}
log("journeycard state has been reset for all convs")
return "", nil
case "journeycard-state":
if len(args) != 1 {
return "", fmt.Errorf("usage: journeycard-state <conv-id> (like 000059aa7f324dad7524b56ed1beb3e3d620b3897d640951710f1417c6b7b85f)")
}
teamID, err := keybase1.TeamIDFromString(args[0])
if err != nil {
return "", err
}
uidGregor := gregor1.UID(m.G().ActiveDevice.UID().ToBytes())
summary, err := t.G().ChatHelper.JourneycardDebugState(m.Ctx(), uidGregor, teamID)
return summary, err
case "":
return "", fmt.Errorf("empty script name")
default:
return t.scriptExtras(ctx, arg)
}
}
func (t *DebuggingHandler) FirstStep(ctx context.Context, arg keybase1.FirstStepArg) (result keybase1.FirstStepResult, err error) {
client := t.rpcClient()
cbArg := keybase1.SecondStepArg{Val: arg.Val + 1, SessionID: arg.SessionID}
var cbReply int
err = client.Call(ctx, "keybase.1.debugging.secondStep", []interface{}{cbArg}, &cbReply, 0)
if err != nil {
return
}
result.ValPlusTwo = cbReply
return
}
func (t *DebuggingHandler) SecondStep(_ context.Context, arg keybase1.SecondStepArg) (val int, err error) {
val = arg.Val + 1
return
}
func (t *DebuggingHandler) Increment(_ context.Context, arg keybase1.IncrementArg) (val int, err error) {
val = arg.Val + 1
return
}