mirror of https://github.com/Rudi9719/kbtui.git
Casper Weiss Bang
5 years ago
9 changed files with 399 additions and 107 deletions
@ -0,0 +1,34 @@
@@ -0,0 +1,34 @@
|
||||
// +build !rm_basic_commands allcommands followcmd
|
||||
|
||||
package main |
||||
|
||||
import ( |
||||
"fmt" |
||||
) |
||||
|
||||
func init() { |
||||
command := Command{ |
||||
Cmd: []string{"follow"}, |
||||
Description: "$username - Follows the given user", |
||||
Help: "", |
||||
Exec: cmdFollow, |
||||
} |
||||
RegisterCommand(command) |
||||
} |
||||
|
||||
func cmdFollow(cmd []string) { |
||||
if len(cmd) == 2 { |
||||
go follow(cmd[1]) |
||||
} else { |
||||
printFollowHelp() |
||||
} |
||||
} |
||||
func follow(username string) { |
||||
k.Exec("follow", username, "-y") |
||||
printInfoF("Now follows $TEXT", config.Colors.Message.LinkKeybase.stylize(username)) |
||||
followedInSteps[username] = 1 |
||||
} |
||||
|
||||
func printFollowHelp() { |
||||
printInfo(fmt.Sprintf("To follow a user use %sfollow <username>", config.Basics.CmdPrefix)) |
||||
} |
||||
@ -0,0 +1,135 @@
@@ -0,0 +1,135 @@
|
||||
// +build !rm_basic_commands allcommands inspectcmd
|
||||
|
||||
package main |
||||
|
||||
import ( |
||||
"fmt" |
||||
"regexp" |
||||
"samhofi.us/x/keybase" |
||||
"strconv" |
||||
"strings" |
||||
) |
||||
|
||||
func init() { |
||||
command := Command{ |
||||
Cmd: []string{"inspect", "id"}, |
||||
Description: "$identifier - shows info about $identifier ($identifier is either username, messageId or team)", |
||||
Help: "", |
||||
Exec: cmdInspect, |
||||
} |
||||
|
||||
RegisterCommand(command) |
||||
} |
||||
|
||||
func cmdInspect(cmd []string) { |
||||
if len(cmd) == 2 { |
||||
regexIsNumeric := regexp.MustCompile(`^\d+$`) |
||||
if regexIsNumeric.MatchString(cmd[1]) { |
||||
// Then it must be a message id
|
||||
id, _ := strconv.Atoi(cmd[1]) |
||||
go printMessage(id) |
||||
|
||||
} else { |
||||
go printUser(strings.ReplaceAll(cmd[1], "@", "")) |
||||
} |
||||
|
||||
} else { |
||||
printInfo(fmt.Sprintf("To inspect something use %sid <username/messageId>", config.Basics.CmdPrefix)) |
||||
} |
||||
|
||||
} |
||||
func printMessage(id int) { |
||||
chat := k.NewChat(channel) |
||||
messages, err := chat.ReadMessage(id) |
||||
if err == nil { |
||||
var response StyledString |
||||
if messages != nil && len((*messages).Result.Messages) > 0 { |
||||
message := (*messages).Result.Messages[0].Msg |
||||
var apiCast keybase.ChatAPI |
||||
apiCast.Msg = &message |
||||
response = formatOutput(apiCast) |
||||
} else { |
||||
response = config.Colors.Feed.Error.stylize("message not found") |
||||
} |
||||
printToView("Chat", response.string()) |
||||
} |
||||
} |
||||
|
||||
func formatProofs(userLookup keybase.UserAPI) StyledString { |
||||
messageColor := config.Colors.Message |
||||
message := basicStyle.stylize("") |
||||
for _, proof := range userLookup.Them[0].ProofsSummary.All { |
||||
style := config.Colors.Feed.Success |
||||
if proof.State != 1 { |
||||
style = config.Colors.Feed.Error |
||||
} |
||||
proofString := style.stylize("Proof [$NAME@$SITE]: $URL\n") |
||||
proofString = proofString.replace("$NAME", messageColor.SenderDefault.stylize(proof.Nametag)) |
||||
proofString = proofString.replace("$SITE", messageColor.SenderDevice.stylize(proof.ProofType)) |
||||
proofString = proofString.replace("$URL", messageColor.LinkURL.stylize(proof.HumanURL)) |
||||
message = message.append(proofString) |
||||
} |
||||
return message.appendString("\n") |
||||
} |
||||
func formatProfile(userLookup keybase.UserAPI) StyledString { |
||||
messageColor := config.Colors.Message |
||||
user := userLookup.Them[0] |
||||
profileText := messageColor.Body.stylize("Name: $FNAME\nLocation: $LOC\nBio: $BIO\n") |
||||
profileText = profileText.replaceString("$FNAME", user.Profile.FullName) |
||||
profileText = profileText.replaceString("$LOC", user.Profile.Location) |
||||
profileText = profileText.replaceString("$BIO", user.Profile.Bio) |
||||
|
||||
return profileText |
||||
} |
||||
|
||||
func formatFollowState(userLookup keybase.UserAPI) StyledString { |
||||
username := userLookup.Them[0].Basics.Username |
||||
followSteps := followedInSteps[username] |
||||
if followSteps == 1 { |
||||
return config.Colors.Feed.Success.stylize("<Followed!>\n\n") |
||||
} else if followSteps > 1 { |
||||
var steps []string |
||||
for head := username; head != ""; head = trustTreeParent[head] { |
||||
steps = append(steps, fmt.Sprintf("[%s]", head)) |
||||
} |
||||
trustLine := fmt.Sprintf("Indirect follow: <%s>\n\n", strings.Join(steps, " Followed by ")) |
||||
return config.Colors.Message.Body.stylize(trustLine) |
||||
} |
||||
|
||||
return basicStyle.stylize("") |
||||
|
||||
} |
||||
|
||||
func formatFollowerAndFollowedList(username string, listType string) StyledString { |
||||
messageColor := config.Colors.Message |
||||
response := basicStyle.stylize("") |
||||
bytes, _ := k.Exec("list-"+listType, username) |
||||
bigString := string(bytes) |
||||
lines := strings.Split(bigString, "\n") |
||||
response = response.appendString(fmt.Sprintf("%s (%d): ", listType, len(lines)-1)) |
||||
for i, user := range lines[:len(lines)-1] { |
||||
if i != 0 { |
||||
response = response.appendString(", ") |
||||
} |
||||
response = response.append(messageColor.LinkKeybase.stylize(user)) |
||||
response = response.append(getUserFlags(user)) |
||||
} |
||||
return response.appendString("\n\n") |
||||
} |
||||
|
||||
func printUser(username string) { |
||||
messageColor := config.Colors.Message |
||||
|
||||
userLookup, _ := k.UserLookup(username) |
||||
|
||||
response := messageColor.Header.stylize("[Inspecting `$USER`]\n") |
||||
response = response.replace("$USER", messageColor.SenderDefault.stylize(username)) |
||||
response = response.append(formatProfile(userLookup)) |
||||
response = response.append(formatFollowState(userLookup)) |
||||
|
||||
response = response.append(formatProofs(userLookup)) |
||||
response = response.append(formatFollowerAndFollowedList(username, "followers")) |
||||
response = response.append(formatFollowerAndFollowedList(username, "following")) |
||||
|
||||
printToView("Chat", response.string()) |
||||
} |
||||
@ -0,0 +1,33 @@
@@ -0,0 +1,33 @@
|
||||
// +build !rm_basic_commands allcommands followcmd
|
||||
|
||||
package main |
||||
|
||||
import ( |
||||
"fmt" |
||||
) |
||||
|
||||
func init() { |
||||
command := Command{ |
||||
Cmd: []string{"unfollow"}, |
||||
Description: "$username - Unfollows the given user", |
||||
Help: "", |
||||
Exec: cmdUnfollow, |
||||
} |
||||
RegisterCommand(command) |
||||
} |
||||
|
||||
func cmdUnfollow(cmd []string) { |
||||
if len(cmd) == 2 { |
||||
go unfollow(cmd[1]) |
||||
} else { |
||||
printUnfollowHelp() |
||||
} |
||||
} |
||||
func unfollow(username string) { |
||||
k.Exec("unfollow", username) |
||||
printInfoF("Now unfollows $TEXT", config.Colors.Message.LinkKeybase.stylize(username)) |
||||
} |
||||
|
||||
func printUnfollowHelp() { |
||||
printInfo(fmt.Sprintf("To unfollow a user use %sunfollow <username>", config.Basics.CmdPrefix)) |
||||
} |
||||
@ -0,0 +1,58 @@
@@ -0,0 +1,58 @@
|
||||
package main |
||||
|
||||
import ( |
||||
"fmt" |
||||
"strings" |
||||
) |
||||
|
||||
var followedInSteps = make(map[string]int) |
||||
var trustTreeParent = make(map[string]string) |
||||
|
||||
func clearFlagCache() { |
||||
followedInSteps = make(map[string]int) |
||||
trustTreeParent = make(map[string]string) |
||||
} |
||||
|
||||
var maxDepth = 4 |
||||
|
||||
func generateFollowersList() { |
||||
// Does a BFS of followedInSteps
|
||||
queue := []string{k.Username} |
||||
printInfo("Generating Tree of Trust...") |
||||
lastDepth := 1 |
||||
for len(queue) > 0 { |
||||
head := queue[0] |
||||
queue = queue[1:] |
||||
depth := followedInSteps[head] + 1 |
||||
if depth > maxDepth { |
||||
continue |
||||
} |
||||
if depth > lastDepth { |
||||
printInfo(fmt.Sprintf("Trust generated at Level #%d", depth-1)) |
||||
lastDepth = depth |
||||
} |
||||
|
||||
bytes, _ := k.Exec("list-following", head) |
||||
bigString := string(bytes) |
||||
following := strings.Split(bigString, "\n") |
||||
for _, user := range following { |
||||
if followedInSteps[user] == 0 && user != k.Username { |
||||
followedInSteps[user] = depth |
||||
trustTreeParent[user] = head |
||||
queue = append(queue, user) |
||||
} |
||||
} |
||||
} |
||||
printInfo(fmt.Sprintf("Trust-level estabilished for %d users", len(followedInSteps))) |
||||
} |
||||
|
||||
func getUserFlags(username string) StyledString { |
||||
tags := "" |
||||
followDepth := followedInSteps[username] |
||||
if followDepth == 1 { |
||||
tags += fmt.Sprintf(" %s", config.Formatting.IconFollowingUser) |
||||
} else if followDepth > 1 { |
||||
tags += fmt.Sprintf(" %s%d", config.Formatting.IconIndirectFollowUser, followDepth-1) |
||||
} |
||||
return config.Colors.Message.SenderTags.stylize(tags) |
||||
} |
||||
Loading…
Reference in new issue