Skip to content
Snippets Groups Projects
Commit 9b8afc50 authored by Josh Brooks's avatar Josh Brooks
Browse files

Fix usages of Search/Lookup in cmd

parent ecc60371
No related branches found
No related tags found
2 merge requests!510Release,!289Have ud.Search/Lookup take in udE2e
...@@ -37,14 +37,14 @@ var udCmd = &cobra.Command{ ...@@ -37,14 +37,14 @@ var udCmd = &cobra.Command{
cmixParams, e2eParams := initParams() cmixParams, e2eParams := initParams()
authCbs := makeAuthCallbacks( authCbs := makeAuthCallbacks(
viper.GetBool(unsafeChannelCreationFlag), e2eParams) viper.GetBool(unsafeChannelCreationFlag), e2eParams)
client := initE2e(cmixParams, e2eParams, authCbs) messenger := initE2e(cmixParams, e2eParams, authCbs)
// get user and save contact to file // get user and save contact to file
user := client.GetReceptionIdentity() user := messenger.GetReceptionIdentity()
jww.INFO.Printf("[UD]User: %s", user.ID) jww.INFO.Printf("[UD]User: %s", user.ID)
writeContact(user.GetContact()) writeContact(user.GetContact())
err := client.StartNetworkFollower(50 * time.Millisecond) err := messenger.StartNetworkFollower(50 * time.Millisecond)
if err != nil { if err != nil {
jww.FATAL.Panicf("%+v", err) jww.FATAL.Panicf("%+v", err)
} }
...@@ -53,7 +53,7 @@ var udCmd = &cobra.Command{ ...@@ -53,7 +53,7 @@ var udCmd = &cobra.Command{
// Wait until connected or crash on timeout // Wait until connected or crash on timeout
connected := make(chan bool, 10) connected := make(chan bool, 10)
client.GetCmix().AddHealthCallback( messenger.GetCmix().AddHealthCallback(
func(isconnected bool) { func(isconnected bool) {
connected <- isconnected connected <- isconnected
}) })
...@@ -62,14 +62,14 @@ var udCmd = &cobra.Command{ ...@@ -62,14 +62,14 @@ var udCmd = &cobra.Command{
jww.TRACE.Printf("[UD] Connected!") jww.TRACE.Printf("[UD] Connected!")
// Make user discovery manager // Make user discovery manager
rng := client.GetRng() rng := messenger.GetRng()
userToRegister := viper.GetString(udRegisterFlag) userToRegister := viper.GetString(udRegisterFlag)
jww.TRACE.Printf("[UD] Registering user %v...", userToRegister) jww.TRACE.Printf("[UD] Registering user %v...", userToRegister)
userDiscoveryMgr, err := ud.NewManager(client, client.GetComms(), userDiscoveryMgr, err := ud.NewManager(messenger, messenger.GetComms(),
client.NetworkFollowerStatus, userToRegister, nil) messenger.NetworkFollowerStatus, userToRegister, nil)
if err != nil { if err != nil {
if strings.Contains(err.Error(), ud.IsRegisteredErr) { if strings.Contains(err.Error(), ud.IsRegisteredErr) {
userDiscoveryMgr, err = ud.LoadManager(client, client.GetComms()) userDiscoveryMgr, err = ud.LoadManager(messenger, messenger.GetComms())
if err != nil { if err != nil {
jww.FATAL.Panicf("Failed to load UD manager: %+v", err) jww.FATAL.Panicf("Failed to load UD manager: %+v", err)
} }
...@@ -146,8 +146,7 @@ var udCmd = &cobra.Command{ ...@@ -146,8 +146,7 @@ var udCmd = &cobra.Command{
} }
stream := rng.GetStream() stream := rng.GetStream()
_, _, err = ud.Lookup(client.GetCmix(), _, _, err = ud.Lookup(messenger,
stream, client.GetE2E().GetGroup(),
udContact, cb, lookupID, single.GetDefaultRequestParams()) udContact, cb, lookupID, single.GetDefaultRequestParams())
if err != nil { if err != nil {
jww.WARN.Printf("Failed UD lookup: %+v", err) jww.WARN.Printf("Failed UD lookup: %+v", err)
...@@ -166,13 +165,13 @@ var udCmd = &cobra.Command{ ...@@ -166,13 +165,13 @@ var udCmd = &cobra.Command{
} }
jww.INFO.Printf("[UD] BATCHADD: Running") jww.INFO.Printf("[UD] BATCHADD: Running")
restored, _, _, err := xxmutils.RestoreContactsFromBackup( restored, _, _, err := xxmutils.RestoreContactsFromBackup(
idListFile, client, userDiscoveryMgr, nil) idListFile, messenger, userDiscoveryMgr, nil)
if err != nil { if err != nil {
jww.FATAL.Panicf("%+v", err) jww.FATAL.Panicf("%+v", err)
} }
for i := 0; i < len(restored); i++ { for i := 0; i < len(restored); i++ {
uid := restored[i] uid := restored[i]
for !client.GetE2E().HasAuthenticatedChannel(uid) { for !messenger.GetE2E().HasAuthenticatedChannel(uid) {
time.Sleep(time.Second) time.Sleep(time.Second)
} }
jww.INFO.Printf("[UD] Authenticated channel established for %s", uid) jww.INFO.Printf("[UD] Authenticated channel established for %s", uid)
...@@ -225,7 +224,7 @@ var udCmd = &cobra.Command{ ...@@ -225,7 +224,7 @@ var udCmd = &cobra.Command{
} }
if len(facts) == 0 { if len(facts) == 0 {
err = client.StopNetworkFollower() err = messenger.StopNetworkFollower()
if err != nil { if err != nil {
jww.WARN.Print(err) jww.WARN.Print(err)
} }
...@@ -241,19 +240,15 @@ var udCmd = &cobra.Command{ ...@@ -241,19 +240,15 @@ var udCmd = &cobra.Command{
} }
} }
stream := rng.GetStream()
defer stream.Close()
jww.INFO.Printf("[UD] Search: %v", facts) jww.INFO.Printf("[UD] Search: %v", facts)
_, _, err = ud.Search(client.GetCmix(), _, _, err = ud.Search(messenger,
client.GetEventReporter(),
stream, client.GetE2E().GetGroup(),
udContact, cb, facts, single.GetDefaultRequestParams()) udContact, cb, facts, single.GetDefaultRequestParams())
if err != nil { if err != nil {
jww.FATAL.Panicf("%+v", err) jww.FATAL.Panicf("%+v", err)
} }
time.Sleep(91 * time.Second) time.Sleep(91 * time.Second)
err = client.StopNetworkFollower() err = messenger.StopNetworkFollower()
if err != nil { if err != nil {
jww.WARN.Print(err) jww.WARN.Print(err)
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment