github.com/keybase/client/go@v0.0.0-20241007131713-f10651d043c8/chat/server.go (about) 1 package chat 2 3 import ( 4 "bytes" 5 "encoding/hex" 6 "encoding/json" 7 "errors" 8 "fmt" 9 "io" 10 "os" 11 "regexp" 12 "sort" 13 "strings" 14 "sync" 15 "time" 16 17 "github.com/keybase/client/go/chat/attachments" 18 "github.com/keybase/client/go/chat/globals" 19 "github.com/keybase/client/go/chat/search" 20 "github.com/keybase/client/go/chat/storage" 21 "github.com/keybase/client/go/chat/types" 22 "github.com/keybase/client/go/chat/unfurl" 23 "github.com/keybase/client/go/chat/utils" 24 "github.com/keybase/client/go/libkb" 25 "github.com/keybase/client/go/protocol/chat1" 26 "github.com/keybase/client/go/protocol/gregor1" 27 "github.com/keybase/client/go/protocol/keybase1" 28 "github.com/keybase/client/go/teambot" 29 "github.com/keybase/client/go/teams" 30 "github.com/keybase/client/go/teams/opensearch" 31 "github.com/keybase/pipeliner" 32 "golang.org/x/net/context" 33 "golang.org/x/sync/errgroup" 34 "golang.org/x/sync/semaphore" 35 ) 36 37 type UISource interface { 38 GetChatUI(sessionID int) libkb.ChatUI 39 GetStreamUICli() *keybase1.StreamUiClient 40 } 41 42 type Server struct { 43 globals.Contextified 44 utils.DebugLabeler 45 46 serverConn types.ServerConnection 47 uiSource UISource 48 boxer *Boxer 49 identNotifier types.IdentifyNotifier 50 51 searchMu sync.Mutex 52 searchInboxMu sync.Mutex 53 loadGalleryMu sync.Mutex 54 searchCancelFn context.CancelFunc 55 searchInboxCancelFn context.CancelFunc 56 loadGalleryCancelFn context.CancelFunc 57 58 fileAttachmentDownloadConfigurationMu sync.RWMutex 59 fileAttachmentDownloadCacheDir string 60 fileAttachmentDownloadDownloadDir string 61 62 // Only for testing 63 rc chat1.RemoteInterface 64 mockChatUI libkb.ChatUI 65 } 66 67 var _ chat1.LocalInterface = (*Server)(nil) 68 69 func NewServer(g *globals.Context, serverConn types.ServerConnection, uiSource UISource) *Server { 70 return &Server{ 71 Contextified: globals.NewContextified(g), 72 DebugLabeler: utils.NewDebugLabeler(g.ExternalG(), "Server", false), 73 serverConn: serverConn, 74 uiSource: uiSource, 75 boxer: NewBoxer(g), 76 identNotifier: NewCachingIdentifyNotifier(g), 77 fileAttachmentDownloadCacheDir: g.GetEnv().GetCacheDir(), 78 fileAttachmentDownloadDownloadDir: g.GetEnv().GetDownloadsDir(), 79 } 80 } 81 82 func (h *Server) getChatUI(sessionID int) libkb.ChatUI { 83 if h.mockChatUI != nil { 84 return h.mockChatUI 85 } 86 return h.uiSource.GetChatUI(sessionID) 87 } 88 89 func (h *Server) getStreamUICli() *keybase1.StreamUiClient { 90 return h.uiSource.GetStreamUICli() 91 } 92 93 func (h *Server) shouldSquashError(err error) bool { 94 // these are not offline errors, but we never want the JS to receive them and potentially 95 // display a black bar 96 switch terr := err.(type) { 97 case storage.AbortedError: 98 return true 99 case TransientUnboxingError: 100 return h.shouldSquashError(terr.Inner()) 101 } 102 switch err { 103 case utils.ErrConvLockTabDeadlock, context.Canceled: 104 return true 105 } 106 return false 107 } 108 109 func (h *Server) squashSquashableErrors(err error) error { 110 if h.shouldSquashError(err) { 111 return nil 112 } 113 return err 114 } 115 116 func (h *Server) handleOfflineError(ctx context.Context, err error, 117 res chat1.OfflinableResult) error { 118 if err == nil { 119 return nil 120 } 121 if h.shouldSquashError(err) { 122 return nil 123 } 124 125 errKind := IsOfflineError(err) 126 h.Debug(ctx, "handleOfflineError: errType: %T", err) 127 if errKind != OfflineErrorKindOnline { 128 h.Debug(ctx, "handleOfflineError: setting offline: err: %s", err) 129 res.SetOffline() 130 switch errKind { 131 case OfflineErrorKindOfflineReconnect: 132 // Reconnect Gregor if we think we are offline (and told to reconnect) 133 h.Debug(ctx, "handleOfflineError: reconnecting to gregor") 134 if _, err := h.serverConn.Reconnect(ctx); err != nil { 135 h.Debug(ctx, "handleOfflineError: error reconnecting: %s", err) 136 } else { 137 h.Debug(ctx, "handleOfflineError: success reconnecting") 138 } 139 default: 140 // Nothing to do for other errors. 141 } 142 return nil 143 } 144 return err 145 } 146 147 func (h *Server) setResultRateLimit(ctx context.Context, res types.RateLimitedResult) { 148 res.SetRateLimits(globals.CtxRateLimits(ctx)) 149 } 150 151 func (h *Server) suspendBgConvLoads(ctx context.Context) func() { 152 return utils.SuspendComponents(ctx, h.G(), []types.Suspendable{ 153 h.G().ConvLoader, 154 h.G().Indexer, 155 }) 156 } 157 158 func (h *Server) suspendInboxSource(ctx context.Context) func() { 159 return utils.SuspendComponent(ctx, h.G(), h.G().InboxSource) 160 } 161 162 func (h *Server) RequestInboxLayout(ctx context.Context, reselectMode chat1.InboxLayoutReselectMode) (err error) { 163 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, nil) 164 defer h.Trace(ctx, &err, "RequestInboxLayout")() 165 h.G().UIInboxLoader.UpdateLayout(ctx, reselectMode, "UI request") 166 return nil 167 } 168 169 func (h *Server) RequestInboxUnbox(ctx context.Context, convIDs []chat1.ConversationID) (err error) { 170 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, nil) 171 ctx = globals.CtxAddLocalizerCancelable(ctx) 172 defer h.Trace(ctx, &err, "RequestInboxUnbox")() 173 defer h.PerfTrace(ctx, &err, "RequestInboxUnbox")() 174 for _, convID := range convIDs { 175 h.GetPerfLog().CDebugf(ctx, "RequestInboxUnbox: queuing unbox for: %s", convID) 176 h.Debug(ctx, "RequestInboxUnbox: queuing unbox for: %s", convID) 177 } 178 if err := h.G().UIInboxLoader.UpdateConvs(ctx, convIDs); err != nil { 179 h.Debug(ctx, "RequestInboxUnbox: failed to update convs: %s", err) 180 } 181 return nil 182 } 183 184 func (h *Server) RequestInboxSmallIncrease(ctx context.Context) (err error) { 185 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, nil) 186 defer h.Trace(ctx, &err, "RequestInboxSmallIncrease")() 187 h.G().UIInboxLoader.UpdateLayoutFromSmallIncrease(ctx) 188 return nil 189 } 190 191 func (h *Server) RequestInboxSmallReset(ctx context.Context) (err error) { 192 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, nil) 193 defer h.Trace(ctx, &err, "RequestInboxSmallReset")() 194 h.G().UIInboxLoader.UpdateLayoutFromSmallReset(ctx) 195 return nil 196 } 197 198 func (h *Server) GetInboxNonblockLocal(ctx context.Context, arg chat1.GetInboxNonblockLocalArg) (res chat1.NonblockFetchRes, err error) { 199 var breaks []keybase1.TLFIdentifyFailure 200 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &breaks, h.identNotifier) 201 ctx = globals.CtxAddLocalizerCancelable(ctx) 202 defer h.Trace(ctx, &err, "GetInboxNonblockLocal")() 203 defer h.PerfTrace(ctx, &err, "GetInboxNonblockLocal")() 204 defer func() { h.setResultRateLimit(ctx, &res) }() 205 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 206 defer func() { 207 if res.Offline { 208 h.Debug(ctx, "GetInboxNonblockLocal: result obtained offline") 209 } 210 }() 211 defer h.suspendBgConvLoads(ctx)() 212 213 if err := h.G().UIInboxLoader.LoadNonblock(ctx, arg.Query, arg.MaxUnbox, 214 arg.SkipUnverified); err != nil { 215 return res, err 216 } 217 res.Offline = h.G().InboxSource.IsOffline(ctx) 218 res.IdentifyFailures = breaks 219 return res, nil 220 } 221 222 func (h *Server) MarkAsReadLocal(ctx context.Context, arg chat1.MarkAsReadLocalArg) (res chat1.MarkAsReadLocalRes, err error) { 223 var identBreaks []keybase1.TLFIdentifyFailure 224 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, &identBreaks, 225 h.identNotifier) 226 defer h.Trace(ctx, &err, 227 fmt.Sprintf("MarkAsReadLocal(%s, %v)", arg.ConversationID, arg.MsgID))() 228 defer func() { h.setResultRateLimit(ctx, &res) }() 229 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 230 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 231 if err != nil { 232 h.Debug(ctx, "MarkAsRead: not logged in: %s", err) 233 return chat1.MarkAsReadLocalRes{}, nil 234 } 235 // Don't send remote mark as read if we somehow get this in the background. 236 if h.G().MobileAppState.State() != keybase1.MobileAppState_FOREGROUND { 237 h.Debug(ctx, "MarkAsReadLocal: not marking as read, app state not foreground: %v", 238 h.G().MobileAppState.State()) 239 return chat1.MarkAsReadLocalRes{ 240 Offline: h.G().InboxSource.IsOffline(ctx), 241 }, nil 242 } 243 if err = h.G().InboxSource.MarkAsRead(ctx, arg.ConversationID, uid, arg.MsgID, arg.ForceUnread); err != nil { 244 switch err { 245 case utils.ErrGetUnverifiedConvNotFound, utils.ErrGetVerifiedConvNotFound: 246 // if we couldn't find the conv, then just act like it worked 247 default: 248 return res, err 249 } 250 } 251 return chat1.MarkAsReadLocalRes{ 252 Offline: h.G().InboxSource.IsOffline(ctx), 253 }, nil 254 } 255 256 func (h *Server) MarkTLFAsReadLocal(ctx context.Context, arg chat1.MarkTLFAsReadLocalArg) (res chat1.MarkTLFAsReadLocalRes, err error) { 257 var identBreaks []keybase1.TLFIdentifyFailure 258 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, &identBreaks, nil) 259 defer h.Trace(ctx, &err, "MarkTLFAsRead")() 260 defer func() { h.setResultRateLimit(ctx, &res) }() 261 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 262 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 263 if err != nil { 264 return res, err 265 } 266 convs, err := h.G().TeamChannelSource.GetChannelsFull(ctx, uid, arg.TlfID, chat1.TopicType_CHAT) 267 if err != nil { 268 return res, err 269 } 270 epick := libkb.FirstErrorPicker{} 271 for _, conv := range convs { 272 _, err = h.MarkAsReadLocal(ctx, chat1.MarkAsReadLocalArg{ 273 ConversationID: conv.GetConvID(), 274 MsgID: &conv.ReaderInfo.MaxMsgid, 275 }) 276 epick.Push(err) 277 } 278 return chat1.MarkTLFAsReadLocalRes{ 279 Offline: h.G().InboxSource.IsOffline(ctx), 280 }, epick.Error() 281 } 282 283 // GetInboxAndUnboxLocal implements keybase.chatLocal.getInboxAndUnboxLocal protocol. 284 func (h *Server) GetInboxAndUnboxLocal(ctx context.Context, arg chat1.GetInboxAndUnboxLocalArg) (res chat1.GetInboxAndUnboxLocalRes, err error) { 285 var identBreaks []keybase1.TLFIdentifyFailure 286 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, h.identNotifier) 287 if arg.Query != nil && arg.Query.TopicType != nil && *arg.Query.TopicType != chat1.TopicType_CHAT { 288 // make this cancelable for things like KBFS file edit convs 289 ctx = globals.CtxAddLocalizerCancelable(ctx) 290 } 291 defer h.Trace(ctx, &err, "GetInboxAndUnboxLocal")() 292 defer func() { h.setResultRateLimit(ctx, &res) }() 293 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 294 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 295 if err != nil { 296 return res, err 297 } 298 // Ignore these requests on mobile 299 if h.G().IsMobileAppType() && arg.Query != nil && arg.Query.TopicType != nil && 300 *arg.Query.TopicType == chat1.TopicType_KBFSFILEEDIT { 301 return chat1.GetInboxAndUnboxLocalRes{ 302 IdentifyFailures: identBreaks, 303 }, nil 304 } 305 306 // Read inbox from the source 307 ib, _, err := h.G().InboxSource.Read(ctx, uid, types.ConversationLocalizerBlocking, 308 types.InboxSourceDataSourceAll, nil, arg.Query) 309 switch err.(type) { 310 case nil: 311 case UnknownTLFNameError: 312 h.Debug(ctx, "GetInboxAndUnboxLocal: got unknown TLF name error, returning blank results") 313 ib.Convs = nil 314 default: 315 return res, err 316 } 317 318 return chat1.GetInboxAndUnboxLocalRes{ 319 Conversations: ib.Convs, 320 Offline: h.G().InboxSource.IsOffline(ctx), 321 IdentifyFailures: identBreaks, 322 }, nil 323 } 324 325 func (h *Server) GetInboxAndUnboxUILocal(ctx context.Context, arg chat1.GetInboxAndUnboxUILocalArg) (res chat1.GetInboxAndUnboxUILocalRes, err error) { 326 var identBreaks []keybase1.TLFIdentifyFailure 327 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, h.identNotifier) 328 defer h.Trace(ctx, &err, "GetInboxAndUnboxUILocal")() 329 defer func() { h.setResultRateLimit(ctx, &res) }() 330 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 331 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 332 if err != nil { 333 return res, err 334 } 335 // Read inbox from the source 336 ib, _, err := h.G().InboxSource.Read(ctx, uid, types.ConversationLocalizerBlocking, 337 types.InboxSourceDataSourceAll, nil, arg.Query) 338 switch err.(type) { 339 case nil: 340 case UnknownTLFNameError: 341 h.Debug(ctx, "GetInboxAndUnboxUILocal: got unknown TLF name error, returning blank results") 342 ib.Convs = nil 343 default: 344 return res, err 345 } 346 return chat1.GetInboxAndUnboxUILocalRes{ 347 Conversations: utils.PresentConversationLocals(ctx, h.G(), uid, ib.Convs, 348 utils.PresentParticipantsModeInclude), 349 IdentifyFailures: identBreaks, 350 }, nil 351 } 352 353 // GetThreadLocal implements keybase.chatLocal.getThreadLocal protocol. 354 func (h *Server) GetThreadLocal(ctx context.Context, arg chat1.GetThreadLocalArg) (res chat1.GetThreadLocalRes, err error) { 355 var identBreaks []keybase1.TLFIdentifyFailure 356 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, h.identNotifier) 357 defer h.Trace(ctx, &err, "GetThreadLocal")() 358 defer func() { h.setResultRateLimit(ctx, &res) }() 359 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 360 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 361 if err != nil { 362 return chat1.GetThreadLocalRes{}, err 363 } 364 thread, err := h.G().UIThreadLoader.Load(ctx, uid, arg.ConversationID, arg.Reason, nil, arg.Query, 365 arg.Pagination) 366 if err != nil { 367 return chat1.GetThreadLocalRes{}, err 368 } 369 return chat1.GetThreadLocalRes{ 370 Thread: thread, 371 IdentifyFailures: identBreaks, 372 }, nil 373 } 374 375 func (h *Server) GetUnreadline(ctx context.Context, arg chat1.GetUnreadlineArg) (res chat1.UnreadlineRes, err error) { 376 var identBreaks []keybase1.TLFIdentifyFailure 377 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, h.identNotifier) 378 defer h.Trace(ctx, &err, 379 fmt.Sprintf("GetUnreadline: convID: %v, readMsgID: %v", arg.ConvID, arg.ReadMsgID))() 380 defer func() { h.setResultRateLimit(ctx, &res) }() 381 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 382 383 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 384 if err != nil { 385 return res, err 386 } 387 388 res.UnreadlineID, err = h.G().ConvSource.GetUnreadline(ctx, arg.ConvID, uid, arg.ReadMsgID) 389 if err != nil { 390 h.Debug(ctx, "GetUnreadline: unable to run UnreadMsgID: %v", err) 391 return res, err 392 } 393 return res, nil 394 } 395 396 func (h *Server) GetThreadNonblock(ctx context.Context, arg chat1.GetThreadNonblockArg) (res chat1.NonblockFetchRes, err error) { 397 var identBreaks []keybase1.TLFIdentifyFailure 398 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, h.identNotifier) 399 defer h.Trace(ctx, &err, 400 fmt.Sprintf("GetThreadNonblock(%s,%v,%v)", arg.ConversationID, arg.CbMode, arg.Reason))() 401 defer h.PerfTrace(ctx, &err, 402 fmt.Sprintf("GetThreadNonblock(%s,%v,%v)", arg.ConversationID, arg.CbMode, arg.Reason))() 403 defer func() { h.setResultRateLimit(ctx, &res) }() 404 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 405 defer h.suspendBgConvLoads(ctx)() 406 defer h.suspendInboxSource(ctx)() 407 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 408 if err != nil { 409 return chat1.NonblockFetchRes{}, err 410 } 411 chatUI := h.getChatUI(arg.SessionID) 412 return res, h.G().UIThreadLoader.LoadNonblock(ctx, chatUI, uid, arg.ConversationID, arg.Reason, 413 arg.Pgmode, arg.CbMode, arg.KnownRemotes, arg.Query, arg.Pagination) 414 } 415 416 func (h *Server) NewConversationsLocal(ctx context.Context, arg chat1.NewConversationsLocalArg) (res chat1.NewConversationsLocalRes, err error) { 417 var identBreaks []keybase1.TLFIdentifyFailure 418 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, h.identNotifier) 419 defer h.Trace(ctx, &err, fmt.Sprintf("NewConversationsLocal(len=%d)", len(arg.NewConversationLocalArguments)))() 420 defer func() { h.setResultRateLimit(ctx, &res) }() 421 422 _, err = utils.AssertLoggedInUID(ctx, h.G()) 423 if err != nil { 424 return chat1.NewConversationsLocalRes{}, err 425 } 426 427 var errs []error 428 for _, convArg := range arg.NewConversationLocalArguments { 429 var result chat1.NewConversationsLocalResult 430 newConvRes, err := h.NewConversationLocal(ctx, chat1.NewConversationLocalArg{ 431 TlfName: convArg.TlfName, 432 TopicType: convArg.TopicType, 433 TlfVisibility: convArg.TlfVisibility, 434 TopicName: convArg.TopicName, 435 MembersType: convArg.MembersType, 436 IdentifyBehavior: arg.IdentifyBehavior, 437 }) 438 if err != nil { 439 e := err.Error() 440 result.Err = &e 441 errs = append(errs, err) 442 } else { 443 result.Result = new(chat1.NewConversationLocalRes) 444 result.Result.Conv = newConvRes.Conv 445 result.Result.UiConv = newConvRes.UiConv 446 } 447 res.Results = append(res.Results, result) 448 } 449 450 res.IdentifyFailures = identBreaks 451 return res, libkb.CombineErrors(errs...) 452 } 453 454 // NewConversationLocal implements keybase.chatLocal.newConversationLocal protocol. 455 // Create a new conversation. Or in the case of CHAT, create-or-get a conversation. 456 func (h *Server) NewConversationLocal(ctx context.Context, arg chat1.NewConversationLocalArg) (res chat1.NewConversationLocalRes, err error) { 457 var identBreaks []keybase1.TLFIdentifyFailure 458 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, h.identNotifier) 459 defer h.Trace(ctx, &err, 460 fmt.Sprintf("NewConversationLocal(%s|%v)", arg.TlfName, arg.MembersType))() 461 defer func() { h.setResultRateLimit(ctx, &res) }() 462 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 463 if err != nil { 464 return chat1.NewConversationLocalRes{}, err 465 } 466 467 conv, created, err := NewConversation(ctx, h.G(), uid, arg.TlfName, arg.TopicName, 468 arg.TopicType, arg.MembersType, arg.TlfVisibility, nil, h.remoteClient, NewConvFindExistingNormal) 469 if err != nil { 470 h.Debug(ctx, "NewConversationLocal: failed to make conv: %s", err) 471 return res, err 472 } 473 474 res.Conv = conv 475 res.UiConv = utils.PresentConversationLocal(ctx, h.G(), uid, conv, utils.PresentParticipantsModeInclude) 476 res.IdentifyFailures = identBreaks 477 478 // If we are making a new channel in a team, send a system message to 479 // indicate this. 480 if created && arg.MembersType == chat1.ConversationMembersType_TEAM && 481 arg.TopicType == chat1.TopicType_CHAT && 482 arg.TopicName != nil && *arg.TopicName != globals.DefaultTeamTopic { 483 subBody := chat1.NewMessageSystemWithNewchannel(chat1.MessageSystemNewChannel{ 484 Creator: h.G().Env.GetUsername().String(), 485 NameAtCreation: *arg.TopicName, 486 ConvID: conv.GetConvID(), 487 }) 488 body := chat1.NewMessageBodyWithSystem(subBody) 489 err = h.G().ChatHelper.SendMsgByName(ctx, conv.Info.TlfName, 490 &globals.DefaultTeamTopic, arg.MembersType, keybase1.TLFIdentifyBehavior_CHAT_CLI, 491 body, chat1.MessageType_SYSTEM) 492 if err != nil { 493 h.Debug(ctx, "NewConversationLocal: unable to post new channel system message: %v", err) 494 } 495 } 496 497 // If we have this conv hidden, then let's bring it back before returning 498 if !utils.GetConversationStatusBehavior(conv.Info.Status).ShowInInbox { 499 h.Debug(ctx, "NewConversationLocal: new conversation not shown, unhiding: %s", conv.GetConvID()) 500 if err := h.G().InboxSource.RemoteSetConversationStatus(ctx, uid, conv.GetConvID(), 501 chat1.ConversationStatus_UNFILED); err != nil { 502 h.Debug(ctx, "NewConversationLocal: unable to unhide conv: %s", err) 503 return res, err 504 } 505 } 506 507 return res, nil 508 } 509 510 func (h *Server) limitConvResults(ctx context.Context, uid gregor1.UID, allConvs []types.RemoteConversation, 511 num int) ([]chat1.ConversationLocal, error) { 512 var convs []types.RemoteConversation 513 sort.Sort(utils.RemoteConvByMtime(allConvs)) 514 if len(allConvs) <= num { 515 convs = allConvs 516 } else { 517 convs = allConvs[:num] 518 } 519 locals, _, err := h.G().InboxSource.Localize(ctx, uid, convs, types.ConversationLocalizerBlocking) 520 if err != nil { 521 return nil, err 522 } 523 return locals, nil 524 } 525 526 func (h *Server) GetInboxSummaryForCLILocal(ctx context.Context, arg chat1.GetInboxSummaryForCLILocalQuery) (res chat1.GetInboxSummaryForCLILocalRes, err error) { 527 var identBreaks []keybase1.TLFIdentifyFailure 528 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_CLI, &identBreaks, 529 h.identNotifier) 530 defer h.Trace(ctx, &err, "GetInboxSummaryForCLILocal")() 531 defer func() { h.setResultRateLimit(ctx, &res) }() 532 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 533 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 534 if err != nil { 535 return chat1.GetInboxSummaryForCLILocalRes{}, err 536 } 537 538 var after time.Time 539 if len(arg.After) > 0 { 540 after, err = utils.ParseTimeFromRFC3339OrDurationFromPast(h.G(), arg.After) 541 if err != nil { 542 return res, fmt.Errorf("parsing time or duration (%s) error: %s", arg.After, err) 543 } 544 } 545 var before time.Time 546 if len(arg.Before) > 0 { 547 before, err = utils.ParseTimeFromRFC3339OrDurationFromPast(h.G(), arg.Before) 548 if err != nil { 549 return res, fmt.Errorf("parsing time or duration (%s) error: %s", arg.Before, err) 550 } 551 } 552 553 var queryBase chat1.GetInboxQuery 554 queryBase.ComputeActiveList = true 555 queryBase.OneChatTypePerTLF = new(bool) 556 *queryBase.OneChatTypePerTLF = true 557 if !after.IsZero() { 558 gafter := gregor1.ToTime(after) 559 queryBase.After = &gafter 560 } 561 if !before.IsZero() { 562 gbefore := gregor1.ToTime(before) 563 queryBase.Before = &gbefore 564 } 565 if arg.TopicType != chat1.TopicType_NONE { 566 queryBase.TopicType = &arg.TopicType 567 } 568 if arg.Visibility != keybase1.TLFVisibility_ANY { 569 queryBase.TlfVisibility = &arg.Visibility 570 } 571 queryBase.Status = arg.Status 572 queryBase.ConvIDs = arg.ConvIDs 573 574 var gires chat1.GetInboxAndUnboxLocalRes 575 if arg.UnreadFirst { 576 if arg.UnreadFirstLimit.AtMost <= 0 { 577 arg.UnreadFirstLimit.AtMost = int(^uint(0) >> 1) // maximum int 578 } 579 if arg.UnreadFirstLimit.AtMost < arg.UnreadFirstLimit.AtLeast { 580 arg.UnreadFirstLimit.AtMost = arg.UnreadFirstLimit.AtLeast 581 } 582 query := queryBase 583 query.UnreadOnly, query.ReadOnly = true, false 584 ib, err := h.G().InboxSource.ReadUnverified(ctx, uid, types.InboxSourceDataSourceAll, &query) 585 if err != nil { 586 return chat1.GetInboxSummaryForCLILocalRes{}, err 587 } 588 res.Conversations, err = h.limitConvResults(ctx, uid, ib.ConvsUnverified, 589 arg.UnreadFirstLimit.AtMost) 590 if err != nil { 591 return chat1.GetInboxSummaryForCLILocalRes{}, err 592 } 593 594 more := utils.Collar( 595 arg.UnreadFirstLimit.AtLeast-len(res.Conversations), 596 arg.UnreadFirstLimit.NumRead, 597 arg.UnreadFirstLimit.AtMost-len(res.Conversations), 598 ) 599 if more > 0 { 600 query := queryBase 601 query.UnreadOnly, query.ReadOnly = false, true 602 ib, err := h.G().InboxSource.ReadUnverified(ctx, uid, types.InboxSourceDataSourceAll, &query) 603 if err != nil { 604 return chat1.GetInboxSummaryForCLILocalRes{}, err 605 } 606 moreConvs, err := h.limitConvResults(ctx, uid, ib.ConvsUnverified, more) 607 if err != nil { 608 return chat1.GetInboxSummaryForCLILocalRes{}, err 609 } 610 res.Conversations = append(res.Conversations, moreConvs...) 611 } 612 } else { 613 if arg.ActivitySortedLimit <= 0 { 614 arg.ActivitySortedLimit = int(^uint(0) >> 1) // maximum int 615 } 616 query := queryBase 617 query.UnreadOnly, query.ReadOnly = false, false 618 ib, err := h.G().InboxSource.ReadUnverified(ctx, uid, types.InboxSourceDataSourceAll, &query) 619 if err != nil { 620 return chat1.GetInboxSummaryForCLILocalRes{}, err 621 } 622 res.Conversations, err = h.limitConvResults(ctx, uid, ib.ConvsUnverified, arg.ActivitySortedLimit) 623 if err != nil { 624 return chat1.GetInboxSummaryForCLILocalRes{}, err 625 } 626 } 627 res.Offline = gires.Offline 628 return res, nil 629 } 630 631 func (h *Server) GetConversationForCLILocal(ctx context.Context, arg chat1.GetConversationForCLILocalQuery) (res chat1.GetConversationForCLILocalRes, err error) { 632 var identBreaks []keybase1.TLFIdentifyFailure 633 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_CLI, &identBreaks, 634 h.identNotifier) 635 defer h.Trace(ctx, &err, "GetConversationForCLILocal")() 636 defer func() { h.setResultRateLimit(ctx, &res) }() 637 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 638 if _, err = utils.AssertLoggedInUID(ctx, h.G()); err != nil { 639 return res, err 640 } 641 642 if arg.Limit.AtMost <= 0 { 643 arg.Limit.AtMost = int(^uint(0) >> 1) // maximum int 644 } 645 if arg.Limit.AtMost < arg.Limit.AtLeast { 646 arg.Limit.AtMost = arg.Limit.AtLeast 647 } 648 649 convLocal := arg.Conv 650 651 var since time.Time 652 if arg.Since != nil { 653 since, err = utils.ParseTimeFromRFC3339OrDurationFromPast(h.G(), *arg.Since) 654 if err != nil { 655 return res, fmt.Errorf("parsing time or duration (%s) error: %s", *arg.Since, since) 656 } 657 } 658 659 query := chat1.GetThreadQuery{ 660 MarkAsRead: arg.MarkAsRead, 661 MessageTypes: arg.MessageTypes, 662 } 663 if !since.IsZero() { 664 gsince := gregor1.ToTime(since) 665 query.After = &gsince 666 } 667 668 tv, err := h.GetThreadLocal(ctx, chat1.GetThreadLocalArg{ 669 ConversationID: convLocal.Info.Id, 670 Query: &query, 671 IdentifyBehavior: keybase1.TLFIdentifyBehavior_CHAT_CLI, 672 }) 673 if err != nil { 674 return chat1.GetConversationForCLILocalRes{}, err 675 } 676 677 // apply message count limits 678 var messages []chat1.MessageUnboxed 679 for _, m := range tv.Thread.Messages { 680 messages = append(messages, m) 681 682 arg.Limit.AtMost-- 683 arg.Limit.AtLeast-- 684 if m.GetMessageID() <= convLocal.ReaderInfo.ReadMsgid { 685 arg.Limit.NumRead-- 686 } 687 if arg.Limit.AtMost <= 0 || 688 (arg.Limit.NumRead <= 0 && arg.Limit.AtLeast <= 0) { 689 break 690 } 691 } 692 693 return chat1.GetConversationForCLILocalRes{ 694 Conversation: convLocal, 695 Messages: messages, 696 Offline: tv.Offline, 697 }, nil 698 } 699 700 func (h *Server) GetMessagesLocal(ctx context.Context, arg chat1.GetMessagesLocalArg) (res chat1.GetMessagesLocalRes, err error) { 701 var identBreaks []keybase1.TLFIdentifyFailure 702 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, h.identNotifier) 703 defer h.Trace(ctx, &err, "GetMessagesLocal")() 704 defer func() { h.setResultRateLimit(ctx, &res) }() 705 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 706 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 707 if err != nil { 708 return res, err 709 } 710 reason := chat1.GetThreadReason_GENERAL 711 messages, err := h.G().ChatHelper.GetMessages(ctx, uid, arg.ConversationID, arg.MessageIDs, 712 !arg.DisableResolveSupersedes, &reason) 713 if err != nil { 714 return res, err 715 } 716 return chat1.GetMessagesLocalRes{ 717 Messages: messages, 718 IdentifyFailures: identBreaks, 719 }, nil 720 } 721 722 func (h *Server) GetNextAttachmentMessageLocal(ctx context.Context, 723 arg chat1.GetNextAttachmentMessageLocalArg) (res chat1.GetNextAttachmentMessageLocalRes, err error) { 724 var identBreaks []keybase1.TLFIdentifyFailure 725 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, h.identNotifier) 726 defer h.Trace(ctx, &err, "GetNextAttachmentMessageLocal(%s,%d,%v)", 727 arg.ConvID, arg.MessageID, arg.BackInTime)() 728 defer func() { h.setResultRateLimit(ctx, &res) }() 729 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 730 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 731 if err != nil { 732 return res, err 733 } 734 gallery := attachments.NewGallery(h.G()) 735 unboxed, _, err := gallery.NextMessage(ctx, uid, arg.ConvID, arg.MessageID, 736 attachments.NextMessageOptions{ 737 BackInTime: arg.BackInTime, 738 AssetTypes: arg.AssetTypes, 739 }, 740 ) 741 if err != nil { 742 return res, err 743 } 744 var resMsg *chat1.UIMessage 745 if unboxed != nil { 746 resMsg = new(chat1.UIMessage) 747 *resMsg = utils.PresentMessageUnboxed(ctx, h.G(), *unboxed, uid, arg.ConvID) 748 } 749 return chat1.GetNextAttachmentMessageLocalRes{ 750 Message: resMsg, 751 IdentifyFailures: identBreaks, 752 }, nil 753 } 754 755 func (h *Server) SetConversationStatusLocal(ctx context.Context, arg chat1.SetConversationStatusLocalArg) (res chat1.SetConversationStatusLocalRes, err error) { 756 var identBreaks []keybase1.TLFIdentifyFailure 757 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, h.identNotifier) 758 defer h.Trace(ctx, &err, fmt.Sprintf("SetConversationStatusLocal: %v", arg.Status))() 759 defer func() { h.setResultRateLimit(ctx, &res) }() 760 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 761 if err != nil { 762 return res, err 763 } 764 err = h.G().InboxSource.RemoteSetConversationStatus(ctx, uid, arg.ConversationID, arg.Status) 765 return chat1.SetConversationStatusLocalRes{ 766 IdentifyFailures: identBreaks, 767 }, err 768 } 769 770 // PostLocal implements keybase.chatLocal.postLocal protocol. 771 func (h *Server) PostLocal(ctx context.Context, arg chat1.PostLocalArg) (res chat1.PostLocalRes, err error) { 772 var identBreaks []keybase1.TLFIdentifyFailure 773 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, h.identNotifier) 774 defer h.Trace(ctx, &err, "PostLocal")() 775 defer func() { h.setResultRateLimit(ctx, &res) }() 776 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 777 if err != nil { 778 return res, err 779 } 780 781 // Check for any slash command hits for an execute 782 if handled, err := h.G().CommandsSource.AttemptBuiltinCommand(ctx, uid, arg.ConversationID, 783 arg.Msg.ClientHeader.TlfName, arg.Msg.MessageBody, arg.ReplyTo); handled { 784 h.Debug(ctx, "PostLocal: handled slash command with error: %s", err) 785 return res, nil 786 } 787 788 // Run Stellar UI on any payments in the body 789 if !arg.SkipInChatPayments { 790 if arg.Msg.MessageBody, err = h.runStellarSendUI(ctx, arg.SessionID, uid, arg.ConversationID, 791 arg.Msg.MessageBody, arg.ReplyTo); err != nil { 792 return res, err 793 } 794 } 795 796 var prepareOpts chat1.SenderPrepareOptions 797 prepareOpts.ReplyTo = arg.ReplyTo 798 sender := NewBlockingSender(h.G(), h.boxer, h.remoteClient) 799 _, msgBoxed, err := sender.Send(ctx, arg.ConversationID, arg.Msg, 0, nil, nil, &prepareOpts) 800 if err != nil { 801 h.Debug(ctx, "PostLocal: unable to send message: %s", err.Error()) 802 return res, err 803 } 804 805 return chat1.PostLocalRes{ 806 MessageID: msgBoxed.GetMessageID(), 807 IdentifyFailures: identBreaks, 808 }, nil 809 } 810 811 func (h *Server) PostDeleteNonblock(ctx context.Context, arg chat1.PostDeleteNonblockArg) (chat1.PostLocalNonblockRes, error) { 812 var parg chat1.PostLocalNonblockArg 813 parg.ClientPrev = arg.ClientPrev 814 parg.ConversationID = arg.ConversationID 815 parg.IdentifyBehavior = arg.IdentifyBehavior 816 parg.OutboxID = arg.OutboxID 817 parg.Msg.ClientHeader.MessageType = chat1.MessageType_DELETE 818 parg.Msg.ClientHeader.Supersedes = arg.Supersedes 819 parg.Msg.ClientHeader.TlfName = arg.TlfName 820 parg.Msg.ClientHeader.TlfPublic = arg.TlfPublic 821 822 return h.PostLocalNonblock(ctx, parg) 823 } 824 825 func (h *Server) PostEditNonblock(ctx context.Context, arg chat1.PostEditNonblockArg) (res chat1.PostLocalNonblockRes, err error) { 826 var identBreaks []keybase1.TLFIdentifyFailure 827 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, h.identNotifier) 828 defer h.Trace(ctx, &err, "PostEditNonblock")() 829 830 var parg chat1.PostLocalNonblockArg 831 var supersedes chat1.MessageID 832 if arg.Target.MessageID != nil && *arg.Target.MessageID > 0 { 833 supersedes = *arg.Target.MessageID 834 } 835 if supersedes.IsNil() && arg.Target.OutboxID == nil { 836 return res, errors.New("must specify a messageID or outboxID for edit") 837 } 838 parg.ClientPrev = arg.ClientPrev 839 parg.ConversationID = arg.ConversationID 840 parg.IdentifyBehavior = arg.IdentifyBehavior 841 parg.OutboxID = arg.OutboxID 842 parg.Msg.ClientHeader.MessageType = chat1.MessageType_EDIT 843 parg.Msg.ClientHeader.Supersedes = supersedes 844 parg.Msg.ClientHeader.TlfName = arg.TlfName 845 parg.Msg.ClientHeader.TlfPublic = arg.TlfPublic 846 parg.Msg.MessageBody = chat1.NewMessageBodyWithEdit(chat1.MessageEdit{ 847 MessageID: supersedes, 848 Body: arg.Body, 849 }) 850 if supersedes.IsNil() { 851 h.Debug(ctx, "PostEditNonblock: setting supersedes outboxID: %s", arg.Target.OutboxID) 852 parg.Msg.SupersedesOutboxID = arg.Target.OutboxID 853 } 854 return h.PostLocalNonblock(ctx, parg) 855 } 856 857 func (h *Server) runStellarSendUI(ctx context.Context, sessionID int, uid gregor1.UID, 858 convID chat1.ConversationID, msgBody chat1.MessageBody, replyTo *chat1.MessageID) (res chat1.MessageBody, err error) { 859 defer h.Trace(ctx, &err, "runStellarSendUI")() 860 ui := h.getChatUI(sessionID) 861 bodyTyp, err := msgBody.MessageType() 862 if err != nil || bodyTyp != chat1.MessageType_TEXT { 863 return msgBody, nil 864 } 865 body := msgBody.Text().Body 866 parsedPayments := h.G().StellarSender.ParsePayments(ctx, uid, convID, body, replyTo) 867 if len(parsedPayments) == 0 { 868 h.Debug(ctx, "runStellarSendUI: no payments") 869 return msgBody, nil 870 } 871 h.Debug(ctx, "runStellarSendUI: payments found, showing confirm screen") 872 if err := ui.ChatStellarShowConfirm(ctx); err != nil { 873 return res, err 874 } 875 defer func() { 876 _ = ui.ChatStellarDone(ctx, err != nil) 877 }() 878 uiSummary, toSend, err := h.G().StellarSender.DescribePayments(ctx, uid, convID, parsedPayments) 879 if err != nil { 880 if err := libkb.ExportErrorAsStatus(h.G().GlobalContext, err); err != nil { 881 _, _ = ui.ChatStellarDataError(ctx, *err) 882 } else { 883 h.Debug(ctx, "error exported to nothing") // should never happen 884 } 885 return res, err 886 } 887 h.Debug(ctx, "runStellarSendUI: payments described, telling UI") 888 accepted, err := ui.ChatStellarDataConfirm(ctx, uiSummary) 889 if err != nil { 890 return res, err 891 } 892 if !accepted { 893 return res, errors.New("Payment message declined") 894 } 895 h.Debug(ctx, "runStellarSendUI: message confirmed, sending payments") 896 payments, err := h.G().StellarSender.SendPayments(ctx, convID, toSend) 897 if err != nil { 898 // Send regardless here 899 h.Debug(ctx, "runStellarSendUI: failed to send payments, but continuing on: %s", err) 900 return msgBody, nil 901 } 902 newBody := msgBody.Text().DeepCopy() 903 newBody.Body = body 904 newBody.Payments = payments 905 return chat1.NewMessageBodyWithText(newBody), nil 906 } 907 908 var quickReactionPattern = regexp.MustCompile(`(?:^\+:)([^\s]+)(?::)$`) 909 910 func (h *Server) isQuickReaction(ctx context.Context, uid gregor1.UID, convID chat1.ConversationID, 911 body string) (reaction string, msgID chat1.MessageID, ok bool) { 912 body = strings.TrimSpace(body) 913 if !(strings.HasPrefix(body, "+:") && strings.HasSuffix(body, ":")) { 914 return "", 0, false 915 } 916 hits := quickReactionPattern.FindStringSubmatch(body) 917 if len(hits) < 2 { 918 return "", 0, false 919 } 920 tryStock := func() (string, bool) { 921 if h.G().EmojiSource.IsStockEmoji(hits[1]) { 922 return ":" + hits[1] + ":", true 923 } 924 return "", false 925 } 926 tryCustom := func() (string, bool) { 927 emojis, err := h.G().EmojiSource.Harvest(ctx, body, uid, convID, types.EmojiHarvestModeFast) 928 if err != nil { 929 h.Debug(ctx, "isQuickReaction: failed to harvest: %s", err) 930 return "", false 931 } 932 if len(emojis) != 1 { 933 return "", false 934 } 935 return ":" + emojis[0].Alias + ":", true 936 } 937 var hit *string 938 if reaction, ok = tryStock(); ok { 939 hit = new(string) 940 *hit = reaction 941 } 942 if hit == nil { 943 if reaction, ok = tryCustom(); ok { 944 hit = new(string) 945 *hit = reaction 946 } 947 } 948 if hit == nil { 949 return "", 0, false 950 } 951 conv, err := utils.GetUnverifiedConv(ctx, h.G(), uid, convID, types.InboxSourceDataSourceLocalOnly) 952 if err != nil { 953 h.Debug(ctx, "isQuickReaction: failed to get conv: %s", err) 954 return "", 0, false 955 } 956 return *hit, conv.MaxVisibleMsgID(), true 957 } 958 959 func (h *Server) PostTextNonblock(ctx context.Context, arg chat1.PostTextNonblockArg) (res chat1.PostLocalNonblockRes, err error) { 960 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, nil, h.identNotifier) 961 defer h.Trace(ctx, &err, "PostTextNonblock")() 962 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 963 if err != nil { 964 return res, err 965 } 966 reaction, msgID, ok := h.isQuickReaction(ctx, uid, arg.ConversationID, arg.Body) 967 if ok { 968 h.Debug(ctx, "PostTextNonblock: detected quick reaction") 969 return h.PostReactionNonblock(ctx, chat1.PostReactionNonblockArg{ 970 ConversationID: arg.ConversationID, 971 TlfName: arg.TlfName, 972 TlfPublic: arg.TlfPublic, 973 Supersedes: msgID, 974 Body: reaction, 975 OutboxID: arg.OutboxID, 976 ClientPrev: arg.ClientPrev, 977 IdentifyBehavior: arg.IdentifyBehavior, 978 }) 979 } 980 981 var parg chat1.PostLocalNonblockArg 982 parg.SessionID = arg.SessionID 983 parg.ClientPrev = arg.ClientPrev 984 parg.ConversationID = arg.ConversationID 985 parg.IdentifyBehavior = arg.IdentifyBehavior 986 parg.OutboxID = arg.OutboxID 987 parg.ReplyTo = arg.ReplyTo 988 parg.Msg.ClientHeader.MessageType = chat1.MessageType_TEXT 989 parg.Msg.ClientHeader.TlfName = arg.TlfName 990 parg.Msg.ClientHeader.TlfPublic = arg.TlfPublic 991 parg.Msg.MessageBody = chat1.NewMessageBodyWithText(chat1.MessageText{ 992 Body: arg.Body, 993 }) 994 if arg.EphemeralLifetime != nil { 995 parg.Msg.ClientHeader.EphemeralMetadata = &chat1.MsgEphemeralMetadata{ 996 Lifetime: *arg.EphemeralLifetime, 997 } 998 } 999 return h.PostLocalNonblock(ctx, parg) 1000 } 1001 1002 func (h *Server) PostReactionNonblock(ctx context.Context, arg chat1.PostReactionNonblockArg) (res chat1.PostLocalNonblockRes, err error) { 1003 var parg chat1.PostLocalNonblockArg 1004 parg.ClientPrev = arg.ClientPrev 1005 parg.ConversationID = arg.ConversationID 1006 parg.IdentifyBehavior = arg.IdentifyBehavior 1007 parg.OutboxID = arg.OutboxID 1008 parg.Msg.ClientHeader.MessageType = chat1.MessageType_REACTION 1009 parg.Msg.ClientHeader.Supersedes = arg.Supersedes 1010 parg.Msg.ClientHeader.TlfName = arg.TlfName 1011 parg.Msg.ClientHeader.TlfPublic = arg.TlfPublic 1012 parg.Msg.MessageBody = chat1.NewMessageBodyWithReaction(chat1.MessageReaction{ 1013 MessageID: arg.Supersedes, 1014 Body: arg.Body, 1015 }) 1016 1017 return h.PostLocalNonblock(ctx, parg) 1018 } 1019 1020 func (h *Server) PostHeadlineNonblock(ctx context.Context, arg chat1.PostHeadlineNonblockArg) (chat1.PostLocalNonblockRes, error) { 1021 var parg chat1.PostLocalNonblockArg 1022 parg.ClientPrev = arg.ClientPrev 1023 parg.ConversationID = arg.ConversationID 1024 parg.IdentifyBehavior = arg.IdentifyBehavior 1025 parg.OutboxID = arg.OutboxID 1026 parg.Msg.ClientHeader.MessageType = chat1.MessageType_HEADLINE 1027 parg.Msg.ClientHeader.TlfName = arg.TlfName 1028 parg.Msg.ClientHeader.TlfPublic = arg.TlfPublic 1029 parg.Msg.MessageBody = chat1.NewMessageBodyWithHeadline(chat1.MessageHeadline{ 1030 Headline: arg.Headline, 1031 }) 1032 1033 return h.PostLocalNonblock(ctx, parg) 1034 } 1035 1036 func (h *Server) PostHeadline(ctx context.Context, arg chat1.PostHeadlineArg) (chat1.PostLocalRes, error) { 1037 var parg chat1.PostLocalArg 1038 parg.ConversationID = arg.ConversationID 1039 parg.IdentifyBehavior = arg.IdentifyBehavior 1040 parg.Msg.ClientHeader.MessageType = chat1.MessageType_HEADLINE 1041 parg.Msg.ClientHeader.TlfName = arg.TlfName 1042 parg.Msg.ClientHeader.TlfPublic = arg.TlfPublic 1043 parg.Msg.MessageBody = chat1.NewMessageBodyWithHeadline(chat1.MessageHeadline{ 1044 Headline: arg.Headline, 1045 }) 1046 1047 return h.PostLocal(ctx, parg) 1048 } 1049 1050 func (h *Server) PostMetadataNonblock(ctx context.Context, arg chat1.PostMetadataNonblockArg) (chat1.PostLocalNonblockRes, error) { 1051 var parg chat1.PostLocalNonblockArg 1052 parg.ClientPrev = arg.ClientPrev 1053 parg.ConversationID = arg.ConversationID 1054 parg.IdentifyBehavior = arg.IdentifyBehavior 1055 parg.OutboxID = arg.OutboxID 1056 parg.Msg.ClientHeader.MessageType = chat1.MessageType_METADATA 1057 parg.Msg.ClientHeader.TlfName = arg.TlfName 1058 parg.Msg.ClientHeader.TlfPublic = arg.TlfPublic 1059 parg.Msg.MessageBody = chat1.NewMessageBodyWithMetadata(chat1.MessageConversationMetadata{ 1060 ConversationTitle: arg.ChannelName, 1061 }) 1062 1063 return h.PostLocalNonblock(ctx, parg) 1064 } 1065 1066 func (h *Server) PostMetadata(ctx context.Context, arg chat1.PostMetadataArg) (chat1.PostLocalRes, error) { 1067 var parg chat1.PostLocalArg 1068 parg.ConversationID = arg.ConversationID 1069 parg.IdentifyBehavior = arg.IdentifyBehavior 1070 parg.Msg.ClientHeader.MessageType = chat1.MessageType_METADATA 1071 parg.Msg.ClientHeader.TlfName = arg.TlfName 1072 parg.Msg.ClientHeader.TlfPublic = arg.TlfPublic 1073 parg.Msg.MessageBody = chat1.NewMessageBodyWithMetadata(chat1.MessageConversationMetadata{ 1074 ConversationTitle: arg.ChannelName, 1075 }) 1076 1077 return h.PostLocal(ctx, parg) 1078 } 1079 1080 func (h *Server) PostDeleteHistoryUpto(ctx context.Context, arg chat1.PostDeleteHistoryUptoArg) (res chat1.PostLocalRes, err error) { 1081 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, nil, h.identNotifier) 1082 defer h.Trace(ctx, &err, "PostDeleteHistoryUpto")() 1083 1084 delh := chat1.MessageDeleteHistory{Upto: arg.Upto} 1085 1086 var parg chat1.PostLocalArg 1087 parg.ConversationID = arg.ConversationID 1088 parg.IdentifyBehavior = arg.IdentifyBehavior 1089 parg.Msg.ClientHeader.MessageType = chat1.MessageType_DELETEHISTORY 1090 parg.Msg.ClientHeader.TlfName = arg.TlfName 1091 parg.Msg.ClientHeader.TlfPublic = arg.TlfPublic 1092 parg.Msg.ClientHeader.DeleteHistory = &delh 1093 parg.Msg.MessageBody = chat1.NewMessageBodyWithDeletehistory(delh) 1094 1095 h.Debug(ctx, "PostDeleteHistoryUpto: deleting upto msgid:%v", delh.Upto) 1096 1097 return h.PostLocal(ctx, parg) 1098 } 1099 1100 func (h *Server) PostDeleteHistoryThrough(ctx context.Context, arg chat1.PostDeleteHistoryThroughArg) (res chat1.PostLocalRes, err error) { 1101 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, nil, h.identNotifier) 1102 defer h.Trace(ctx, &err, "PostDeleteHistoryThrough")() 1103 return h.PostDeleteHistoryUpto(ctx, chat1.PostDeleteHistoryUptoArg{ 1104 ConversationID: arg.ConversationID, 1105 TlfName: arg.TlfName, 1106 TlfPublic: arg.TlfPublic, 1107 IdentifyBehavior: arg.IdentifyBehavior, 1108 Upto: arg.Through + 1, 1109 }) 1110 } 1111 1112 func (h *Server) PostDeleteHistoryByAge(ctx context.Context, arg chat1.PostDeleteHistoryByAgeArg) (res chat1.PostLocalRes, err error) { 1113 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, nil, h.identNotifier) 1114 defer h.Trace(ctx, &err, "PostDeleteHistoryByAge")() 1115 1116 gmRes, err := h.remoteClient().GetMessageBefore(ctx, chat1.GetMessageBeforeArg{ 1117 ConvID: arg.ConversationID, 1118 Age: arg.Age, 1119 }) 1120 if err != nil { 1121 return res, err 1122 } 1123 upto := gmRes.MsgID + 1 1124 h.Debug(ctx, "PostDeleteHistoryByAge: deleting upto msgid:%v (age:%v)", upto, arg.Age) 1125 return h.PostDeleteHistoryUpto(ctx, chat1.PostDeleteHistoryUptoArg{ 1126 ConversationID: arg.ConversationID, 1127 TlfName: arg.TlfName, 1128 TlfPublic: arg.TlfPublic, 1129 IdentifyBehavior: arg.IdentifyBehavior, 1130 Upto: upto, 1131 }) 1132 } 1133 1134 func (h *Server) GenerateOutboxID(ctx context.Context) (res chat1.OutboxID, err error) { 1135 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_SKIP, nil, h.identNotifier) 1136 defer h.Trace(ctx, &err, "GenerateOutboxID")() 1137 return storage.NewOutboxID() 1138 } 1139 1140 func (h *Server) PostLocalNonblock(ctx context.Context, arg chat1.PostLocalNonblockArg) (res chat1.PostLocalNonblockRes, err error) { 1141 var identBreaks []keybase1.TLFIdentifyFailure 1142 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, h.identNotifier) 1143 defer h.Trace(ctx, &err, "PostLocalNonblock")() 1144 defer h.suspendBgConvLoads(ctx)() 1145 defer func() { h.setResultRateLimit(ctx, &res) }() 1146 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 1147 if err != nil { 1148 return res, err 1149 } 1150 1151 // Clear draft 1152 if err := h.G().InboxSource.Draft(ctx, uid, arg.ConversationID, nil); err != nil { 1153 h.Debug(ctx, "PostLocalNonblock: failed to clear draft: %s", err) 1154 } 1155 1156 // Check for any slash command hits for an execute 1157 if handled, err := h.G().CommandsSource.AttemptBuiltinCommand(ctx, uid, arg.ConversationID, 1158 arg.Msg.ClientHeader.TlfName, arg.Msg.MessageBody, arg.ReplyTo); handled { 1159 h.Debug(ctx, "PostLocalNonblock: handled slash command with error: %s", err) 1160 return res, nil 1161 } 1162 1163 if !arg.SkipInChatPayments { 1164 // Determine if the messages contains any Stellar payments, and execute 1165 // them if so 1166 if arg.Msg.MessageBody, err = h.runStellarSendUI(ctx, arg.SessionID, uid, arg.ConversationID, 1167 arg.Msg.MessageBody, arg.ReplyTo); err != nil { 1168 return res, err 1169 } 1170 } 1171 1172 // Create non block sender 1173 var prepareOpts chat1.SenderPrepareOptions 1174 sender := NewBlockingSender(h.G(), h.boxer, h.remoteClient) 1175 nonblockSender := NewNonblockingSender(h.G(), sender) 1176 prepareOpts.ReplyTo = arg.ReplyTo 1177 if arg.Msg.ClientHeader.Conv.TopicType == chat1.TopicType_NONE { 1178 arg.Msg.ClientHeader.Conv.TopicType = chat1.TopicType_CHAT 1179 } 1180 obid, _, err := nonblockSender.Send(ctx, arg.ConversationID, arg.Msg, arg.ClientPrev, arg.OutboxID, 1181 nil, &prepareOpts) 1182 if err != nil { 1183 return res, fmt.Errorf("PostLocalNonblock: unable to send message: err: %s", err.Error()) 1184 } 1185 h.Debug(ctx, "PostLocalNonblock: using outboxID: %s", obid) 1186 1187 return chat1.PostLocalNonblockRes{ 1188 OutboxID: obid, 1189 IdentifyFailures: identBreaks, 1190 }, nil 1191 } 1192 1193 // MakePreview implements chat1.LocalInterface.MakePreview. 1194 func (h *Server) MakePreview(ctx context.Context, arg chat1.MakePreviewArg) (res chat1.MakePreviewRes, err error) { 1195 defer h.Trace(ctx, &err, "MakePreview")() 1196 return attachments.NewSender(h.G()).MakePreview(ctx, arg.Filename, arg.OutboxID) 1197 } 1198 1199 func (h *Server) MakeAudioPreview(ctx context.Context, arg chat1.MakeAudioPreviewArg) (res chat1.MakePreviewRes, err error) { 1200 defer h.Trace(ctx, &err, "MakeAudioPreview")() 1201 return attachments.NewSender(h.G()).MakeAudioPreview(ctx, arg.Amps, arg.Duration) 1202 } 1203 1204 func (h *Server) GetUploadTempFile(ctx context.Context, arg chat1.GetUploadTempFileArg) (res string, err error) { 1205 defer h.Trace(ctx, &err, "GetUploadTempFile")() 1206 return h.G().AttachmentUploader.GetUploadTempFile(ctx, arg.OutboxID, arg.Filename) 1207 } 1208 1209 func (h *Server) MakeUploadTempFile(ctx context.Context, arg chat1.MakeUploadTempFileArg) (res string, err error) { 1210 defer h.Trace(ctx, &err, "MakeUploadTempFile")() 1211 if res, err = h.G().AttachmentUploader.GetUploadTempFile(ctx, arg.OutboxID, arg.Filename); err != nil { 1212 return res, err 1213 } 1214 return res, os.WriteFile(res, arg.Data, 0644) 1215 } 1216 1217 func (h *Server) CancelUploadTempFile(ctx context.Context, outboxID chat1.OutboxID) (err error) { 1218 defer h.Trace(ctx, &err, "CancelUploadTempFile: %s", outboxID)() 1219 return h.G().AttachmentUploader.CancelUploadTempFile(ctx, outboxID) 1220 } 1221 1222 func (h *Server) PostFileAttachmentLocalNonblock(ctx context.Context, 1223 arg chat1.PostFileAttachmentLocalNonblockArg) (res chat1.PostLocalNonblockRes, err error) { 1224 var identBreaks []keybase1.TLFIdentifyFailure 1225 ctx = globals.ChatCtx(ctx, h.G(), arg.Arg.IdentifyBehavior, &identBreaks, h.identNotifier) 1226 defer h.Trace(ctx, &err, "PostFileAttachmentLocalNonblock")() 1227 defer h.suspendBgConvLoads(ctx)() 1228 defer func() { h.setResultRateLimit(ctx, &res) }() 1229 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 1230 if err != nil { 1231 return res, err 1232 } 1233 1234 // Create non block sender 1235 sender := NewNonblockingSender(h.G(), NewBlockingSender(h.G(), h.boxer, h.remoteClient)) 1236 outboxID, _, err := attachments.NewSender(h.G()).PostFileAttachmentMessage(ctx, sender, 1237 arg.Arg.ConversationID, arg.Arg.TlfName, arg.Arg.Visibility, arg.Arg.OutboxID, arg.Arg.Filename, 1238 arg.Arg.Title, arg.Arg.Metadata, arg.ClientPrev, arg.Arg.EphemeralLifetime, 1239 arg.Arg.CallerPreview) 1240 if err != nil { 1241 return res, err 1242 } 1243 _, err = h.G().AttachmentUploader.Register(ctx, uid, arg.Arg.ConversationID, outboxID, arg.Arg.Title, 1244 arg.Arg.Filename, nil, arg.Arg.CallerPreview) 1245 if err != nil { 1246 return res, err 1247 } 1248 return chat1.PostLocalNonblockRes{ 1249 OutboxID: outboxID, 1250 IdentifyFailures: identBreaks, 1251 }, nil 1252 } 1253 1254 // PostFileAttachmentLocal implements chat1.LocalInterface.PostFileAttachmentLocal. 1255 func (h *Server) PostFileAttachmentLocal(ctx context.Context, arg chat1.PostFileAttachmentLocalArg) (res chat1.PostLocalRes, err error) { 1256 var identBreaks []keybase1.TLFIdentifyFailure 1257 ctx = globals.ChatCtx(ctx, h.G(), arg.Arg.IdentifyBehavior, &identBreaks, h.identNotifier) 1258 defer h.Trace(ctx, &err, "PostFileAttachmentLocal")() 1259 defer h.suspendBgConvLoads(ctx)() 1260 defer func() { h.setResultRateLimit(ctx, &res) }() 1261 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 1262 if err != nil { 1263 return res, err 1264 } 1265 1266 // Get base of message we are going to send 1267 sender := NewBlockingSender(h.G(), h.boxer, h.remoteClient) 1268 _, msgID, err := attachments.NewSender(h.G()).PostFileAttachment(ctx, sender, uid, arg.Arg.ConversationID, 1269 arg.Arg.TlfName, arg.Arg.Visibility, arg.Arg.OutboxID, arg.Arg.Filename, arg.Arg.Title, 1270 arg.Arg.Metadata, 0, arg.Arg.EphemeralLifetime, arg.Arg.CallerPreview) 1271 if err != nil { 1272 return res, err 1273 } 1274 if msgID == nil { 1275 return res, errors.New("no message ID returned from post") 1276 } 1277 return chat1.PostLocalRes{ 1278 MessageID: *msgID, 1279 IdentifyFailures: identBreaks, 1280 }, nil 1281 } 1282 1283 // DownloadAttachmentLocal implements chat1.LocalInterface.DownloadAttachmentLocal. 1284 func (h *Server) DownloadAttachmentLocal(ctx context.Context, arg chat1.DownloadAttachmentLocalArg) (res chat1.DownloadAttachmentLocalRes, err error) { 1285 var identBreaks []keybase1.TLFIdentifyFailure 1286 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, h.identNotifier) 1287 defer h.Trace(ctx, &err, "DownloadAttachmentLocal")() 1288 defer func() { h.setResultRateLimit(ctx, &res) }() 1289 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 1290 if err != nil { 1291 return res, err 1292 } 1293 if arg.MessageID == 0 { 1294 return res, errors.New("invalid message ID provided") 1295 } 1296 darg := downloadAttachmentArg{ 1297 SessionID: arg.SessionID, 1298 ConversationID: arg.ConversationID, 1299 MessageID: arg.MessageID, 1300 Preview: arg.Preview, 1301 IdentifyBehavior: arg.IdentifyBehavior, 1302 } 1303 cli := h.getStreamUICli() 1304 darg.Sink = libkb.NewRemoteStreamBuffered(arg.Sink, cli, arg.SessionID) 1305 1306 return h.downloadAttachmentLocal(ctx, uid, darg) 1307 } 1308 1309 func (h *Server) getFileAttachmentDownloadDirs() (cacheDir, downloadDir string) { 1310 h.fileAttachmentDownloadConfigurationMu.RLock() 1311 defer h.fileAttachmentDownloadConfigurationMu.RUnlock() 1312 return h.fileAttachmentDownloadCacheDir, h.fileAttachmentDownloadDownloadDir 1313 } 1314 1315 func (h *Server) ConfigureFileAttachmentDownloadLocal(ctx context.Context, arg chat1.ConfigureFileAttachmentDownloadLocalArg) (err error) { 1316 h.fileAttachmentDownloadConfigurationMu.Lock() 1317 defer h.fileAttachmentDownloadConfigurationMu.Unlock() 1318 h.fileAttachmentDownloadCacheDir, h.fileAttachmentDownloadDownloadDir = arg.CacheDirOverride, arg.DownloadDirOverride 1319 return nil 1320 } 1321 1322 // DownloadFileAttachmentLocal implements chat1.LocalInterface.DownloadFileAttachmentLocal. 1323 func (h *Server) DownloadFileAttachmentLocal(ctx context.Context, arg chat1.DownloadFileAttachmentLocalArg) (res chat1.DownloadFileAttachmentLocalRes, err error) { 1324 var identBreaks []keybase1.TLFIdentifyFailure 1325 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, h.identNotifier) 1326 defer h.Trace(ctx, &err, "DownloadFileAttachmentLocal")() 1327 defer func() { h.setResultRateLimit(ctx, &res) }() 1328 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 1329 if err != nil { 1330 return res, err 1331 } 1332 if arg.MessageID == 0 { 1333 return res, errors.New("invalid message ID provided") 1334 } 1335 darg := downloadAttachmentArg{ 1336 SessionID: arg.SessionID, 1337 ConversationID: arg.ConversationID, 1338 MessageID: arg.MessageID, 1339 Preview: arg.Preview, 1340 IdentifyBehavior: arg.IdentifyBehavior, 1341 } 1342 cacheDir, downloadDir := h.getFileAttachmentDownloadDirs() 1343 downloadParentdir, useArbitraryName := downloadDir, false 1344 if arg.DownloadToCache { 1345 downloadParentdir, useArbitraryName = cacheDir, true 1346 } 1347 filePath, sink, err := attachments.SinkFromFilename(ctx, h.G(), uid, 1348 arg.ConversationID, arg.MessageID, downloadParentdir, useArbitraryName) 1349 if err != nil { 1350 return res, err 1351 } 1352 defer func() { 1353 // In the event of any error delete the file if it's empty. 1354 if err != nil { 1355 h.Debug(ctx, "DownloadFileAttachmentLocal: deleteFileIfEmpty: %v", deleteFileIfEmpty(filePath)) 1356 } 1357 }() 1358 if err := attachments.Quarantine(ctx, filePath); err != nil { 1359 h.Debug(ctx, "DownloadFileAttachmentLocal: failed to quarantine download: %s", err) 1360 return res, err 1361 } 1362 darg.Sink = sink 1363 ires, err := h.downloadAttachmentLocal(ctx, uid, darg) 1364 if err != nil { 1365 return res, err 1366 } 1367 return chat1.DownloadFileAttachmentLocalRes{ 1368 FilePath: filePath, 1369 IdentifyFailures: ires.IdentifyFailures, 1370 }, nil 1371 } 1372 1373 func deleteFileIfEmpty(filename string) (err error) { 1374 f, err := os.Stat(filename) 1375 if err != nil { 1376 return err 1377 } 1378 if f.Size() == 0 { 1379 return os.Remove(filename) 1380 } 1381 return nil 1382 } 1383 1384 type downloadAttachmentArg struct { 1385 SessionID int 1386 ConversationID chat1.ConversationID 1387 MessageID chat1.MessageID 1388 Sink io.WriteCloser 1389 Preview bool 1390 IdentifyBehavior keybase1.TLFIdentifyBehavior 1391 } 1392 1393 func (h *Server) downloadAttachmentLocal(ctx context.Context, uid gregor1.UID, arg downloadAttachmentArg) (res chat1.DownloadAttachmentLocalRes, err error) { 1394 1395 var identBreaks []keybase1.TLFIdentifyFailure 1396 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, h.identNotifier) 1397 progress := func(bytesComplete, bytesTotal int64) { 1398 h.G().NotifyRouter.HandleChatAttachmentDownloadProgress(ctx, keybase1.UID(uid.String()), 1399 arg.ConversationID, arg.MessageID, bytesComplete, bytesTotal) 1400 } 1401 1402 h.Debug(ctx, "downloadAttachmentLocal: fetching asset from attachment message: convID: %s messageID: %d", 1403 arg.ConversationID, arg.MessageID) 1404 1405 err = attachments.Download(ctx, h.G(), uid, arg.ConversationID, 1406 arg.MessageID, arg.Sink, arg.Preview, progress, h.remoteClient) 1407 if err != nil { 1408 return res, err 1409 } 1410 h.G().NotifyRouter.HandleChatAttachmentDownloadComplete(ctx, keybase1.UID(uid.String()), 1411 arg.ConversationID, arg.MessageID) 1412 1413 return chat1.DownloadAttachmentLocalRes{ 1414 IdentifyFailures: identBreaks, 1415 }, nil 1416 } 1417 1418 func (h *Server) presentUIItem(ctx context.Context, uid gregor1.UID, conv *chat1.ConversationLocal, 1419 partMode utils.PresentParticipantsMode) (res *chat1.InboxUIItem) { 1420 if conv != nil { 1421 pc := utils.PresentConversationLocal(ctx, h.G(), uid, *conv, partMode) 1422 res = &pc 1423 } 1424 return res 1425 } 1426 1427 func (h *Server) CancelPost(ctx context.Context, outboxID chat1.OutboxID) (err error) { 1428 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_SKIP, nil, h.identNotifier) 1429 defer h.Trace(ctx, &err, "CancelPost(%s)", outboxID)() 1430 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 1431 if err != nil { 1432 return err 1433 } 1434 outbox := storage.NewOutbox(h.G(), uid) 1435 obr, err := outbox.RemoveMessage(ctx, outboxID) 1436 if err != nil { 1437 return err 1438 } 1439 // Alert the attachment uploader as well, in case this outboxID corresponds to an attachment upload 1440 if err := h.G().AttachmentUploader.Cancel(ctx, outboxID); err != nil { 1441 return err 1442 } 1443 convLocal, err := h.G().InboxSource.IncrementLocalConvVersion(ctx, uid, obr.ConvID) 1444 if err != nil { 1445 h.Debug(ctx, "CancelPost: failed to get IncrementLocalConvVersion") 1446 } 1447 act := chat1.NewChatActivityWithFailedMessage(chat1.FailedMessageInfo{ 1448 OutboxRecords: []chat1.OutboxRecord{obr}, 1449 Conv: h.presentUIItem(ctx, uid, convLocal, utils.PresentParticipantsModeSkip), 1450 }) 1451 h.G().ActivityNotifier.Activity(context.Background(), uid, chat1.TopicType_NONE, &act, 1452 chat1.ChatActivitySource_LOCAL) 1453 return h.G().Badger.Send(ctx) 1454 } 1455 1456 func (h *Server) RetryPost(ctx context.Context, arg chat1.RetryPostArg) (err error) { 1457 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_SKIP, nil, h.identNotifier) 1458 defer h.Trace(ctx, &err, fmt.Sprintf("RetryPost: obr: %v", arg.OutboxID))() 1459 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 1460 if err != nil { 1461 return err 1462 } 1463 1464 // Mark as retry in the outbox 1465 outbox := storage.NewOutbox(h.G(), uid) 1466 obr, err := outbox.RetryMessage(ctx, arg.OutboxID, arg.IdentifyBehavior) 1467 if err != nil { 1468 return err 1469 } else if obr == nil { 1470 return nil 1471 } 1472 if obr.IsAttachment() { 1473 if _, err := h.G().AttachmentUploader.Retry(ctx, obr.OutboxID); err != nil { 1474 h.Debug(ctx, "RetryPost: failed to retry attachment upload: %s", err) 1475 } 1476 } 1477 1478 // Force the send loop to try again 1479 h.G().MessageDeliverer.ForceDeliverLoop(ctx) 1480 1481 return nil 1482 } 1483 1484 // remoteClient returns a client connection to gregord. 1485 func (h *Server) remoteClient() chat1.RemoteInterface { 1486 if h.rc != nil { 1487 return h.rc 1488 } 1489 return h.serverConn.GetClient() 1490 } 1491 1492 func (h *Server) setTestRemoteClient(ri chat1.RemoteInterface) { 1493 h.rc = ri 1494 } 1495 1496 func (h *Server) FindGeneralConvFromTeamID(ctx context.Context, teamID keybase1.TeamID) (res chat1.InboxUIItem, err error) { 1497 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, nil) 1498 defer h.Trace(ctx, &err, "FindGeneralConvFromTeamID")() 1499 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 1500 if err != nil { 1501 return res, err 1502 } 1503 tlfID, err := chat1.TeamIDToTLFID(teamID) 1504 if err != nil { 1505 return res, err 1506 } 1507 vis := keybase1.TLFVisibility_PRIVATE 1508 topicName := globals.DefaultTeamTopic 1509 topicType := chat1.TopicType_CHAT 1510 query := &chat1.GetInboxLocalQuery{ 1511 Name: &chat1.NameQuery{ 1512 TlfID: &tlfID, 1513 MembersType: chat1.ConversationMembersType_TEAM, 1514 }, 1515 TlfVisibility: &vis, 1516 TopicName: &topicName, 1517 TopicType: &topicType, 1518 } 1519 ib, _, err := h.G().InboxSource.Read(ctx, uid, types.ConversationLocalizerBlocking, 1520 types.InboxSourceDataSourceAll, nil, query) 1521 if err != nil { 1522 return res, err 1523 } 1524 if len(ib.Convs) != 1 { 1525 return res, libkb.NotFoundError{} 1526 } 1527 return utils.PresentConversationLocal(ctx, h.G(), uid, ib.Convs[0], utils.PresentParticipantsModeSkip), 1528 nil 1529 } 1530 1531 func (h *Server) FindConversationsLocal(ctx context.Context, 1532 arg chat1.FindConversationsLocalArg) (res chat1.FindConversationsLocalRes, err error) { 1533 var identBreaks []keybase1.TLFIdentifyFailure 1534 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, h.identNotifier) 1535 defer h.Trace(ctx, &err, "FindConversationsLocal")() 1536 defer func() { h.setResultRateLimit(ctx, &res) }() 1537 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 1538 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 1539 if err != nil { 1540 return res, err 1541 } 1542 1543 res.Conversations, err = FindConversations(ctx, h.G(), h.DebugLabeler, 1544 types.InboxSourceDataSourceAll, h.remoteClient, 1545 uid, arg.TlfName, arg.TopicType, arg.MembersType, arg.Visibility, arg.TopicName, arg.OneChatPerTLF) 1546 if err != nil { 1547 return res, err 1548 } 1549 res.UiConversations = utils.PresentConversationLocals(ctx, h.G(), uid, res.Conversations, 1550 utils.PresentParticipantsModeInclude) 1551 return res, nil 1552 } 1553 1554 func (h *Server) UpdateUnsentText(ctx context.Context, arg chat1.UpdateUnsentTextArg) (err error) { 1555 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 1556 defer h.Trace(ctx, &err, 1557 fmt.Sprintf("UpdateUnsentText convID: %s", arg.ConversationID))() 1558 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 1559 if err != nil { 1560 h.Debug(ctx, "UpdateUnsentText: not logged in: %s", err) 1561 return nil 1562 } 1563 1564 // Save draft 1565 var draftText *string 1566 if len(arg.Text) > 0 { 1567 draftText = &arg.Text 1568 } 1569 if err := h.G().InboxSource.Draft(ctx, uid, arg.ConversationID, draftText); err != nil { 1570 h.Debug(ctx, "UpdateUnsentText: failed to save draft: %s", err) 1571 } 1572 1573 // Attempt to prefetch any unfurls in the background that are in the message text 1574 go h.G().Unfurler.Prefetch(globals.BackgroundChatCtx(ctx, h.G()), uid, arg.ConversationID, arg.Text) 1575 1576 // Preview any slash commands in the text 1577 go h.G().CommandsSource.PreviewBuiltinCommand(globals.BackgroundChatCtx(ctx, h.G()), uid, 1578 arg.ConversationID, arg.TlfName, arg.Text) 1579 1580 return nil 1581 } 1582 1583 func (h *Server) UpdateTyping(ctx context.Context, arg chat1.UpdateTypingArg) (err error) { 1584 var identBreaks []keybase1.TLFIdentifyFailure 1585 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, 1586 &identBreaks, h.identNotifier) 1587 defer h.Trace(ctx, &err, 1588 fmt.Sprintf("UpdateTyping convID: %s", arg.ConversationID))() 1589 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 1590 if err != nil { 1591 h.Debug(ctx, "UpdateTyping: not logged in: %s", err) 1592 return nil 1593 } 1594 // Just bail out if we are offline 1595 if !h.G().Syncer.IsConnected(ctx) { 1596 return nil 1597 } 1598 deviceID := make([]byte, libkb.DeviceIDLen) 1599 if err := h.G().Env.GetDeviceID().ToBytes(deviceID); err != nil { 1600 h.Debug(ctx, "UpdateTyping: failed to get device: %s", err) 1601 return nil 1602 } 1603 if err := h.remoteClient().UpdateTypingRemote(ctx, chat1.UpdateTypingRemoteArg{ 1604 Uid: uid, 1605 DeviceID: deviceID, 1606 ConvID: arg.ConversationID, 1607 Typing: arg.Typing, 1608 }); err != nil { 1609 h.Debug(ctx, "UpdateTyping: failed to hit the server: %s", err.Error()) 1610 } 1611 return nil 1612 } 1613 1614 func (h *Server) JoinConversationByIDLocal(ctx context.Context, convID chat1.ConversationID) (res chat1.JoinLeaveConversationLocalRes, err error) { 1615 var identBreaks []keybase1.TLFIdentifyFailure 1616 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, 1617 &identBreaks, h.identNotifier) 1618 defer h.Trace(ctx, &err, fmt.Sprintf("JoinConversationByIDLocal: convID: %s", convID))() 1619 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 1620 defer func() { h.setResultRateLimit(ctx, &res) }() 1621 defer func() { 1622 if res.Offline { 1623 h.Debug(ctx, "JoinConversationByIDLocal: result obtained offline") 1624 } 1625 }() 1626 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 1627 if err != nil { 1628 return res, err 1629 } 1630 err = JoinConversation(ctx, h.G(), h.DebugLabeler, h.remoteClient, uid, convID) 1631 if err != nil { 1632 return res, err 1633 } 1634 res.Offline = h.G().InboxSource.IsOffline(ctx) 1635 return res, nil 1636 } 1637 1638 func (h *Server) JoinConversationLocal(ctx context.Context, arg chat1.JoinConversationLocalArg) (res chat1.JoinLeaveConversationLocalRes, err error) { 1639 var identBreaks []keybase1.TLFIdentifyFailure 1640 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, 1641 &identBreaks, h.identNotifier) 1642 defer h.Trace(ctx, &err, fmt.Sprintf("JoinConversation(%s)", 1643 arg.TopicName))() 1644 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 1645 defer func() { h.setResultRateLimit(ctx, &res) }() 1646 defer func() { 1647 if res.Offline { 1648 h.Debug(ctx, "JoinConversationLocal: result obtained offline") 1649 } 1650 }() 1651 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 1652 if err != nil { 1653 return res, err 1654 } 1655 if err = JoinConversationByName(ctx, h.G(), h.DebugLabeler, h.remoteClient, uid, arg.TlfName, 1656 arg.TopicName, arg.TopicType, arg.Visibility); err != nil { 1657 return res, err 1658 } 1659 res.Offline = h.G().InboxSource.IsOffline(ctx) 1660 return res, nil 1661 } 1662 1663 func (h *Server) LeaveConversationLocal(ctx context.Context, convID chat1.ConversationID) (res chat1.JoinLeaveConversationLocalRes, err error) { 1664 var identBreaks []keybase1.TLFIdentifyFailure 1665 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, 1666 &identBreaks, h.identNotifier) 1667 defer h.Trace(ctx, &err, fmt.Sprintf("LeaveConversation(%s)", convID))() 1668 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 1669 defer func() { h.setResultRateLimit(ctx, &res) }() 1670 defer func() { 1671 if res.Offline { 1672 h.Debug(ctx, "LeaveConversationLocal: result obtained offline") 1673 } 1674 }() 1675 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 1676 if err != nil { 1677 return res, err 1678 } 1679 err = LeaveConversation(ctx, h.G(), h.DebugLabeler, h.remoteClient, uid, convID) 1680 if err != nil { 1681 return res, err 1682 } 1683 res.Offline = h.G().InboxSource.IsOffline(ctx) 1684 return res, nil 1685 } 1686 1687 func (h *Server) PreviewConversationByIDLocal(ctx context.Context, convID chat1.ConversationID) (res chat1.PreviewConversationLocalRes, err error) { 1688 var identBreaks []keybase1.TLFIdentifyFailure 1689 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, 1690 &identBreaks, h.identNotifier) 1691 defer h.Trace(ctx, &err, fmt.Sprintf("PreviewConversationByIDLocal: convID: %s", convID))() 1692 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 1693 defer func() { h.setResultRateLimit(ctx, &res) }() 1694 defer func() { 1695 if res.Offline { 1696 h.Debug(ctx, "PreviewConversationByIDLocal: result obtained offline") 1697 } 1698 }() 1699 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 1700 if err != nil { 1701 return res, err 1702 } 1703 conv, err := PreviewConversation(ctx, h.G(), h.DebugLabeler, h.remoteClient, uid, convID) 1704 if err != nil { 1705 return res, err 1706 } 1707 res.Conv = utils.PresentConversationLocal(ctx, h.G(), uid, conv, utils.PresentParticipantsModeInclude) 1708 res.Offline = h.G().InboxSource.IsOffline(ctx) 1709 return res, nil 1710 } 1711 1712 func (h *Server) DeleteConversationLocal(ctx context.Context, arg chat1.DeleteConversationLocalArg) (res chat1.DeleteConversationLocalRes, err error) { 1713 var identBreaks []keybase1.TLFIdentifyFailure 1714 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, 1715 &identBreaks, h.identNotifier) 1716 defer h.Trace(ctx, &err, fmt.Sprintf("DeleteConversation(%s)", arg.ConvID))() 1717 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 1718 defer func() { h.setResultRateLimit(ctx, &res) }() 1719 defer func() { 1720 if res.Offline { 1721 h.Debug(ctx, "DeleteConversationLocal: result obtained offline") 1722 } 1723 }() 1724 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 1725 if err != nil { 1726 return res, err 1727 } 1728 ui := h.getChatUI(arg.SessionID) 1729 confirmed := arg.Confirmed 1730 if !confirmed { 1731 confirmed, err = ui.ChatConfirmChannelDelete(ctx, chat1.ChatConfirmChannelDeleteArg{ 1732 SessionID: arg.SessionID, 1733 Channel: arg.ChannelName, 1734 }) 1735 if err != nil { 1736 return res, err 1737 } 1738 } 1739 if !confirmed { 1740 return res, errors.New("channel delete unconfirmed") 1741 } 1742 if err := h.G().InboxSource.RemoteDeleteConversation(ctx, uid, arg.ConvID); err != nil { 1743 return res, err 1744 } 1745 res.Offline = h.G().InboxSource.IsOffline(ctx) 1746 return res, nil 1747 } 1748 1749 func (h *Server) RemoveFromConversationLocal(ctx context.Context, arg chat1.RemoveFromConversationLocalArg) (res chat1.RemoveFromConversationLocalRes, err error) { 1750 var identBreaks []keybase1.TLFIdentifyFailure 1751 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, 1752 &identBreaks, h.identNotifier) 1753 defer h.Trace(ctx, &err, fmt.Sprintf("RemoveFromConversation(%s)", arg.ConvID))() 1754 defer func() { h.setResultRateLimit(ctx, &res) }() 1755 if _, err = utils.AssertLoggedInUID(ctx, h.G()); err != nil { 1756 return res, err 1757 } 1758 err = RemoveFromConversation(ctx, h.G(), h.DebugLabeler, h.remoteClient, arg.ConvID, arg.Usernames) 1759 if err != nil { 1760 return res, err 1761 } 1762 return res, nil 1763 } 1764 1765 func (h *Server) GetTLFConversationsLocal(ctx context.Context, arg chat1.GetTLFConversationsLocalArg) (res chat1.GetTLFConversationsLocalRes, err error) { 1766 var identBreaks []keybase1.TLFIdentifyFailure 1767 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, 1768 &identBreaks, h.identNotifier) 1769 defer h.Trace(ctx, &err, fmt.Sprintf("GetTLFConversationsLocal(%s)", 1770 arg.TlfName))() 1771 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 1772 defer func() { h.setResultRateLimit(ctx, &res) }() 1773 defer func() { 1774 if res.Offline { 1775 h.Debug(ctx, "GetTLFConversationsLocal: result obtained offline") 1776 } 1777 }() 1778 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 1779 if err != nil { 1780 return res, err 1781 } 1782 1783 // Fetch the TLF ID from specified name 1784 nameInfo, err := CreateNameInfoSource(ctx, h.G(), arg.MembersType).LookupID(ctx, arg.TlfName, false) 1785 if err != nil { 1786 h.Debug(ctx, "GetTLFConversationsLocal: failed to get TLFID from name: %v", err) 1787 return res, err 1788 } 1789 1790 var convs []chat1.ConversationLocal 1791 convs, err = h.G().TeamChannelSource.GetChannelsFull(ctx, uid, nameInfo.ID, arg.TopicType) 1792 if err != nil { 1793 return res, err 1794 } 1795 res.Convs = utils.PresentConversationLocals(ctx, h.G(), uid, convs, utils.PresentParticipantsModeInclude) 1796 res.Offline = h.G().InboxSource.IsOffline(ctx) 1797 return res, nil 1798 } 1799 1800 func (h *Server) GetChannelMembershipsLocal(ctx context.Context, arg chat1.GetChannelMembershipsLocalArg) (res chat1.GetChannelMembershipsLocalRes, err error) { 1801 var identBreaks []keybase1.TLFIdentifyFailure 1802 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, 1803 &identBreaks, h.identNotifier) 1804 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 1805 defer func() { h.setResultRateLimit(ctx, &res) }() 1806 defer func() { 1807 if res.Offline { 1808 h.Debug(ctx, "GetTLFConversationsLocal: result obtained offline") 1809 } 1810 }() 1811 myUID, err := utils.AssertLoggedInUID(ctx, h.G()) 1812 if err != nil { 1813 return chat1.GetChannelMembershipsLocalRes{}, err 1814 } 1815 1816 chatTopicType := chat1.TopicType_CHAT 1817 tlfID := chat1.TLFID(arg.TeamID.ToBytes()) 1818 1819 // fetch all conversations in the supplied team 1820 inbox, err := h.G().InboxSource.ReadUnverified(ctx, myUID, types.InboxSourceDataSourceAll, 1821 &chat1.GetInboxQuery{ 1822 TlfID: &tlfID, 1823 MembersTypes: []chat1.ConversationMembersType{chat1.ConversationMembersType_TEAM}, 1824 TopicType: &chatTopicType, 1825 }) 1826 if err != nil { 1827 return res, err 1828 } 1829 1830 // find a list of conversations that the provided uid is a member of 1831 var memberConvs []types.RemoteConversation 1832 for _, conv := range inbox.ConvsUnverified { 1833 uids, err := h.G().ParticipantsSource.Get(ctx, myUID, conv.GetConvID(), 1834 types.InboxSourceDataSourceAll) 1835 if err != nil { 1836 return res, err 1837 } 1838 for _, uid := range uids { 1839 if bytes.Equal(uid, arg.Uid) { 1840 memberConvs = append(memberConvs, conv) 1841 break 1842 } 1843 } 1844 } 1845 1846 // localize those conversations so we can get the topic name 1847 convsLocal, _, err := h.G().InboxSource.Localize(ctx, myUID, memberConvs, 1848 types.ConversationLocalizerBlocking) 1849 for _, conv := range convsLocal { 1850 res.Channels = append(res.Channels, chat1.ChannelNameMention{ 1851 ConvID: conv.GetConvID(), 1852 TopicName: conv.GetTopicName(), 1853 }) 1854 } 1855 return res, nil 1856 } 1857 1858 func (h *Server) SetAppNotificationSettingsLocal(ctx context.Context, 1859 arg chat1.SetAppNotificationSettingsLocalArg) (res chat1.SetAppNotificationSettingsLocalRes, err error) { 1860 var identBreaks []keybase1.TLFIdentifyFailure 1861 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, 1862 &identBreaks, h.identNotifier) 1863 defer h.Trace(ctx, &err, fmt.Sprintf("SetAppNotificationSettings(%s)", 1864 arg.ConvID))() 1865 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 1866 defer func() { h.setResultRateLimit(ctx, &res) }() 1867 defer func() { 1868 if res.Offline { 1869 h.Debug(ctx, "SetAppNotificationSettingsLocal: result obtained offline") 1870 } 1871 }() 1872 if _, err = utils.AssertLoggedInUID(ctx, h.G()); err != nil { 1873 return res, err 1874 } 1875 1876 var nsettings chat1.ConversationNotificationInfo 1877 nsettings.ChannelWide = arg.ChannelWide 1878 nsettings.Settings = make(map[keybase1.DeviceType]map[chat1.NotificationKind]bool) 1879 nsettings.Settings[keybase1.DeviceType_MOBILE] = make(map[chat1.NotificationKind]bool) 1880 nsettings.Settings[keybase1.DeviceType_DESKTOP] = make(map[chat1.NotificationKind]bool) 1881 for _, setting := range arg.Settings { 1882 nsettings.Settings[setting.DeviceType][setting.Kind] = setting.Enabled 1883 } 1884 _, err = h.remoteClient().SetAppNotificationSettings(ctx, chat1.SetAppNotificationSettingsArg{ 1885 ConvID: arg.ConvID, 1886 Settings: nsettings, 1887 }) 1888 if err != nil { 1889 h.Debug(ctx, "SetAppNotificationSettings: failed to post to remote: %s", err.Error()) 1890 return res, err 1891 } 1892 res.Offline = h.G().InboxSource.IsOffline(ctx) 1893 return res, nil 1894 } 1895 1896 func (h *Server) UnboxMobilePushNotification(ctx context.Context, arg chat1.UnboxMobilePushNotificationArg) (res string, err error) { 1897 var identBreaks []keybase1.TLFIdentifyFailure 1898 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, &identBreaks, h.identNotifier) 1899 defer h.Trace(ctx, &err, fmt.Sprintf("UnboxMobilePushNotification(%s)", 1900 arg.ConvID))() 1901 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 1902 if err != nil { 1903 return "", err 1904 } 1905 bConvID, err := hex.DecodeString(arg.ConvID) 1906 if err != nil { 1907 h.Debug(ctx, "UnboxMobilePushNotification: invalid convID: %s msg: %s", arg.ConvID, err.Error()) 1908 return "", err 1909 } 1910 convID := chat1.ConversationID(bConvID) 1911 mp := NewMobilePush(h.G()) 1912 msg, err := mp.UnboxPushNotification(ctx, uid, convID, arg.MembersType, arg.Payload) 1913 if err != nil { 1914 return "", err 1915 } 1916 if _, err := utils.GetVerifiedConv(ctx, h.G(), uid, convID, types.InboxSourceDataSourceAll); err != nil { 1917 return "", err 1918 } 1919 if arg.ShouldAck { 1920 mp.AckNotificationSuccess(ctx, arg.PushIDs) 1921 } 1922 1923 if msg.IsValid() { 1924 body := msg.Valid().MessageBody 1925 bodyTyp, err := body.MessageType() 1926 if err != nil { 1927 return "", err 1928 } 1929 if bodyTyp == chat1.MessageType_TEXT { 1930 return body.Text().Body, nil 1931 } 1932 } 1933 return res, nil 1934 } 1935 1936 func (h *Server) SetGlobalAppNotificationSettingsLocal(ctx context.Context, 1937 strSettings map[string]bool) (err error) { 1938 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 1939 defer h.Trace(ctx, &err, "SetGlobalAppNotificationSettings")() 1940 if _, err = utils.AssertLoggedInUID(ctx, h.G()); err != nil { 1941 return err 1942 } 1943 return setGlobalAppNotificationSettings(ctx, h.G(), h.remoteClient, strSettings) 1944 } 1945 1946 func (h *Server) GetGlobalAppNotificationSettingsLocal(ctx context.Context) (res chat1.GlobalAppNotificationSettings, err error) { 1947 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 1948 defer h.Trace(ctx, &err, "GetGlobalAppNotificationSettings")() 1949 if _, err = utils.AssertLoggedInUID(ctx, h.G()); err != nil { 1950 return res, err 1951 } 1952 return getGlobalAppNotificationSettings(ctx, h.G(), h.remoteClient) 1953 } 1954 1955 func (h *Server) AddTeamMemberAfterReset(ctx context.Context, 1956 arg chat1.AddTeamMemberAfterResetArg) (err error) { 1957 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 1958 defer h.Trace(ctx, &err, "AddTeamMemberAfterReset")() 1959 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 1960 if err != nil { 1961 return err 1962 } 1963 1964 // Lookup conversation to get team ID 1965 iboxRes, _, err := h.G().InboxSource.Read(ctx, uid, types.ConversationLocalizerBlocking, 1966 types.InboxSourceDataSourceAll, nil, 1967 &chat1.GetInboxLocalQuery{ 1968 ConvIDs: []chat1.ConversationID{arg.ConvID}, 1969 }) 1970 if err != nil { 1971 return err 1972 } 1973 if len(iboxRes.Convs) != 1 { 1974 return errors.New("failed to find conversation to add reset user back into") 1975 } 1976 var teamID keybase1.TeamID 1977 conv := iboxRes.Convs[0] 1978 switch conv.Info.MembersType { 1979 case chat1.ConversationMembersType_IMPTEAMUPGRADE: 1980 team, err := NewTeamLoader(h.G().ExternalG()).loadTeam(ctx, conv.Info.Triple.Tlfid, conv.Info.TlfName, 1981 conv.Info.MembersType, conv.Info.Visibility == keybase1.TLFVisibility_PUBLIC, nil) 1982 if err != nil { 1983 return err 1984 } 1985 teamID = team.ID 1986 case chat1.ConversationMembersType_IMPTEAMNATIVE, chat1.ConversationMembersType_TEAM: 1987 teamID = keybase1.TeamID(conv.Info.Triple.Tlfid.String()) 1988 default: 1989 return fmt.Errorf("unable to add member back to non team conversation: %v", 1990 conv.Info.MembersType) 1991 } 1992 return teams.ReAddMemberAfterReset(ctx, h.G().ExternalG(), teamID, arg.Username) 1993 } 1994 1995 func (h *Server) GetAllResetConvMembers(ctx context.Context) (res chat1.GetAllResetConvMembersRes, err error) { 1996 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 1997 defer h.Trace(ctx, &err, "GetAllResetConvMembers")() 1998 defer func() { h.setResultRateLimit(ctx, &res) }() 1999 if _, err := utils.AssertLoggedInUID(ctx, h.G()); err != nil { 2000 return res, err 2001 } 2002 resetConvs, err := h.remoteClient().GetResetConversations(ctx) 2003 if err != nil { 2004 return res, err 2005 } 2006 for _, resetMember := range resetConvs.ResetConvs { 2007 username, err := h.G().GetUPAKLoader().LookupUsername(ctx, keybase1.UID(resetMember.Uid.String())) 2008 if err != nil { 2009 return res, err 2010 } 2011 res.Members = append(res.Members, chat1.ResetConvMember{ 2012 Uid: resetMember.Uid, 2013 Conv: resetMember.ConvID, 2014 Username: username.String(), 2015 }) 2016 } 2017 return res, nil 2018 } 2019 2020 func (h *Server) SetConvRetentionLocal(ctx context.Context, arg chat1.SetConvRetentionLocalArg) (err error) { 2021 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 2022 defer h.Trace(ctx, &err, "SetConvRetentionLocal(%v, %v)", arg.ConvID, 2023 arg.Policy.Summary())() 2024 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 2025 if err != nil { 2026 return err 2027 } 2028 // short circuit if the policy is unchanged. 2029 policy := arg.Policy 2030 conv, err := utils.GetVerifiedConv(ctx, h.G(), uid, arg.ConvID, types.InboxSourceDataSourceAll) 2031 if err != nil { 2032 return err 2033 } 2034 if convRetention := conv.ConvRetention; convRetention != nil && policy.Eq(*convRetention) { 2035 h.Debug(ctx, "retention policy unchanged, skipping update") 2036 return nil 2037 } 2038 2039 if _, err = h.remoteClient().SetConvRetention(ctx, chat1.SetConvRetentionArg{ 2040 ConvID: arg.ConvID, 2041 Policy: policy, 2042 }); err != nil { 2043 return err 2044 } 2045 2046 // Post a SYSTEM message to conversation about the change. If we're 2047 // inheriting the team policy, fetch that for the message. 2048 isInherit := false 2049 typ, err := policy.Typ() 2050 if err != nil { 2051 return err 2052 } 2053 switch typ { 2054 case chat1.RetentionPolicyType_INHERIT: 2055 isInherit = true 2056 default: 2057 // Nothing to do for other policy types. 2058 } 2059 2060 if isInherit { 2061 teamRetention := conv.TeamRetention 2062 if teamRetention == nil { 2063 policy = chat1.RetentionPolicy{} 2064 } else { 2065 policy = *teamRetention 2066 } 2067 } 2068 username := h.G().Env.GetUsername() 2069 subBody := chat1.NewMessageSystemWithChangeretention(chat1.MessageSystemChangeRetention{ 2070 User: username.String(), 2071 IsTeam: false, 2072 IsInherit: isInherit, 2073 MembersType: conv.GetMembersType(), 2074 Policy: policy, 2075 }) 2076 body := chat1.NewMessageBodyWithSystem(subBody) 2077 return h.G().ChatHelper.SendMsgByID(ctx, arg.ConvID, conv.Info.TlfName, body, chat1.MessageType_SYSTEM, 2078 conv.Info.Visibility) 2079 } 2080 2081 func (h *Server) SetTeamRetentionLocal(ctx context.Context, arg chat1.SetTeamRetentionLocalArg) (err error) { 2082 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 2083 defer h.Trace(ctx, &err, "SetTeamRetentionLocal(%v, %v)", arg.TeamID, arg.Policy.Summary())() 2084 if _, err = utils.AssertLoggedInUID(ctx, h.G()); err != nil { 2085 return err 2086 } 2087 teamRetention, err := h.GetTeamRetentionLocal(ctx, arg.TeamID) 2088 if err != nil { 2089 return err 2090 } 2091 policy := arg.Policy 2092 if teamRetention != nil && policy.Eq(*teamRetention) { 2093 h.Debug(ctx, "retention policy unchanged, skipping update") 2094 return nil 2095 } 2096 2097 if _, err = h.remoteClient().SetTeamRetention(ctx, chat1.SetTeamRetentionArg{ 2098 TeamID: arg.TeamID, 2099 Policy: policy, 2100 }); err != nil { 2101 return err 2102 } 2103 2104 // Post a SYSTEM message to the #general channel about the change. 2105 tlfID, err := chat1.MakeTLFID(arg.TeamID.String()) 2106 if err != nil { 2107 return err 2108 } 2109 username := h.G().Env.GetUsername() 2110 subBody := chat1.NewMessageSystemWithChangeretention(chat1.MessageSystemChangeRetention{ 2111 User: username.String(), 2112 IsTeam: true, 2113 IsInherit: false, 2114 MembersType: chat1.ConversationMembersType_TEAM, 2115 Policy: arg.Policy, 2116 }) 2117 body := chat1.NewMessageBodyWithSystem(subBody) 2118 info, err := CreateNameInfoSource(ctx, h.G(), chat1.ConversationMembersType_TEAM).LookupName(ctx, tlfID, 2119 false, "") 2120 if err != nil { 2121 return err 2122 } 2123 return h.G().ChatHelper.SendMsgByName(ctx, info.CanonicalName, &globals.DefaultTeamTopic, 2124 chat1.ConversationMembersType_TEAM, keybase1.TLFIdentifyBehavior_CHAT_CLI, body, 2125 chat1.MessageType_SYSTEM) 2126 } 2127 2128 func (h *Server) GetTeamRetentionLocal(ctx context.Context, teamID keybase1.TeamID) (res *chat1.RetentionPolicy, err error) { 2129 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 2130 defer h.Trace(ctx, &err, "GetTeamRetentionLocal(%s)", teamID)() 2131 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 2132 if err != nil { 2133 return res, err 2134 } 2135 tlfID, err := chat1.MakeTLFID(teamID.String()) 2136 if err != nil { 2137 return res, err 2138 } 2139 ib, err := h.G().InboxSource.ReadUnverified(ctx, uid, types.InboxSourceDataSourceAll, 2140 &chat1.GetInboxQuery{ 2141 TlfID: &tlfID, 2142 }) 2143 if err != nil { 2144 return res, err 2145 } 2146 if len(ib.ConvsUnverified) == 0 { 2147 return res, errors.New("no conversations found") 2148 } 2149 return ib.ConvsUnverified[0].Conv.TeamRetention, nil 2150 } 2151 2152 func (h *Server) SetConvMinWriterRoleLocal(ctx context.Context, arg chat1.SetConvMinWriterRoleLocalArg) (err error) { 2153 defer h.Trace(ctx, &err, "SetConvMinWriterRole(%v, %v)", arg.ConvID, arg.Role)() 2154 _, err = h.remoteClient().SetConvMinWriterRole(ctx, chat1.SetConvMinWriterRoleArg(arg)) 2155 return err 2156 } 2157 2158 func (h *Server) UpgradeKBFSConversationToImpteam(ctx context.Context, convID chat1.ConversationID) (err error) { 2159 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 2160 defer h.Trace(ctx, &err, "UpgradeKBFSConversationToImpteam(%s)", convID)() 2161 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 2162 if err != nil { 2163 return err 2164 } 2165 2166 ibox, _, err := h.G().InboxSource.Read(ctx, uid, types.ConversationLocalizerBlocking, 2167 types.InboxSourceDataSourceAll, nil, 2168 &chat1.GetInboxLocalQuery{ 2169 ConvIDs: []chat1.ConversationID{convID}, 2170 }) 2171 if err != nil { 2172 return err 2173 } 2174 if len(ibox.Convs) == 0 { 2175 return errors.New("no conversation found") 2176 } 2177 conv := ibox.Convs[0] 2178 if conv.GetMembersType() != chat1.ConversationMembersType_KBFS { 2179 return fmt.Errorf("cannot upgrade %v conversation", conv.GetMembersType()) 2180 } 2181 tlfID := conv.Info.Triple.Tlfid 2182 tlfName := conv.Info.TlfName 2183 public := conv.Info.Visibility == keybase1.TLFVisibility_PUBLIC 2184 return h.G().ChatHelper.UpgradeKBFSToImpteam(ctx, tlfName, tlfID, public) 2185 } 2186 2187 func (h *Server) cancelActiveSearchLocked() { 2188 if h.searchCancelFn != nil { 2189 h.searchCancelFn() 2190 h.searchCancelFn = nil 2191 } 2192 } 2193 2194 func (h *Server) getSearchContext(ctx context.Context) context.Context { 2195 // enforce a single search happening at a time 2196 h.searchMu.Lock() 2197 h.cancelActiveSearchLocked() 2198 ctx, h.searchCancelFn = context.WithCancel(ctx) 2199 h.searchMu.Unlock() 2200 return ctx 2201 } 2202 2203 func (h *Server) CancelActiveSearch(ctx context.Context) (err error) { 2204 defer h.Trace(ctx, &err, "CancelActiveSearch")() 2205 h.searchMu.Lock() 2206 h.cancelActiveSearchLocked() 2207 h.searchMu.Unlock() 2208 return nil 2209 } 2210 2211 func (h *Server) getSearchRegexp(query string, opts chat1.SearchOpts) (re *regexp.Regexp, err error) { 2212 if opts.IsRegex { 2213 re, err = regexp.Compile(query) 2214 } else { 2215 // String queries are set case insensitive 2216 re, err = utils.GetQueryRe(query) 2217 } 2218 if err != nil { 2219 return nil, err 2220 } 2221 return re, nil 2222 } 2223 2224 func (h *Server) SearchRegexp(ctx context.Context, arg chat1.SearchRegexpArg) (res chat1.SearchRegexpRes, err error) { 2225 var identBreaks []keybase1.TLFIdentifyFailure 2226 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, h.identNotifier) 2227 defer h.Trace(ctx, &err, "SearchRegexp")() 2228 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 2229 defer func() { h.setResultRateLimit(ctx, &res) }() 2230 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 2231 if err != nil { 2232 return res, err 2233 } 2234 ctx = h.getSearchContext(ctx) 2235 2236 query, opts := search.UpgradeSearchOptsFromQuery(arg.Query, arg.Opts, 2237 h.G().GetEnv().GetUsername().String()) 2238 re, err := h.getSearchRegexp(query, opts) 2239 if err != nil { 2240 return res, err 2241 } 2242 2243 chatUI := h.getChatUI(arg.SessionID) 2244 uiCh := make(chan chat1.ChatSearchHit, 10) 2245 ch := make(chan struct{}) 2246 go func() { 2247 for searchHit := range uiCh { 2248 select { 2249 case <-ctx.Done(): 2250 return 2251 default: 2252 _ = chatUI.ChatSearchHit(ctx, chat1.ChatSearchHitArg{ 2253 SearchHit: searchHit, 2254 }) 2255 } 2256 } 2257 close(ch) 2258 }() 2259 hits, _, err := h.G().RegexpSearcher.Search(ctx, uid, arg.ConvID, re, uiCh, opts) 2260 if err != nil { 2261 return res, err 2262 } 2263 2264 <-ch 2265 _ = chatUI.ChatSearchDone(ctx, chat1.ChatSearchDoneArg{ 2266 NumHits: len(hits), 2267 }) 2268 return chat1.SearchRegexpRes{ 2269 Hits: hits, 2270 IdentifyFailures: identBreaks, 2271 }, nil 2272 } 2273 2274 func (h *Server) cancelActiveInboxSearchLocked() { 2275 if h.searchInboxCancelFn != nil { 2276 h.searchInboxCancelFn() 2277 h.searchInboxCancelFn = nil 2278 } 2279 } 2280 2281 func (h *Server) getInboxSearchContext(ctx context.Context) context.Context { 2282 // enforce a single search happening at a time 2283 h.searchInboxMu.Lock() 2284 h.cancelActiveInboxSearchLocked() 2285 ctx, h.searchInboxCancelFn = context.WithCancel(ctx) 2286 h.searchInboxMu.Unlock() 2287 return ctx 2288 } 2289 2290 func (h *Server) CancelActiveInboxSearch(ctx context.Context) (err error) { 2291 defer h.Trace(ctx, &err, "CancelActiveInboxSearch")() 2292 h.searchInboxMu.Lock() 2293 h.cancelActiveInboxSearchLocked() 2294 h.searchInboxMu.Unlock() 2295 return nil 2296 } 2297 2298 func (h *Server) delegateInboxSearch(ctx context.Context, uid gregor1.UID, query, origQuery string, 2299 opts chat1.SearchOpts, ui libkb.ChatUI) (res chat1.ChatSearchInboxResults, err error) { 2300 defer h.Trace(ctx, &err, "delegateInboxSearch")() 2301 convs, err := h.G().Indexer.SearchableConvs(ctx, opts.ConvID) 2302 if err != nil { 2303 return res, err 2304 } 2305 re, err := h.getSearchRegexp(query, opts) 2306 if err != nil { 2307 return res, err 2308 } 2309 select { 2310 case <-ctx.Done(): 2311 return res, ctx.Err() 2312 default: 2313 _ = ui.ChatSearchConvHits(ctx, chat1.UIChatSearchConvHits{}) 2314 } 2315 var numHits, numConvs int 2316 for index, conv := range convs { 2317 uiCh := make(chan chat1.ChatSearchHit, 10) 2318 ch := make(chan struct{}) 2319 go func() { 2320 for searchHit := range uiCh { 2321 select { 2322 case <-ctx.Done(): 2323 return 2324 default: 2325 _ = ui.ChatSearchHit(ctx, chat1.ChatSearchHitArg{ 2326 SearchHit: searchHit, 2327 }) 2328 } 2329 } 2330 close(ch) 2331 }() 2332 hits, _, err := h.G().RegexpSearcher.Search(ctx, uid, conv.GetConvID(), re, uiCh, opts) 2333 if err != nil { 2334 h.Debug(ctx, "delegateInboxSearch: failed to search conv: %s", err) 2335 continue 2336 } 2337 <-ch 2338 if len(hits) == 0 { 2339 continue 2340 } 2341 numHits += len(hits) 2342 numConvs++ 2343 inboxHit := chat1.ChatSearchInboxHit{ 2344 ConvID: conv.GetConvID(), 2345 TeamType: conv.GetTeamType(), 2346 ConvName: utils.GetRemoteConvDisplayName(conv), 2347 Query: origQuery, 2348 Time: hits[0].HitMessage.Valid().Ctime, 2349 Hits: hits, 2350 } 2351 select { 2352 case <-ctx.Done(): 2353 return res, ctx.Err() 2354 default: 2355 _ = ui.ChatSearchInboxHit(ctx, chat1.ChatSearchInboxHitArg{ 2356 SearchHit: inboxHit, 2357 }) 2358 } 2359 res.Hits = append(res.Hits, inboxHit) 2360 if opts.MaxConvsHit > 0 && len(res.Hits) > opts.MaxConvsHit { 2361 break 2362 } 2363 if opts.MaxConvsSearched > 0 && index > opts.MaxConvsSearched { 2364 break 2365 } 2366 } 2367 2368 select { 2369 case <-ctx.Done(): 2370 return res, ctx.Err() 2371 default: 2372 _ = ui.ChatSearchInboxDone(ctx, chat1.ChatSearchInboxDoneArg{ 2373 Res: chat1.ChatSearchInboxDone{ 2374 NumHits: numHits, 2375 NumConvs: numConvs, 2376 Delegated: true, 2377 }, 2378 }) 2379 } 2380 return res, nil 2381 } 2382 2383 func (h *Server) SearchInbox(ctx context.Context, arg chat1.SearchInboxArg) (res chat1.SearchInboxRes, err error) { 2384 var identBreaks []keybase1.TLFIdentifyFailure 2385 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, h.identNotifier) 2386 defer h.Trace(ctx, &err, "SearchInbox")() 2387 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 2388 defer func() { h.setResultRateLimit(ctx, &res) }() 2389 defer h.suspendBgConvLoads(ctx)() 2390 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 2391 if err != nil { 2392 return res, err 2393 } 2394 2395 chatUI := h.getChatUI(arg.SessionID) 2396 select { 2397 case <-ctx.Done(): 2398 return res, ctx.Err() 2399 default: 2400 _ = chatUI.ChatSearchInboxStart(ctx) 2401 } 2402 2403 username := h.G().GetEnv().GetUsernameForUID(keybase1.UID(uid.String())).String() 2404 query, opts := search.UpgradeSearchOptsFromQuery(arg.Query, arg.Opts, username) 2405 doSearch := !arg.NamesOnly && len(query) > 0 2406 forceDelegate := false 2407 if arg.Opts.ConvID != nil { 2408 fullyIndexed, err := h.G().Indexer.FullyIndexed(ctx, *arg.Opts.ConvID) 2409 if err != nil { 2410 h.Debug(ctx, "SearchInbox: failed to check fully indexed, delegating... err: %s", err) 2411 forceDelegate = true 2412 } else { 2413 forceDelegate = !fullyIndexed 2414 } 2415 if len(query) < search.MinTokenLength { 2416 forceDelegate = true 2417 } 2418 if forceDelegate { 2419 h.Debug(ctx, "SearchInbox: force delegating since not indexed") 2420 } 2421 ctx = h.getSearchContext(ctx) 2422 } else { 2423 ctx = h.getInboxSearchContext(ctx) 2424 } 2425 2426 if doSearch && (opts.IsRegex || forceDelegate) { 2427 inboxRes, err := h.delegateInboxSearch(ctx, uid, query, arg.Query, opts, chatUI) 2428 if err != nil { 2429 return res, err 2430 } 2431 res.Res = &inboxRes 2432 res.IdentifyFailures = identBreaks 2433 return res, nil 2434 } 2435 eg := errgroup.Group{} 2436 2437 // stream hits back to client UI 2438 hitUICh := make(chan chat1.ChatSearchInboxHit, 10) 2439 var numHits int 2440 eg.Go(func() error { 2441 if !doSearch { 2442 return nil 2443 } 2444 for searchHit := range hitUICh { 2445 numHits += len(searchHit.Hits) 2446 select { 2447 case <-ctx.Done(): 2448 return nil 2449 default: 2450 _ = chatUI.ChatSearchInboxHit(ctx, chat1.ChatSearchInboxHitArg{ 2451 SearchHit: searchHit, 2452 }) 2453 } 2454 } 2455 return nil 2456 }) 2457 2458 // stream index status back to client UI 2459 indexUICh := make(chan chat1.ChatSearchIndexStatus, 10) 2460 eg.Go(func() error { 2461 if !doSearch { 2462 return nil 2463 } 2464 for status := range indexUICh { 2465 select { 2466 case <-ctx.Done(): 2467 return nil 2468 default: 2469 _ = chatUI.ChatSearchIndexStatus(ctx, chat1.ChatSearchIndexStatusArg{ 2470 Status: status, 2471 }) 2472 } 2473 } 2474 return nil 2475 }) 2476 2477 // send up conversation name matches 2478 eg.Go(func() error { 2479 if opts.MaxNameConvs == 0 { 2480 return nil 2481 } 2482 convHits, err := h.G().InboxSource.Search(ctx, uid, query, opts.MaxNameConvs, 2483 types.InboxSourceSearchEmptyModeUnread) 2484 if err != nil { 2485 h.Debug(ctx, "SearchInbox: failed to get conv hits: %s", err) 2486 return nil 2487 } 2488 select { 2489 case <-ctx.Done(): 2490 return ctx.Err() 2491 default: 2492 _ = chatUI.ChatSearchConvHits(ctx, chat1.UIChatSearchConvHits{ 2493 Hits: utils.PresentRemoteConversationsAsSearchHits(convHits, username), 2494 UnreadMatches: len(query) == 0, 2495 }) 2496 } 2497 return nil 2498 }) 2499 2500 // send up team name matches 2501 g := h.G().ExternalG() 2502 mctx := libkb.NewMetaContext(ctx, g) 2503 eg.Go(func() error { 2504 if opts.MaxTeams == 0 { 2505 return nil 2506 } 2507 hits, err := opensearch.Local(mctx, query, 2508 opts.MaxTeams) 2509 if err != nil { 2510 h.Debug(ctx, "SearchInbox: failed to get team hits: %s", err) 2511 return nil 2512 } 2513 select { 2514 case <-ctx.Done(): 2515 return nil 2516 default: 2517 _ = chatUI.ChatSearchTeamHits(ctx, chat1.UIChatSearchTeamHits{ 2518 Hits: hits, 2519 SuggestedMatches: len(query) == 0, 2520 }) 2521 } 2522 return nil 2523 }) 2524 2525 eg.Go(func() error { 2526 hits, err := teambot.NewFeaturedBotLoader(g).SearchLocal( 2527 mctx, keybase1.SearchLocalArg{ 2528 Query: query, 2529 Limit: opts.MaxBots, 2530 SkipCache: opts.SkipBotCache, 2531 }) 2532 if err != nil { 2533 h.Debug(ctx, "SearchInbox: failed to get bot hits: %s", err) 2534 return nil 2535 } 2536 select { 2537 case <-ctx.Done(): 2538 return nil 2539 default: 2540 _ = chatUI.ChatSearchBotHits(ctx, chat1.UIChatSearchBotHits{ 2541 Hits: hits.Bots, 2542 SuggestedMatches: len(query) == 0, 2543 }) 2544 } 2545 return nil 2546 }) 2547 2548 var searchRes *chat1.ChatSearchInboxResults 2549 if doSearch { 2550 select { 2551 case <-time.After(50 * time.Millisecond): 2552 case <-ctx.Done(): 2553 return 2554 } 2555 if searchRes, err = h.G().Indexer.Search(ctx, query, arg.Query, opts, hitUICh, indexUICh); err != nil { 2556 return res, err 2557 } 2558 } 2559 2560 if err := eg.Wait(); err != nil { 2561 h.Debug(ctx, "unable to wait for search: %v") 2562 } 2563 2564 var doneRes chat1.ChatSearchInboxDone 2565 if searchRes != nil { 2566 doneRes = chat1.ChatSearchInboxDone{ 2567 NumHits: numHits, 2568 NumConvs: len(searchRes.Hits), 2569 PercentIndexed: searchRes.PercentIndexed, 2570 } 2571 } 2572 select { 2573 case <-ctx.Done(): 2574 return res, ctx.Err() 2575 default: 2576 _ = chatUI.ChatSearchInboxDone(ctx, chat1.ChatSearchInboxDoneArg{ 2577 Res: doneRes, 2578 }) 2579 } 2580 return chat1.SearchInboxRes{ 2581 Res: searchRes, 2582 IdentifyFailures: identBreaks, 2583 }, nil 2584 } 2585 2586 func (h *Server) ProfileChatSearch(ctx context.Context, identifyBehavior keybase1.TLFIdentifyBehavior) ( 2587 res map[chat1.ConvIDStr]chat1.ProfileSearchConvStats, err error) { 2588 var identBreaks []keybase1.TLFIdentifyFailure 2589 ctx = globals.ChatCtx(ctx, h.G(), identifyBehavior, &identBreaks, h.identNotifier) 2590 defer h.Trace(ctx, &err, "ProfileChatSearch")() 2591 _, err = utils.AssertLoggedInUID(ctx, h.G()) 2592 if err != nil { 2593 return nil, err 2594 } 2595 2596 res, err = h.G().Indexer.IndexInbox(ctx) 2597 if err != nil { 2598 return nil, err 2599 } 2600 b, err := json.Marshal(res) 2601 if err != nil { 2602 return nil, err 2603 } 2604 h.Debug(ctx, "%s\n", string(b)) 2605 return res, err 2606 } 2607 2608 func (h *Server) GetStaticConfig(ctx context.Context) (res chat1.StaticConfig, err error) { 2609 defer h.Trace(ctx, &err, "GetStaticConfig")() 2610 return chat1.StaticConfig{ 2611 DeletableByDeleteHistory: chat1.DeletableMessageTypesByDeleteHistory(), 2612 BuiltinCommands: h.G().CommandsSource.GetBuiltins(ctx), 2613 }, nil 2614 } 2615 2616 func (h *Server) ResolveUnfurlPrompt(ctx context.Context, arg chat1.ResolveUnfurlPromptArg) (err error) { 2617 var identBreaks []keybase1.TLFIdentifyFailure 2618 defer func() { 2619 // squash all errors coming out of here 2620 err = nil 2621 }() 2622 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, h.identNotifier) 2623 defer h.Trace(ctx, &err, "ResolveUnfurlPrompt")() 2624 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 2625 if err != nil { 2626 return err 2627 } 2628 fetchAndUnfurl := func() error { 2629 msgs, err := h.G().ConvSource.GetMessages(ctx, arg.ConvID, uid, []chat1.MessageID{arg.MsgID}, 2630 nil, nil, true) 2631 if err != nil { 2632 return err 2633 } 2634 if len(msgs) != 1 { 2635 return errors.New("message not found") 2636 } 2637 h.G().Unfurler.UnfurlAndSend(ctx, uid, arg.ConvID, msgs[0]) 2638 return nil 2639 } 2640 atyp, err := arg.Result.ActionType() 2641 if err != nil { 2642 return err 2643 } 2644 switch atyp { 2645 case chat1.UnfurlPromptAction_NOTNOW: 2646 // do nothing 2647 case chat1.UnfurlPromptAction_ACCEPT: 2648 if err = h.G().Unfurler.WhitelistAdd(ctx, uid, arg.Result.Accept()); err != nil { 2649 return fmt.Errorf("failed to add to whitelist, doing nothing: %s", err) 2650 } 2651 if err = fetchAndUnfurl(); err != nil { 2652 return fmt.Errorf("failed to fetch and unfurl: %s", err) 2653 } 2654 case chat1.UnfurlPromptAction_ONETIME: 2655 h.G().Unfurler.WhitelistAddExemption(ctx, uid, 2656 unfurl.NewSingleMessageWhitelistExemption(arg.ConvID, arg.MsgID, arg.Result.Onetime())) 2657 if err = fetchAndUnfurl(); err != nil { 2658 return fmt.Errorf("failed to fetch and unfurl: %s", err) 2659 } 2660 case chat1.UnfurlPromptAction_NEVER: 2661 if err = h.G().Unfurler.SetMode(ctx, uid, chat1.UnfurlMode_NEVER); err != nil { 2662 return fmt.Errorf("failed to set mode to never: %s", err) 2663 } 2664 case chat1.UnfurlPromptAction_ALWAYS: 2665 if err = h.G().Unfurler.SetMode(ctx, uid, chat1.UnfurlMode_ALWAYS); err != nil { 2666 return fmt.Errorf("failed to set mode to always: %s", err) 2667 } 2668 if err = fetchAndUnfurl(); err != nil { 2669 return fmt.Errorf("failed to fetch and unfurl: %s", err) 2670 } 2671 } 2672 return nil 2673 } 2674 2675 func (h *Server) GetUnfurlSettings(ctx context.Context) (res chat1.UnfurlSettingsDisplay, err error) { 2676 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 2677 defer h.Trace(ctx, &err, "GetUnfurlSettings")() 2678 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 2679 if err != nil { 2680 return res, err 2681 } 2682 settings, err := h.G().Unfurler.GetSettings(ctx, uid) 2683 if err != nil { 2684 return res, err 2685 } 2686 res.Mode = settings.Mode 2687 for w := range settings.Whitelist { 2688 res.Whitelist = append(res.Whitelist, w) 2689 } 2690 sort.Slice(res.Whitelist, func(i, j int) bool { 2691 return res.Whitelist[i] < res.Whitelist[j] 2692 }) 2693 return res, nil 2694 } 2695 2696 func (h *Server) SaveUnfurlSettings(ctx context.Context, arg chat1.SaveUnfurlSettingsArg) (err error) { 2697 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 2698 defer h.Trace(ctx, &err, "SaveUnfurlSettings")() 2699 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 2700 if err != nil { 2701 return err 2702 } 2703 wm := make(map[string]bool) 2704 for _, w := range arg.Whitelist { 2705 wm[w] = true 2706 } 2707 return h.G().Unfurler.SetSettings(ctx, uid, chat1.UnfurlSettings{ 2708 Mode: arg.Mode, 2709 Whitelist: wm, 2710 }) 2711 } 2712 2713 func (h *Server) ToggleMessageCollapse(ctx context.Context, arg chat1.ToggleMessageCollapseArg) (err error) { 2714 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 2715 defer h.Trace(ctx, &err, "ToggleMessageCollapse convID=%s msgID=%d collapsed=%v", 2716 arg.ConvID, arg.MsgID, arg.Collapse)() 2717 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 2718 if err != nil { 2719 return err 2720 } 2721 if err := utils.NewCollapses(h.G()).ToggleSingle(ctx, uid, arg.ConvID, arg.MsgID, arg.Collapse); err != nil { 2722 return err 2723 } 2724 msg, err := h.G().ConvSource.GetMessage(ctx, arg.ConvID, uid, arg.MsgID, nil, nil, true) 2725 if err != nil { 2726 h.Debug(ctx, "ToggleMessageCollapse: failed to get message: %s", err) 2727 return nil 2728 } 2729 if !msg.IsValid() { 2730 h.Debug(ctx, "ToggleMessageCollapse: invalid message") 2731 return nil 2732 } 2733 if msg.Valid().MessageBody.IsType(chat1.MessageType_UNFURL) { 2734 unfurledMsg, err := h.G().ConvSource.GetMessage(ctx, arg.ConvID, uid, 2735 msg.Valid().MessageBody.Unfurl().MessageID, nil, nil, true) 2736 if err != nil { 2737 h.Debug(ctx, "ToggleMessageCollapse: failed to get unfurl base message: %s", err) 2738 return nil 2739 } 2740 // give a notification about the unfurled message 2741 notif := chat1.MessagesUpdated{ 2742 ConvID: arg.ConvID, 2743 Updates: []chat1.UIMessage{utils.PresentMessageUnboxed(ctx, h.G(), unfurledMsg, uid, arg.ConvID)}, 2744 } 2745 act := chat1.NewChatActivityWithMessagesUpdated(notif) 2746 h.G().ActivityNotifier.Activity(ctx, uid, chat1.TopicType_CHAT, 2747 &act, chat1.ChatActivitySource_LOCAL) 2748 } else if msg.Valid().MessageBody.IsType(chat1.MessageType_ATTACHMENT) { 2749 notif := chat1.MessagesUpdated{ 2750 ConvID: arg.ConvID, 2751 Updates: []chat1.UIMessage{utils.PresentMessageUnboxed(ctx, h.G(), msg, uid, arg.ConvID)}, 2752 } 2753 act := chat1.NewChatActivityWithMessagesUpdated(notif) 2754 h.G().ActivityNotifier.Activity(ctx, uid, chat1.TopicType_CHAT, 2755 &act, chat1.ChatActivitySource_LOCAL) 2756 } 2757 return nil 2758 } 2759 2760 func (h *Server) BulkAddToConv(ctx context.Context, arg chat1.BulkAddToConvArg) (err error) { 2761 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 2762 defer h.Trace(ctx, &err, "BulkAddToConv: convID: %v, numUsers: %v", arg.ConvID, len(arg.Usernames))() 2763 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 2764 if err != nil { 2765 return err 2766 } 2767 return h.G().ChatHelper.BulkAddToConv(ctx, uid, arg.ConvID, arg.Usernames) 2768 } 2769 2770 func (h *Server) BulkAddToManyConvs(ctx context.Context, arg chat1.BulkAddToManyConvsArg) (err error) { 2771 for _, convID := range arg.Conversations { 2772 if err = h.BulkAddToConv(ctx, chat1.BulkAddToConvArg{ 2773 ConvID: convID, 2774 Usernames: arg.Usernames, 2775 }); err != nil { 2776 return err 2777 } 2778 } 2779 return nil 2780 } 2781 2782 func (h *Server) PutReacjiSkinTone(ctx context.Context, skinTone keybase1.ReacjiSkinTone) (res keybase1.UserReacjis, err error) { 2783 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 2784 defer h.Trace(ctx, &err, "PutReacjiSkinTone")() 2785 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 2786 if err != nil { 2787 return res, err 2788 } 2789 store := storage.NewReacjiStore(h.G()) 2790 err = store.PutSkinTone(ctx, uid, skinTone) 2791 if err != nil { 2792 return res, err 2793 } 2794 res = store.UserReacjis(ctx, uid) 2795 return res, nil 2796 } 2797 2798 func (h *Server) ResolveMaybeMention(ctx context.Context, mention chat1.MaybeMention) (err error) { 2799 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 2800 defer h.Trace(ctx, &err, "ResolveMaybeMention")() 2801 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 2802 if err != nil { 2803 return err 2804 } 2805 2806 // Try to load as user 2807 if mention.Channel == "" { 2808 nn := libkb.NewNormalizedUsername(mention.Name) 2809 if _, err = h.G().GetUPAKLoader().LookupUID(ctx, nn); err != nil { 2810 h.Debug(ctx, "ResolveMaybeMention: not a user") 2811 } else { 2812 _ = h.getChatUI(0).ChatMaybeMentionUpdate(ctx, mention.Name, mention.Channel, 2813 chat1.NewUIMaybeMentionInfoWithUser()) 2814 return nil 2815 } 2816 } 2817 // Try to load as team 2818 return h.G().TeamMentionLoader.LoadTeamMention(ctx, uid, mention, nil, true) 2819 } 2820 2821 func (h *Server) getLoadGalleryContext(ctx context.Context) context.Context { 2822 // enforce a single search happening at a time 2823 h.loadGalleryMu.Lock() 2824 if h.loadGalleryCancelFn != nil { 2825 h.loadGalleryCancelFn() 2826 h.loadGalleryCancelFn = nil 2827 } 2828 ctx, h.loadGalleryCancelFn = context.WithCancel(ctx) 2829 h.loadGalleryMu.Unlock() 2830 return ctx 2831 } 2832 2833 func (h *Server) LoadGallery(ctx context.Context, arg chat1.LoadGalleryArg) (res chat1.LoadGalleryRes, err error) { 2834 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 2835 defer h.Trace(ctx, &err, "LoadGallery")() 2836 defer func() { err = h.squashSquashableErrors(err) }() 2837 defer func() { h.setResultRateLimit(ctx, &res) }() 2838 defer h.suspendBgConvLoads(ctx)() 2839 2840 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 2841 if err != nil { 2842 return res, err 2843 } 2844 ctx = h.getLoadGalleryContext(ctx) 2845 chatUI := h.getChatUI(arg.SessionID) 2846 convID := arg.ConvID 2847 var opts attachments.NextMessageOptions 2848 opts.BackInTime = true 2849 switch arg.Typ { 2850 case chat1.GalleryItemTyp_MEDIA: 2851 opts.MessageType = chat1.MessageType_ATTACHMENT 2852 opts.AssetTypes = []chat1.AssetMetadataType{chat1.AssetMetadataType_IMAGE, 2853 chat1.AssetMetadataType_VIDEO} 2854 case chat1.GalleryItemTyp_LINK: 2855 opts.MessageType = chat1.MessageType_TEXT 2856 opts.FilterLinks = true 2857 case chat1.GalleryItemTyp_DOC: 2858 opts.MessageType = chat1.MessageType_ATTACHMENT 2859 opts.AssetTypes = []chat1.AssetMetadataType{chat1.AssetMetadataType_NONE} 2860 default: 2861 return res, errors.New("invalid gallery type") 2862 } 2863 var msgID chat1.MessageID 2864 if arg.FromMsgID != nil { 2865 msgID = *arg.FromMsgID 2866 } else { 2867 conv, err := utils.GetUnverifiedConv(ctx, h.G(), uid, convID, types.InboxSourceDataSourceAll) 2868 if err != nil { 2869 return res, err 2870 } 2871 msgID = conv.Conv.ReaderInfo.MaxMsgid + 1 2872 } 2873 2874 hitCh := make(chan chat1.UIMessage) 2875 go func(ctx context.Context) { 2876 for msg := range hitCh { 2877 _ = chatUI.ChatLoadGalleryHit(ctx, msg) 2878 } 2879 }(ctx) 2880 gallery := attachments.NewGallery(h.G()) 2881 msgs, last, err := gallery.NextMessages(ctx, uid, convID, msgID, arg.Num, opts, hitCh) 2882 if err != nil { 2883 return res, err 2884 } 2885 return chat1.LoadGalleryRes{ 2886 Last: last, 2887 Messages: utils.PresentMessagesUnboxed(ctx, h.G(), msgs, uid, convID), 2888 }, nil 2889 } 2890 2891 func (h *Server) LoadFlip(ctx context.Context, arg chat1.LoadFlipArg) (res chat1.LoadFlipRes, err error) { 2892 var identBreaks []keybase1.TLFIdentifyFailure 2893 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, &identBreaks, h.identNotifier) 2894 defer h.Trace(ctx, &err, "LoadFlip")() 2895 defer func() { h.setResultRateLimit(ctx, &res) }() 2896 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 2897 if err != nil { 2898 return res, err 2899 } 2900 statusCh, errCh := h.G().CoinFlipManager.LoadFlip(ctx, uid, arg.HostConvID, arg.HostMsgID, 2901 arg.FlipConvID, arg.GameID) 2902 select { 2903 case status := <-statusCh: 2904 res.Status = status 2905 case err = <-errCh: 2906 return res, err 2907 } 2908 res.IdentifyFailures = identBreaks 2909 return res, nil 2910 } 2911 2912 func (h *Server) LocationUpdate(ctx context.Context, coord chat1.Coordinate) (err error) { 2913 var identBreaks []keybase1.TLFIdentifyFailure 2914 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, &identBreaks, h.identNotifier) 2915 defer h.Trace(ctx, &err, "LocationUpdate")() 2916 _, err = utils.AssertLoggedInUID(ctx, h.G()) 2917 if err != nil { 2918 return err 2919 } 2920 h.G().LiveLocationTracker.LocationUpdate(ctx, coord) 2921 return nil 2922 } 2923 2924 func (h *Server) AdvertiseBotCommandsLocal(ctx context.Context, arg chat1.AdvertiseBotCommandsLocalArg) (res chat1.AdvertiseBotCommandsLocalRes, err error) { 2925 var identBreaks []keybase1.TLFIdentifyFailure 2926 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, &identBreaks, h.identNotifier) 2927 defer h.Trace(ctx, &err, "AdvertiseBotCommandsLocal")() 2928 defer func() { h.setResultRateLimit(ctx, &res) }() 2929 _, err = utils.AssertLoggedInUID(ctx, h.G()) 2930 if err != nil { 2931 return res, err 2932 } 2933 if err := h.G().BotCommandManager.Advertise(ctx, arg.Alias, arg.Advertisements); err != nil { 2934 return res, err 2935 } 2936 return res, nil 2937 } 2938 2939 func (h *Server) ClearBotCommandsLocal(ctx context.Context, filter *chat1.ClearBotCommandsFilter) (res chat1.ClearBotCommandsLocalRes, err error) { 2940 var identBreaks []keybase1.TLFIdentifyFailure 2941 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, &identBreaks, h.identNotifier) 2942 defer h.Trace(ctx, &err, "ClearBotCommandsLocal")() 2943 defer func() { h.setResultRateLimit(ctx, &res) }() 2944 _, err = utils.AssertLoggedInUID(ctx, h.G()) 2945 if err != nil { 2946 return res, err 2947 } 2948 if err := h.G().BotCommandManager.Clear(ctx, filter); err != nil { 2949 return res, err 2950 } 2951 return res, nil 2952 } 2953 2954 func (h *Server) ListPublicBotCommandsLocal(ctx context.Context, username string) (res chat1.ListBotCommandsLocalRes, err error) { 2955 var identBreaks []keybase1.TLFIdentifyFailure 2956 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, &identBreaks, h.identNotifier) 2957 defer h.Trace(ctx, &err, "ListPublicBotCommandsLocal")() 2958 defer func() { h.setResultRateLimit(ctx, &res) }() 2959 _, err = utils.AssertLoggedInUID(ctx, h.G()) 2960 if err != nil { 2961 return res, err 2962 } 2963 2964 convID, err := h.G().BotCommandManager.PublicCommandsConv(ctx, username) 2965 if err != nil { 2966 if _, ok := err.(UnknownTLFNameError); ok { 2967 h.Debug(ctx, "ListPublicBotCommandsLocal: unknown conv name") 2968 return res, nil 2969 } 2970 return res, err 2971 } 2972 if convID == nil { 2973 return res, nil 2974 } 2975 return h.ListBotCommandsLocal(ctx, *convID) 2976 } 2977 2978 func (h *Server) ListBotCommandsLocal(ctx context.Context, convID chat1.ConversationID) (res chat1.ListBotCommandsLocalRes, err error) { 2979 var identBreaks []keybase1.TLFIdentifyFailure 2980 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, &identBreaks, h.identNotifier) 2981 defer h.Trace(ctx, &err, "ListBotCommandsLocal")() 2982 defer func() { h.setResultRateLimit(ctx, &res) }() 2983 _, err = utils.AssertLoggedInUID(ctx, h.G()) 2984 if err != nil { 2985 return res, err 2986 } 2987 completeCh, err := h.G().BotCommandManager.UpdateCommands(ctx, convID, nil) 2988 if err != nil { 2989 return res, err 2990 } 2991 if err := <-completeCh; err != nil { 2992 h.Debug(ctx, "ListBotCommandsLocal: failed to update commands, list might be stale: %s", err) 2993 } 2994 lres, _, err := h.G().BotCommandManager.ListCommands(ctx, convID) 2995 if err != nil { 2996 return res, err 2997 } 2998 res.Commands = lres 2999 return res, nil 3000 } 3001 3002 func (h *Server) GetMutualTeamsLocal(ctx context.Context, usernames []string) (res chat1.GetMutualTeamsLocalRes, err error) { 3003 var identBreaks []keybase1.TLFIdentifyFailure 3004 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, &identBreaks, h.identNotifier) 3005 defer h.Trace(ctx, &err, "GetMutualTeamsLocal")() 3006 defer func() { h.setResultRateLimit(ctx, &res) }() 3007 defer func() { err = h.handleOfflineError(ctx, err, &res) }() 3008 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3009 if err != nil { 3010 return res, err 3011 } 3012 3013 providedUIDs := make([]keybase1.UID, 0, len(usernames)) 3014 for _, username := range usernames { 3015 providedUIDs = append(providedUIDs, libkb.GetUIDByUsername(h.G().GlobalContext, username)) 3016 } 3017 3018 // get all the default channels for all the teams you're in 3019 chatTopic := chat1.TopicType_CHAT 3020 inbox, err := h.G().InboxSource.ReadUnverified(ctx, uid, types.InboxSourceDataSourceLocalOnly, 3021 &chat1.GetInboxQuery{ 3022 MembersTypes: []chat1.ConversationMembersType{chat1.ConversationMembersType_TEAM}, 3023 TopicName: &globals.DefaultTeamTopic, 3024 TopicType: &chatTopic, 3025 AllowUnseenQuery: true, 3026 }) 3027 if err != nil { 3028 return res, err 3029 } 3030 3031 // loop through convs 3032 var resLock sync.Mutex 3033 pipeliner := pipeliner.NewPipeliner(4) 3034 for _, conv := range inbox.ConvsUnverified { 3035 if conv.GetMembersType() != chat1.ConversationMembersType_TEAM { 3036 continue 3037 } 3038 if err := pipeliner.WaitForRoom(ctx); err != nil { 3039 return res, err 3040 } 3041 go func(conv types.RemoteConversation) { 3042 var err error 3043 userPresent := make(map[keybase1.UID]bool) 3044 for _, uid := range providedUIDs { 3045 userPresent[keybase1.UID(uid.String())] = false 3046 } 3047 members, err := h.G().ParticipantsSource.Get(ctx, uid, conv.GetConvID(), 3048 types.InboxSourceDataSourceAll) 3049 if err != nil { 3050 pipeliner.CompleteOne(err) 3051 return 3052 } 3053 for _, uid := range members { 3054 if _, exists := userPresent[keybase1.UID(uid.String())]; exists { 3055 // if we see a user in a team that we're looking for, mark that in the userPresent map 3056 userPresent[keybase1.UID(uid.String())] = true 3057 } 3058 } 3059 allOK := true 3060 for _, inTeam := range userPresent { 3061 if !inTeam { 3062 allOK = false 3063 break 3064 } 3065 } 3066 if allOK { 3067 resLock.Lock() 3068 res.TeamIDs = append(res.TeamIDs, 3069 keybase1.TeamID(conv.Conv.Metadata.IdTriple.Tlfid.String())) 3070 resLock.Unlock() 3071 } 3072 pipeliner.CompleteOne(nil) 3073 }(conv) 3074 } 3075 err = pipeliner.Flush(ctx) 3076 return res, err 3077 } 3078 3079 func (h *Server) PinMessage(ctx context.Context, arg chat1.PinMessageArg) (res chat1.PinMessageRes, err error) { 3080 var identBreaks []keybase1.TLFIdentifyFailure 3081 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, &identBreaks, h.identNotifier) 3082 defer h.Trace(ctx, &err, "PinMessage")() 3083 defer func() { h.setResultRateLimit(ctx, &res) }() 3084 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3085 if err != nil { 3086 return res, err 3087 } 3088 msg, err := h.G().ConvSource.GetMessage(ctx, arg.ConvID, uid, arg.MsgID, nil, nil, true) 3089 if err != nil { 3090 return res, err 3091 } 3092 if !msg.IsValid() { 3093 return res, fmt.Errorf("unable to pin message, message invalid") 3094 } 3095 bod := msg.Valid().MessageBody 3096 typ, err := bod.MessageType() 3097 if err != nil { 3098 return res, err 3099 } 3100 switch typ { 3101 case chat1.MessageType_TEXT, 3102 chat1.MessageType_ATTACHMENT: 3103 default: 3104 return res, fmt.Errorf("Unable to pin messageof type %v, expected %v or %v", 3105 typ, chat1.MessageType_TEXT, chat1.MessageType_ATTACHMENT) 3106 } 3107 if _, err := h.PostLocalNonblock(ctx, chat1.PostLocalNonblockArg{ 3108 ConversationID: arg.ConvID, 3109 Msg: chat1.MessagePlaintext{ 3110 ClientHeader: chat1.MessageClientHeader{ 3111 MessageType: chat1.MessageType_PIN, 3112 }, 3113 MessageBody: chat1.NewMessageBodyWithPin(chat1.MessagePin{ 3114 MsgID: arg.MsgID, 3115 }), 3116 }, 3117 }); err != nil { 3118 return res, err 3119 } 3120 return res, nil 3121 } 3122 3123 func (h *Server) UnpinMessage(ctx context.Context, convID chat1.ConversationID) (res chat1.PinMessageRes, err error) { 3124 var identBreaks []keybase1.TLFIdentifyFailure 3125 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, &identBreaks, h.identNotifier) 3126 defer h.Trace(ctx, &err, "UnpinMessage")() 3127 defer func() { h.setResultRateLimit(ctx, &res) }() 3128 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3129 if err != nil { 3130 return res, err 3131 } 3132 conv, err := utils.GetVerifiedConv(ctx, h.G(), uid, convID, types.InboxSourceDataSourceAll) 3133 if err != nil { 3134 return res, err 3135 } 3136 pin, err := conv.GetMaxMessage(chat1.MessageType_PIN) 3137 if err != nil { 3138 return res, err 3139 } 3140 if _, err := h.PostLocal(ctx, chat1.PostLocalArg{ 3141 ConversationID: convID, 3142 Msg: chat1.MessagePlaintext{ 3143 ClientHeader: chat1.MessageClientHeader{ 3144 MessageType: chat1.MessageType_DELETE, 3145 Supersedes: pin.GetMessageID(), 3146 }, 3147 }, 3148 }); err != nil { 3149 return res, err 3150 } 3151 return res, nil 3152 } 3153 3154 func (h *Server) IgnorePinnedMessage(ctx context.Context, convID chat1.ConversationID) (err error) { 3155 var identBreaks []keybase1.TLFIdentifyFailure 3156 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, &identBreaks, h.identNotifier) 3157 defer h.Trace(ctx, &err, "IgnorePinnedMessage")() 3158 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3159 if err != nil { 3160 return err 3161 } 3162 conv, err := utils.GetVerifiedConv(ctx, h.G(), uid, convID, types.InboxSourceDataSourceAll) 3163 if err != nil { 3164 return err 3165 } 3166 pin, err := conv.GetMaxMessage(chat1.MessageType_PIN) 3167 if err != nil { 3168 return err 3169 } 3170 if err := storage.NewPinIgnore(h.G(), uid).Ignore(ctx, convID, pin.GetMessageID()); err != nil { 3171 return err 3172 } 3173 h.G().InboxSource.NotifyUpdate(ctx, uid, convID) 3174 return nil 3175 } 3176 3177 func (h *Server) validateBotRole(ctx context.Context, role keybase1.TeamRole) error { 3178 switch role { 3179 case keybase1.TeamRole_BOT, 3180 keybase1.TeamRole_RESTRICTEDBOT: 3181 return nil 3182 default: 3183 return fmt.Errorf("Only %v and %v are valid roles. Found %v", 3184 keybase1.TeamRole_BOT, keybase1.TeamRole_RESTRICTEDBOT, role) 3185 } 3186 } 3187 3188 func (h *Server) teamIDFromTLFName(ctx context.Context, membersType chat1.ConversationMembersType, 3189 tlfName string, isPublic bool) (res keybase1.TeamID, err error) { 3190 3191 switch membersType { 3192 case chat1.ConversationMembersType_KBFS: 3193 return res, errors.New("unable to find a team for KBFS conv") 3194 case chat1.ConversationMembersType_TEAM, chat1.ConversationMembersType_IMPTEAMNATIVE, 3195 chat1.ConversationMembersType_IMPTEAMUPGRADE: 3196 nameInfo, err := CreateNameInfoSource(ctx, h.G(), membersType).LookupID(ctx, tlfName, isPublic) 3197 if err != nil { 3198 return "", err 3199 } 3200 if membersType == chat1.ConversationMembersType_IMPTEAMUPGRADE { 3201 team, err := NewTeamLoader(h.G().ExternalG()).loadTeam(ctx, nameInfo.ID, tlfName, 3202 membersType, isPublic, nil) 3203 if err != nil { 3204 return res, err 3205 } 3206 return team.ID, nil 3207 } 3208 return keybase1.TeamIDFromString(nameInfo.ID.String()) 3209 } 3210 return res, fmt.Errorf("unknown members type: %v", membersType) 3211 } 3212 3213 func (h *Server) fixupTeamErrorWithTLFName(ctx context.Context, username, tlfName string, err error) error { 3214 switch err.(type) { 3215 case nil: 3216 return nil 3217 case libkb.ExistsError: 3218 h.Debug(ctx, "fixupTeamErrorWithTLFName: %v", err) 3219 return libkb.ExistsError{Msg: fmt.Sprintf( 3220 "user %q is already a member of team %q", username, tlfName)} 3221 case libkb.NotFoundError: 3222 h.Debug(ctx, "fixupTeamErrorWithTLFName: %v", err) 3223 return libkb.NotFoundError{Msg: fmt.Sprintf( 3224 "user %q is not a member of team %q", username, tlfName)} 3225 default: 3226 return err 3227 } 3228 } 3229 3230 func (h *Server) teamIDFromConvID(ctx context.Context, uid gregor1.UID, convID chat1.ConversationID) (id keybase1.TeamID, conv chat1.ConversationLocal, err error) { 3231 if conv, err = utils.GetVerifiedConv(ctx, h.G(), uid, convID, types.InboxSourceDataSourceAll); err != nil { 3232 return id, conv, err 3233 } 3234 team, err := NewTeamLoader(h.G().ExternalG()).loadTeam(ctx, conv.Info.Triple.Tlfid, 3235 conv.Info.TlfName, conv.GetMembersType(), conv.IsPublic(), nil) 3236 if err != nil { 3237 return id, conv, err 3238 } 3239 return team.ID, conv, nil 3240 } 3241 3242 func (h *Server) AddBotMember(ctx context.Context, arg chat1.AddBotMemberArg) (err error) { 3243 var identBreaks []keybase1.TLFIdentifyFailure 3244 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, &identBreaks, h.identNotifier) 3245 defer h.Trace(ctx, &err, "AddBotMember")() 3246 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3247 if err != nil { 3248 return err 3249 } 3250 3251 if err := h.validateBotRole(ctx, arg.Role); err != nil { 3252 return err 3253 } 3254 teamID, conv, err := h.teamIDFromConvID(ctx, uid, arg.ConvID) 3255 if err != nil { 3256 return err 3257 } 3258 defer func() { err = h.fixupTeamErrorWithTLFName(ctx, arg.Username, conv.Info.TlfName, err) }() 3259 _, err = teams.AddMemberByID(ctx, h.G().ExternalG(), teamID, arg.Username, arg.Role, arg.BotSettings, nil /* emailInviteMsg */) 3260 if err != nil { 3261 return err 3262 } 3263 err = teams.SendTeamChatWelcomeMessage(ctx, h.G().ExternalG(), teamID, conv.Info.TlfName, 3264 arg.Username, conv.GetMembersType(), arg.Role) 3265 if err != nil { 3266 return err 3267 } 3268 return nil 3269 } 3270 3271 func (h *Server) EditBotMember(ctx context.Context, arg chat1.EditBotMemberArg) (err error) { 3272 var identBreaks []keybase1.TLFIdentifyFailure 3273 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, &identBreaks, h.identNotifier) 3274 defer h.Trace(ctx, &err, "EditBotMember")() 3275 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3276 if err != nil { 3277 return err 3278 } 3279 if err := h.validateBotRole(ctx, arg.Role); err != nil { 3280 return err 3281 } 3282 teamID, _, err := h.teamIDFromConvID(ctx, uid, arg.ConvID) 3283 if err != nil { 3284 return err 3285 } 3286 return teams.EditMemberByID(ctx, h.G().ExternalG(), teamID, arg.Username, arg.Role, arg.BotSettings) 3287 } 3288 3289 func (h *Server) RemoveBotMember(ctx context.Context, arg chat1.RemoveBotMemberArg) (err error) { 3290 var identBreaks []keybase1.TLFIdentifyFailure 3291 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, &identBreaks, h.identNotifier) 3292 defer h.Trace(ctx, &err, "RemoveBotMember")() 3293 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3294 if err != nil { 3295 return err 3296 } 3297 teamID, conv, err := h.teamIDFromConvID(ctx, uid, arg.ConvID) 3298 if err != nil { 3299 return err 3300 } 3301 defer func() { err = h.fixupTeamErrorWithTLFName(ctx, arg.Username, conv.Info.TlfName, err) }() 3302 return teams.RemoveMemberByID(ctx, h.G().ExternalG(), teamID, arg.Username) 3303 } 3304 3305 func (h *Server) SetBotMemberSettings(ctx context.Context, arg chat1.SetBotMemberSettingsArg) (err error) { 3306 var identBreaks []keybase1.TLFIdentifyFailure 3307 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, &identBreaks, h.identNotifier) 3308 defer h.Trace(ctx, &err, "SetBotMemberSettings")() 3309 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3310 if err != nil { 3311 return err 3312 } 3313 teamID, conv, err := h.teamIDFromConvID(ctx, uid, arg.ConvID) 3314 if err != nil { 3315 return err 3316 } 3317 defer func() { err = h.fixupTeamErrorWithTLFName(ctx, arg.Username, conv.Info.TlfName, err) }() 3318 return teams.SetBotSettingsByID(ctx, h.G().ExternalG(), teamID, arg.Username, arg.BotSettings) 3319 } 3320 3321 func (h *Server) GetBotMemberSettings(ctx context.Context, arg chat1.GetBotMemberSettingsArg) (res keybase1.TeamBotSettings, err error) { 3322 var identBreaks []keybase1.TLFIdentifyFailure 3323 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, &identBreaks, h.identNotifier) 3324 defer h.Trace(ctx, &err, "GetBotMemberSettings")() 3325 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3326 if err != nil { 3327 return res, err 3328 } 3329 teamID, conv, err := h.teamIDFromConvID(ctx, uid, arg.ConvID) 3330 if err != nil { 3331 return res, err 3332 } 3333 defer func() { err = h.fixupTeamErrorWithTLFName(ctx, arg.Username, conv.Info.TlfName, err) }() 3334 return teams.GetBotSettingsByID(ctx, h.G().ExternalG(), teamID, arg.Username) 3335 } 3336 3337 func (h *Server) GetTeamRoleInConversation(ctx context.Context, arg chat1.GetTeamRoleInConversationArg) (res keybase1.TeamRole, err error) { 3338 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, nil) 3339 defer h.Trace(ctx, &err, "GetTeamRoleInConversation")() 3340 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3341 if err != nil { 3342 return res, err 3343 } 3344 teamID, _, err := h.teamIDFromConvID(ctx, uid, arg.ConvID) 3345 if err != nil { 3346 return res, err 3347 } 3348 return teams.MemberRoleFromID(ctx, h.G().ExternalG(), teamID, arg.Username) 3349 } 3350 3351 func (h *Server) SimpleSearchInboxConvNames(ctx context.Context, query string) (res []chat1.SimpleSearchInboxConvNamesHit, err error) { 3352 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, nil) 3353 defer h.Trace(ctx, &err, "SimpleSearchInboxConvNames")() 3354 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3355 if err != nil { 3356 return res, err 3357 } 3358 username := h.G().GetEnv().GetUsername().String() 3359 allConvs, err := h.G().InboxSource.Search(ctx, uid, query, 3360 100, types.InboxSourceSearchEmptyModeAllBySendCtime) 3361 if err != nil { 3362 return res, err 3363 } 3364 for _, conv := range allConvs { 3365 switch conv.GetTeamType() { 3366 case chat1.TeamType_NONE: 3367 searchable := utils.SearchableRemoteConversationName(conv, username) 3368 res = append(res, chat1.SimpleSearchInboxConvNamesHit{ 3369 Name: searchable, 3370 ConvID: conv.GetConvID(), 3371 Parts: strings.Split(searchable, ","), 3372 TlfName: utils.GetRemoteConvTLFName(conv), 3373 }) 3374 case chat1.TeamType_SIMPLE, chat1.TeamType_COMPLEX: 3375 res = append(res, chat1.SimpleSearchInboxConvNamesHit{ 3376 Name: utils.SearchableRemoteConversationName(conv, username), 3377 ConvID: conv.GetConvID(), 3378 IsTeam: true, 3379 TlfName: utils.GetRemoteConvTLFName(conv), 3380 }) 3381 } 3382 } 3383 return res, nil 3384 } 3385 3386 func (h *Server) AddBotConvSearch(ctx context.Context, term string) (res []chat1.ConvSearchHit, err error) { 3387 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, nil) 3388 defer h.Trace(ctx, &err, "AddBotConvSearch")() 3389 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3390 if err != nil { 3391 return res, err 3392 } 3393 username := h.G().GetEnv().GetUsername().String() 3394 allConvs, err := h.G().InboxSource.Search(ctx, uid, term, 100, types.InboxSourceSearchEmptyModeAll) 3395 if err != nil { 3396 return res, err 3397 } 3398 res = make([]chat1.ConvSearchHit, 0, len(allConvs)) 3399 for _, conv := range allConvs { 3400 switch conv.GetTeamType() { 3401 case chat1.TeamType_NONE: 3402 searchable := utils.SearchableRemoteConversationName(conv, username) 3403 res = append(res, chat1.ConvSearchHit{ 3404 Name: searchable, 3405 ConvID: conv.GetConvID(), 3406 Parts: strings.Split(searchable, ","), 3407 }) 3408 case chat1.TeamType_SIMPLE: 3409 res = append(res, chat1.ConvSearchHit{ 3410 Name: utils.SearchableRemoteConversationName(conv, username), 3411 ConvID: conv.GetConvID(), 3412 IsTeam: true, 3413 }) 3414 case chat1.TeamType_COMPLEX: 3415 if conv.Conv.Metadata.IsDefaultConv { 3416 res = append(res, chat1.ConvSearchHit{ 3417 Name: utils.GetRemoteConvTLFName(conv), 3418 ConvID: conv.GetConvID(), 3419 IsTeam: true, 3420 }) 3421 } 3422 } 3423 } 3424 return res, nil 3425 } 3426 3427 func (h *Server) TeamIDFromTLFName(ctx context.Context, arg chat1.TeamIDFromTLFNameArg) (res keybase1.TeamID, err error) { 3428 var identBreaks []keybase1.TLFIdentifyFailure 3429 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, &identBreaks, h.identNotifier) 3430 defer h.Trace(ctx, &err, "TeamIDFromTLFName")() 3431 if _, err = utils.AssertLoggedInUID(ctx, h.G()); err != nil { 3432 return res, err 3433 } 3434 3435 return h.teamIDFromTLFName(ctx, arg.MembersType, arg.TlfName, arg.TlfPublic) 3436 } 3437 3438 func (h *Server) DismissJourneycard(ctx context.Context, arg chat1.DismissJourneycardArg) (err error) { 3439 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 3440 defer h.Trace(ctx, &err, "DismissJourneycard")() 3441 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3442 if err != nil { 3443 return err 3444 } 3445 inbox, err := h.G().InboxSource.ReadUnverified(ctx, uid, types.InboxSourceDataSourceLocalOnly, 3446 &chat1.GetInboxQuery{ 3447 ConvID: &arg.ConvID, 3448 SkipBgLoads: true, 3449 AllowUnseenQuery: true, 3450 }) 3451 if err != nil { 3452 return err 3453 } 3454 switch len(inbox.ConvsUnverified) { 3455 case 0: 3456 return fmt.Errorf("could not find conversation") 3457 case 1: 3458 if inbox.ConvsUnverified[0].LocalMetadata == nil { 3459 return fmt.Errorf("no local metadata") 3460 } 3461 teamID, err := keybase1.TeamIDFromString(inbox.ConvsUnverified[0].Conv.Metadata.IdTriple.Tlfid.String()) 3462 if err != nil { 3463 return err 3464 } 3465 h.G().JourneyCardManager.Dismiss(ctx, uid, teamID, arg.ConvID, arg.CardType) 3466 return nil 3467 default: 3468 return fmt.Errorf("got %v conversations but expected 1", len(inbox.ConvsUnverified)) 3469 } 3470 } 3471 3472 const welcomeMessageName = "__welcome_message" 3473 3474 // welcomeMessageMaxLen is duplicated at 3475 // shared/teams/edit-welcome-message/index.tsx:welcomeMessageMaxLen; keep the 3476 // values in sync! 3477 const welcomeMessageMaxLen = 400 3478 3479 func getWelcomeMessageConv(ctx context.Context, g *globals.Context, uid gregor1.UID, teamID keybase1.TeamID) (res types.RemoteConversation, err error) { 3480 onePerTlf := true 3481 tlfID := chat1.TLFID(teamID.ToBytes()) 3482 topicType := chat1.TopicType_CHAT 3483 inbox, err := g.InboxSource.ReadUnverified(ctx, uid, types.InboxSourceDataSourceLocalOnly, 3484 &chat1.GetInboxQuery{ 3485 MembersTypes: []chat1.ConversationMembersType{chat1.ConversationMembersType_TEAM}, 3486 TlfID: &tlfID, 3487 TopicType: &topicType, 3488 AllowUnseenQuery: true, 3489 OneChatTypePerTLF: &onePerTlf, 3490 }) 3491 if err != nil { 3492 return res, err 3493 } 3494 if len(inbox.ConvsUnverified) == 0 { 3495 return res, libkb.NotFoundError{} 3496 } 3497 return inbox.ConvsUnverified[0], nil 3498 } 3499 3500 func getWelcomeMessage(ctx context.Context, g *globals.Context, ri func() chat1.RemoteInterface, uid gregor1.UID, teamID keybase1.TeamID) (message chat1.WelcomeMessageDisplay, err error) { 3501 conv, err := getWelcomeMessageConv(ctx, g, uid, teamID) 3502 if err != nil { 3503 return message, err 3504 } 3505 message = chat1.WelcomeMessageDisplay{Set: false} 3506 s := NewConvDevConversationBackedStorage(g, chat1.TopicType_DEV, true /* adminOnly */, ri) 3507 found, _, err := s.Get(ctx, uid, conv.GetConvID(), welcomeMessageName, &message, false) 3508 if !found { 3509 return message, nil 3510 } 3511 switch err.(type) { 3512 case nil: 3513 case *DevStorageAdminOnlyError: 3514 return message, nil 3515 default: 3516 return message, err 3517 } 3518 if len(message.Raw) > welcomeMessageMaxLen { 3519 return message, nil 3520 } 3521 message.Display = utils.PresentDecoratedTextNoMentions(ctx, message.Raw) 3522 return message, nil 3523 } 3524 3525 func setWelcomeMessage(ctx context.Context, g *globals.Context, ri func() chat1.RemoteInterface, uid gregor1.UID, teamID keybase1.TeamID, message chat1.WelcomeMessage) (err error) { 3526 if len(message.Raw) > welcomeMessageMaxLen { 3527 return fmt.Errorf("welcome message must be at most %d characters; was %d", welcomeMessageMaxLen, len(message.Raw)) 3528 } 3529 conv, err := getWelcomeMessageConv(ctx, g, uid, teamID) 3530 if err != nil { 3531 return err 3532 } 3533 s := NewConvDevConversationBackedStorage(g, chat1.TopicType_DEV, true /* adminOnly */, ri) 3534 return s.Put(ctx, uid, conv.GetConvID(), welcomeMessageName, message) 3535 } 3536 3537 func (h *Server) SetWelcomeMessage(ctx context.Context, arg chat1.SetWelcomeMessageArg) (err error) { 3538 var identBreaks []keybase1.TLFIdentifyFailure 3539 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, &identBreaks, h.identNotifier) 3540 defer h.Trace(ctx, &err, "SetWelcomeMessage")() 3541 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3542 if err != nil { 3543 return err 3544 } 3545 return setWelcomeMessage(ctx, h.G(), h.remoteClient, uid, arg.TeamID, arg.Message) 3546 } 3547 3548 func (h *Server) GetWelcomeMessage(ctx context.Context, teamID keybase1.TeamID) (res chat1.WelcomeMessageDisplay, err error) { 3549 var identBreaks []keybase1.TLFIdentifyFailure 3550 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, &identBreaks, h.identNotifier) 3551 defer h.Trace(ctx, &err, "GetWelcomeMessage")() 3552 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3553 if err != nil { 3554 return res, err 3555 } 3556 return getWelcomeMessage(ctx, h.G(), h.remoteClient, uid, teamID) 3557 } 3558 3559 func (h *Server) GetDefaultTeamChannelsLocal(ctx context.Context, teamID keybase1.TeamID) (res chat1.GetDefaultTeamChannelsLocalRes, err error) { 3560 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 3561 defer h.Trace(ctx, &err, fmt.Sprintf("GetDefaultTeamChannelsLocal %v", teamID))() 3562 defer func() { h.setResultRateLimit(ctx, &res) }() 3563 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3564 if err != nil { 3565 return res, err 3566 } 3567 3568 resp, err := h.remoteClient().GetDefaultTeamChannels(ctx, teamID) 3569 if err != nil { 3570 return res, err 3571 } 3572 if len(resp.Convs) == 0 { 3573 return res, nil 3574 } 3575 topicType := chat1.TopicType_CHAT 3576 query := &chat1.GetInboxLocalQuery{ 3577 ConvIDs: resp.Convs, 3578 TopicType: &topicType, 3579 } 3580 ib, _, err := h.G().InboxSource.Read(ctx, uid, types.ConversationLocalizerBlocking, 3581 types.InboxSourceDataSourceAll, nil, query) 3582 if err != nil { 3583 return res, err 3584 } 3585 res.Convs = utils.PresentConversationLocals(ctx, h.G(), uid, ib.Convs, 3586 utils.PresentParticipantsModeSkip) 3587 return res, nil 3588 } 3589 3590 func (h *Server) SetDefaultTeamChannelsLocal(ctx context.Context, arg chat1.SetDefaultTeamChannelsLocalArg) (res chat1.SetDefaultTeamChannelsLocalRes, err error) { 3591 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 3592 defer h.Trace(ctx, &err, fmt.Sprintf("SetDefaultTeamChannelsLocal: %v", arg.TeamID))() 3593 defer func() { h.setResultRateLimit(ctx, &res) }() 3594 _, err = utils.AssertLoggedInUID(ctx, h.G()) 3595 if err != nil { 3596 return res, err 3597 } 3598 3599 convs := make([]chat1.ConversationID, 0, len(arg.Convs)) 3600 for _, conv := range arg.Convs { 3601 convID, err := chat1.MakeConvID(conv.String()) 3602 if err != nil { 3603 return res, err 3604 } 3605 convs = append(convs, convID) 3606 } 3607 _, err = h.remoteClient().SetDefaultTeamChannels(ctx, chat1.SetDefaultTeamChannelsArg{ 3608 TeamID: arg.TeamID, 3609 Convs: convs, 3610 }) 3611 return res, err 3612 } 3613 3614 func (h *Server) GetLastActiveForTLF(ctx context.Context, tlfIDStr chat1.TLFIDStr) (res chat1.LastActiveStatus, err error) { 3615 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 3616 defer h.Trace(ctx, &err, "GetLastActiveForTLF")() 3617 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3618 if err != nil { 3619 return res, err 3620 } 3621 tlfID, err := chat1.MakeTLFID(tlfIDStr.String()) 3622 if err != nil { 3623 return res, err 3624 } 3625 mtime, err := h.G().TeamChannelSource.GetLastActiveForTLF(ctx, uid, tlfID, chat1.TopicType_CHAT) 3626 if err != nil { 3627 return res, err 3628 } 3629 return utils.ToLastActiveStatus(mtime), nil 3630 } 3631 3632 func (h *Server) GetLastActiveForTeams(ctx context.Context) (res chat1.LastActiveStatusAll, err error) { 3633 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 3634 defer h.Trace(ctx, &err, "GetLastActiveForTeams")() 3635 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3636 if err != nil { 3637 return res, err 3638 } 3639 activity, err := h.G().TeamChannelSource.GetLastActiveForTeams(ctx, uid, chat1.TopicType_CHAT) 3640 if err != nil { 3641 return res, err 3642 } 3643 res.Teams = make(map[chat1.TLFIDStr]chat1.LastActiveStatus) 3644 res.Channels = make(map[chat1.ConvIDStr]chat1.LastActiveStatus) 3645 for tlfID, mtime := range activity.Teams { 3646 res.Teams[tlfID] = utils.ToLastActiveStatus(mtime) 3647 } 3648 for convID, mtime := range activity.Channels { 3649 res.Channels[convID] = utils.ToLastActiveStatus(mtime) 3650 } 3651 return res, nil 3652 } 3653 3654 func (h *Server) GetRecentJoinsLocal(ctx context.Context, convID chat1.ConversationID) (numJoins int, err error) { 3655 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 3656 defer h.Trace(ctx, &err, "GetRecentJoinsLocal")() 3657 _, err = utils.AssertLoggedInUID(ctx, h.G()) 3658 if err != nil { 3659 return 0, err 3660 } 3661 return h.G().TeamChannelSource.GetRecentJoins(ctx, convID, h.remoteClient()) 3662 } 3663 3664 func (h *Server) RefreshParticipants(ctx context.Context, convID chat1.ConversationID) (err error) { 3665 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 3666 defer h.Trace(ctx, &err, "RefreshParticipants")() 3667 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3668 if err != nil { 3669 return err 3670 } 3671 h.G().ParticipantsSource.GetWithNotifyNonblock(ctx, uid, convID, types.InboxSourceDataSourceAll) 3672 return nil 3673 } 3674 3675 func (h *Server) GetParticipants(ctx context.Context, convID chat1.ConversationID) (participants []chat1.ConversationLocalParticipant, err error) { 3676 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_CLI, nil, h.identNotifier) 3677 defer h.Trace(ctx, &err, "GetParticipants")() 3678 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3679 if err != nil { 3680 return nil, err 3681 } 3682 uids, err := h.G().ParticipantsSource.Get(ctx, uid, convID, types.InboxSourceDataSourceAll) 3683 if err != nil { 3684 return nil, err 3685 } 3686 participants, err = h.G().ParticipantsSource.GetParticipantsFromUids(ctx, uids) 3687 if err != nil { 3688 return nil, err 3689 } 3690 return participants, nil 3691 } 3692 3693 func (h *Server) GetLastActiveAtLocal(ctx context.Context, arg chat1.GetLastActiveAtLocalArg) (lastActiveAt gregor1.Time, err error) { 3694 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 3695 defer h.Trace(ctx, &err, "GetLastActiveAtLocal")() 3696 _, err = utils.AssertLoggedInUID(ctx, h.G()) 3697 if err != nil { 3698 return 0, err 3699 } 3700 uid := gregor1.UID(libkb.UsernameToUID(arg.Username).ToBytes()) 3701 return h.G().TeamChannelSource.GetLastActiveAt(ctx, arg.TeamID, uid, h.remoteClient()) 3702 } 3703 3704 func (h *Server) GetLastActiveAtMultiLocal(ctx context.Context, arg chat1.GetLastActiveAtMultiLocalArg) (res map[keybase1.TeamID]gregor1.Time, err error) { 3705 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 3706 defer h.Trace(ctx, &err, "GetLastActiveAtMultiLocal")() 3707 _, err = utils.AssertLoggedInUID(ctx, h.G()) 3708 if err != nil { 3709 return nil, err 3710 } 3711 uid := gregor1.UID(libkb.UsernameToUID(arg.Username).ToBytes()) 3712 3713 res = map[keybase1.TeamID]gregor1.Time{} 3714 resMu := sync.Mutex{} 3715 sem := semaphore.NewWeighted(10) 3716 eg, subctx := errgroup.WithContext(ctx) 3717 for _, teamID := range arg.TeamIDs { 3718 if err := sem.Acquire(subctx, 1); err != nil { 3719 return nil, err 3720 } 3721 3722 teamID := teamID 3723 eg.Go(func() error { 3724 defer sem.Release(1) 3725 3726 lastActive, err := h.G().TeamChannelSource.GetLastActiveAt(subctx, teamID, uid, h.remoteClient()) 3727 if err != nil { 3728 return err 3729 } 3730 resMu.Lock() 3731 res[teamID] = lastActive 3732 resMu.Unlock() 3733 return nil 3734 }) 3735 } 3736 if err := eg.Wait(); err != nil { 3737 return nil, err 3738 } 3739 3740 return res, nil 3741 } 3742 3743 func (h *Server) getEmojiError(err error) *chat1.EmojiError { 3744 if err == nil { 3745 return nil 3746 } 3747 if verr, ok := err.(*EmojiValidationError); ok { 3748 return verr.Export() 3749 } 3750 return &chat1.EmojiError{ 3751 Clidisplay: err.Error(), 3752 Uidisplay: "unknown error", 3753 } 3754 } 3755 3756 func (h *Server) AddEmoji(ctx context.Context, arg chat1.AddEmojiArg) (res chat1.AddEmojiRes, err error) { 3757 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 3758 defer h.Trace(ctx, &err, "AddEmoji")() 3759 defer func() { h.setResultRateLimit(ctx, &res) }() 3760 defer func() { 3761 res.Error = h.getEmojiError(err) 3762 err = nil 3763 }() 3764 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3765 if err != nil { 3766 return res, err 3767 } 3768 if _, err := h.G().EmojiSource.Add(ctx, uid, arg.ConvID, arg.Alias, arg.Filename, arg.AllowOverwrite); err != nil { 3769 return res, err 3770 } 3771 return res, nil 3772 } 3773 3774 func (h *Server) AddEmojis(ctx context.Context, arg chat1.AddEmojisArg) (res chat1.AddEmojisRes, err error) { 3775 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 3776 defer h.Trace(ctx, &err, "AddEmojis")() 3777 defer func() { h.setResultRateLimit(ctx, &res) }() 3778 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3779 if err != nil { 3780 return res, err 3781 } 3782 if len(arg.Aliases) != len(arg.Filenames) { 3783 return chat1.AddEmojisRes{}, errors.New("aliases and filenames have different length") 3784 } 3785 res.FailedFilenames = make(map[string]chat1.EmojiError) 3786 res.SuccessFilenames = make([]string, 0, len(arg.Aliases)) 3787 for i := range arg.Aliases { 3788 _, err := h.G().EmojiSource.Add(ctx, uid, arg.ConvID, arg.Aliases[i], arg.Filenames[i], arg.AllowOverwrite[i]) 3789 if err != nil { 3790 res.FailedFilenames[arg.Filenames[i]] = *h.getEmojiError(err) 3791 } else { 3792 res.SuccessFilenames = append(res.SuccessFilenames, arg.Filenames[i]) 3793 } 3794 } 3795 return res, nil 3796 } 3797 3798 func (h *Server) AddEmojiAlias(ctx context.Context, arg chat1.AddEmojiAliasArg) (res chat1.AddEmojiAliasRes, err error) { 3799 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 3800 defer h.Trace(ctx, &err, "AddEmojiAlias")() 3801 defer func() { h.setResultRateLimit(ctx, &res) }() 3802 defer func() { 3803 res.Error = h.getEmojiError(err) 3804 err = nil 3805 }() 3806 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3807 if err != nil { 3808 return res, err 3809 } 3810 if _, err := h.G().EmojiSource.AddAlias(ctx, uid, arg.ConvID, arg.NewAlias, arg.ExistingAlias); err != nil { 3811 return res, err 3812 } 3813 return res, nil 3814 } 3815 3816 func (h *Server) RemoveEmoji(ctx context.Context, arg chat1.RemoveEmojiArg) (res chat1.RemoveEmojiRes, err error) { 3817 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 3818 defer h.Trace(ctx, &err, "RemoveEmoji")() 3819 defer func() { h.setResultRateLimit(ctx, &res) }() 3820 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3821 if err != nil { 3822 return res, err 3823 } 3824 if err := h.G().EmojiSource.Remove(ctx, uid, arg.ConvID, arg.Alias); err != nil { 3825 return res, err 3826 } 3827 return res, nil 3828 } 3829 3830 func (h *Server) UserEmojis(ctx context.Context, arg chat1.UserEmojisArg) (res chat1.UserEmojiRes, err error) { 3831 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 3832 defer h.Trace(ctx, &err, "UserEmojis")() 3833 defer func() { h.setResultRateLimit(ctx, &res) }() 3834 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3835 if err != nil { 3836 return res, err 3837 } 3838 res.Emojis, err = h.G().EmojiSource.Get(ctx, uid, arg.ConvID, arg.Opts) 3839 return res, err 3840 } 3841 3842 func (h *Server) ToggleEmojiAnimations(ctx context.Context, enabled bool) (err error) { 3843 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, h.identNotifier) 3844 defer h.Trace(ctx, &err, "ToggleEmojiAnimations")() 3845 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3846 if err != nil { 3847 return err 3848 } 3849 return h.G().EmojiSource.ToggleAnimations(ctx, uid, enabled) 3850 } 3851 3852 func (h *Server) ForwardMessage(ctx context.Context, arg chat1.ForwardMessageArg) (res chat1.PostLocalRes, err error) { 3853 var identBreaks []keybase1.TLFIdentifyFailure 3854 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, h.identNotifier) 3855 defer h.Trace(ctx, &err, "ForwardMessage")() 3856 defer func() { h.setResultRateLimit(ctx, &res) }() 3857 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3858 if err != nil { 3859 return res, err 3860 } 3861 reason := chat1.GetThreadReason_FORWARDMSG 3862 msg, err := h.G().ConvSource.GetMessage(ctx, arg.SrcConvID, uid, arg.MsgID, &reason, nil, true) 3863 if err != nil { 3864 return res, err 3865 } else if !msg.IsValid() { 3866 return res, fmt.Errorf("unable to foward message, source is invalid") 3867 } 3868 dstConv, err := utils.GetVerifiedConv(ctx, h.G(), uid, arg.DstConvID, types.InboxSourceDataSourceAll) 3869 if err != nil { 3870 return res, err 3871 } 3872 mvalid := msg.Valid() 3873 switch mvalid.ClientHeader.MessageType { 3874 case chat1.MessageType_ATTACHMENT: 3875 // download from the original source 3876 mbod := msg.Valid().MessageBody.Attachment() 3877 sink, _, err := h.G().AttachmentUploader.GetUploadTempSink(ctx, mbod.Object.Filename) 3878 if err != nil { 3879 return res, err 3880 } 3881 _, err = h.downloadAttachmentLocal(ctx, uid, downloadAttachmentArg{ 3882 SessionID: arg.SessionID, 3883 ConversationID: arg.SrcConvID, 3884 MessageID: arg.MsgID, 3885 IdentifyBehavior: arg.IdentifyBehavior, 3886 Sink: sink, 3887 }) 3888 if err != nil { 3889 return res, err 3890 } 3891 var ephemeralLifetime *gregor1.DurationSec 3892 if md := mvalid.EphemeralMetadata(); md != nil { 3893 ephemeralLifetime = &md.Lifetime 3894 } 3895 return h.PostFileAttachmentLocal(ctx, chat1.PostFileAttachmentLocalArg{ 3896 SessionID: arg.SessionID, 3897 Arg: chat1.PostFileAttachmentArg{ 3898 ConversationID: arg.DstConvID, 3899 TlfName: dstConv.Info.TlfName, 3900 Visibility: dstConv.Info.Visibility, 3901 Filename: sink.Name(), 3902 Title: arg.Title, 3903 Metadata: mbod.Metadata, 3904 IdentifyBehavior: arg.IdentifyBehavior, 3905 EphemeralLifetime: ephemeralLifetime, 3906 }, 3907 }) 3908 default: 3909 return h.PostLocal(ctx, chat1.PostLocalArg{ 3910 SessionID: arg.SessionID, 3911 ConversationID: arg.DstConvID, 3912 Msg: chat1.MessagePlaintext{ 3913 ClientHeader: chat1.MessageClientHeader{ 3914 Conv: dstConv.Info.Triple, 3915 TlfName: dstConv.Info.TlfName, 3916 TlfPublic: dstConv.Info.Visibility == keybase1.TLFVisibility_PUBLIC, 3917 MessageType: mvalid.ClientHeader.MessageType, 3918 EphemeralMetadata: mvalid.EphemeralMetadata(), 3919 }, 3920 MessageBody: mvalid.MessageBody.DeepCopy(), 3921 }, 3922 IdentifyBehavior: arg.IdentifyBehavior, 3923 SkipInChatPayments: true, 3924 }) 3925 } 3926 } 3927 3928 func (h *Server) ForwardMessageNonblock(ctx context.Context, arg chat1.ForwardMessageNonblockArg) (res chat1.PostLocalNonblockRes, err error) { 3929 var identBreaks []keybase1.TLFIdentifyFailure 3930 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, h.identNotifier) 3931 defer h.Trace(ctx, &err, "ForwardMessageNonblock")() 3932 defer h.suspendBgConvLoads(ctx)() 3933 defer func() { h.setResultRateLimit(ctx, &res) }() 3934 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 3935 if err != nil { 3936 return res, err 3937 } 3938 reason := chat1.GetThreadReason_FORWARDMSG 3939 msg, err := h.G().ConvSource.GetMessage(ctx, arg.SrcConvID, uid, arg.MsgID, &reason, nil, true) 3940 if err != nil { 3941 return res, err 3942 } else if !msg.IsValid() { 3943 return res, fmt.Errorf("unable to forward message, source is invalid") 3944 } 3945 dstConv, err := utils.GetVerifiedConv(ctx, h.G(), uid, arg.DstConvID, types.InboxSourceDataSourceAll) 3946 if err != nil { 3947 return res, err 3948 } 3949 mvalid := msg.Valid() 3950 switch mvalid.ClientHeader.MessageType { 3951 case chat1.MessageType_ATTACHMENT: 3952 mbod := msg.Valid().MessageBody.Attachment() 3953 sink, _, err := h.G().AttachmentUploader.GetUploadTempSink(ctx, mbod.Object.Filename) 3954 if err != nil { 3955 return res, err 3956 } 3957 _, err = h.downloadAttachmentLocal(ctx, uid, downloadAttachmentArg{ 3958 SessionID: arg.SessionID, 3959 ConversationID: arg.SrcConvID, 3960 MessageID: arg.MsgID, 3961 IdentifyBehavior: arg.IdentifyBehavior, 3962 Sink: sink, 3963 }) 3964 if err != nil { 3965 return res, err 3966 } 3967 var ephemeralLifetime *gregor1.DurationSec 3968 if md := mvalid.EphemeralMetadata(); md != nil { 3969 ephemeralLifetime = &md.Lifetime 3970 } 3971 return h.PostFileAttachmentLocalNonblock(ctx, chat1.PostFileAttachmentLocalNonblockArg{ 3972 SessionID: arg.SessionID, 3973 Arg: chat1.PostFileAttachmentArg{ 3974 ConversationID: arg.DstConvID, 3975 TlfName: dstConv.Info.TlfName, 3976 Visibility: dstConv.Info.Visibility, 3977 Filename: sink.Name(), 3978 Title: arg.Title, 3979 Metadata: mbod.Metadata, 3980 IdentifyBehavior: arg.IdentifyBehavior, 3981 EphemeralLifetime: ephemeralLifetime, 3982 }, 3983 }) 3984 default: 3985 return h.PostLocalNonblock(ctx, chat1.PostLocalNonblockArg{ 3986 SessionID: arg.SessionID, 3987 ConversationID: arg.DstConvID, 3988 Msg: chat1.MessagePlaintext{ 3989 ClientHeader: chat1.MessageClientHeader{ 3990 Conv: dstConv.Info.Triple, 3991 TlfName: dstConv.Info.TlfName, 3992 TlfPublic: dstConv.Info.Visibility == keybase1.TLFVisibility_PUBLIC, 3993 MessageType: mvalid.ClientHeader.MessageType, 3994 EphemeralMetadata: mvalid.EphemeralMetadata(), 3995 }, 3996 MessageBody: mvalid.MessageBody.DeepCopy(), 3997 }, 3998 IdentifyBehavior: arg.IdentifyBehavior, 3999 SkipInChatPayments: true, 4000 }) 4001 } 4002 } 4003 4004 func (h *Server) ForwardMessageConvSearch(ctx context.Context, term string) (res []chat1.ConvSearchHit, err error) { 4005 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, nil, nil) 4006 defer h.Trace(ctx, &err, "ForwardMessageConvSearch")() 4007 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 4008 if err != nil { 4009 return res, err 4010 } 4011 username := h.G().GetEnv().GetUsername().String() 4012 allConvs, err := h.G().InboxSource.Search(ctx, uid, term, 100, types.InboxSourceSearchEmptyModeAllBySendCtime) 4013 if err != nil { 4014 return res, err 4015 } 4016 res = make([]chat1.ConvSearchHit, 0, len(allConvs)) 4017 for _, conv := range allConvs { 4018 if conv.CannotWrite() { 4019 continue 4020 } 4021 switch conv.GetTeamType() { 4022 case chat1.TeamType_NONE: 4023 searchable := utils.SearchableRemoteConversationName(conv, username) 4024 res = append(res, chat1.ConvSearchHit{ 4025 Name: searchable, 4026 ConvID: conv.GetConvID(), 4027 Parts: strings.Split(searchable, ","), 4028 }) 4029 case chat1.TeamType_SIMPLE, 4030 chat1.TeamType_COMPLEX: 4031 res = append(res, chat1.ConvSearchHit{ 4032 Name: utils.SearchableRemoteConversationName(conv, username), 4033 ConvID: conv.GetConvID(), 4034 IsTeam: true, 4035 }) 4036 } 4037 } 4038 return res, nil 4039 } 4040 4041 func (h *Server) TrackGiphySelect(ctx context.Context, arg chat1.TrackGiphySelectArg) (res chat1.TrackGiphySelectRes, err error) { 4042 var identBreaks []keybase1.TLFIdentifyFailure 4043 ctx = globals.ChatCtx(ctx, h.G(), keybase1.TLFIdentifyBehavior_CHAT_GUI, &identBreaks, 4044 h.identNotifier) 4045 // Never log user content. 4046 defer h.Trace(ctx, &err, "TrackGiphySelect")() 4047 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 4048 if err != nil { 4049 h.Debug(ctx, "TrackGiphySelect: not logged in: %s", err) 4050 return chat1.TrackGiphySelectRes{}, nil 4051 } 4052 err = storage.NewGiphyStore(h.G()).Put(ctx, uid, arg.Result) 4053 return chat1.TrackGiphySelectRes{}, err 4054 } 4055 4056 func (h *Server) ArchiveChat(ctx context.Context, arg chat1.ArchiveChatJobRequest) (res chat1.ArchiveChatRes, err error) { 4057 var identBreaks []keybase1.TLFIdentifyFailure 4058 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, 4059 h.identNotifier) 4060 defer h.Trace(ctx, &err, "ArchiveChat")() 4061 uid, err := utils.AssertLoggedInUID(ctx, h.G()) 4062 if err != nil { 4063 h.Debug(ctx, "ArchiveChat: not logged in: %s", err) 4064 return chat1.ArchiveChatRes{}, nil 4065 } 4066 4067 outpath, err := NewChatArchiver(h.G(), uid, h.remoteClient).ArchiveChat(ctx, arg) 4068 if err != nil { 4069 return chat1.ArchiveChatRes{}, err 4070 } 4071 4072 return chat1.ArchiveChatRes{ 4073 IdentifyFailures: identBreaks, 4074 OutputPath: outpath, 4075 }, err 4076 } 4077 4078 func (h *Server) ArchiveChatList(ctx context.Context, identifyBehavior keybase1.TLFIdentifyBehavior) (res chat1.ArchiveChatListRes, err error) { 4079 var identBreaks []keybase1.TLFIdentifyFailure 4080 ctx = globals.ChatCtx(ctx, h.G(), identifyBehavior, &identBreaks, 4081 h.identNotifier) 4082 defer h.Trace(ctx, &err, "ArchiveChatList")() 4083 _, err = utils.AssertLoggedInUID(ctx, h.G()) 4084 if err != nil { 4085 h.Debug(ctx, "ArchiveChatList: not logged in: %s", err) 4086 return chat1.ArchiveChatListRes{}, nil 4087 } 4088 4089 return h.G().ArchiveRegistry.List(ctx) 4090 } 4091 4092 func (h *Server) ArchiveChatDelete(ctx context.Context, arg chat1.ArchiveChatDeleteArg) (err error) { 4093 var identBreaks []keybase1.TLFIdentifyFailure 4094 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, 4095 h.identNotifier) 4096 defer h.Trace(ctx, &err, "ArchiveChatDelete")() 4097 _, err = utils.AssertLoggedInUID(ctx, h.G()) 4098 if err != nil { 4099 h.Debug(ctx, "ArchiveChatDelete: not logged in: %s", err) 4100 return nil 4101 } 4102 4103 return h.G().ArchiveRegistry.Delete(ctx, arg.JobID, arg.DeleteOutputPath) 4104 } 4105 4106 func (h *Server) ArchiveChatPause(ctx context.Context, arg chat1.ArchiveChatPauseArg) (err error) { 4107 var identBreaks []keybase1.TLFIdentifyFailure 4108 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, 4109 h.identNotifier) 4110 defer h.Trace(ctx, &err, "ArchiveChatPause")() 4111 _, err = utils.AssertLoggedInUID(ctx, h.G()) 4112 if err != nil { 4113 h.Debug(ctx, "ArchiveChatPause: not logged in: %s", err) 4114 return nil 4115 } 4116 4117 return h.G().ArchiveRegistry.Pause(ctx, arg.JobID) 4118 } 4119 4120 func (h *Server) ArchiveChatResume(ctx context.Context, arg chat1.ArchiveChatResumeArg) (err error) { 4121 var identBreaks []keybase1.TLFIdentifyFailure 4122 ctx = globals.ChatCtx(ctx, h.G(), arg.IdentifyBehavior, &identBreaks, 4123 h.identNotifier) 4124 defer h.Trace(ctx, &err, "ArchiveChatResume")() 4125 _, err = utils.AssertLoggedInUID(ctx, h.G()) 4126 if err != nil { 4127 h.Debug(ctx, "ArchiveChatResume: not logged in: %s", err) 4128 return nil 4129 } 4130 4131 return h.G().ArchiveRegistry.Resume(ctx, arg.JobID) 4132 }