me/offline-twitter
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

REFACTOR: make helper function use the scraper tweet-url parsing method

Browse Source
This commit is contained in:
Alessio 2022-03-13 14:50:08 -07:00
parent 8c2fb965aa
commit 8261087103
1 changed files with 6 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
cmd/twitter/helpers.go
View File

@ -3,9 +3,7 @@ package main
import ( import (
"fmt" "fmt"
"os" "os"
"regexp"
"strconv" "strconv"
"strings"
"offline_twitter/scraper" "offline_twitter/scraper"
"offline_twitter/terminal_utils" "offline_twitter/terminal_utils"
@ -104,23 +102,11 @@ func happy_exit(text string) {
* returns: the id at the end of the tweet: e.g., 1395882872729477131 * returns: the id at the end of the tweet: e.g., 1395882872729477131
*/ */
func extract_id_from(url string) (scraper.TweetID, error) { func extract_id_from(url string) (scraper.TweetID, error) {
var id_str string _, id, is_ok := scraper.TryParseTweetUrl(url)
if is_ok {
return id, nil
}
if regexp.MustCompile(`^\d+$`).MatchString(url) { num, err := strconv.Atoi(url)
id_str = url return scraper.TweetID(num), err
} else {
parts := strings.Split(url, "/")
if len(parts) != 6 {
return 0, fmt.Errorf("Tweet format isn't right (%d)", len(parts))
}
if parts[0] != "https:" || parts[1] != "" || parts[2] != "twitter.com" || parts[4] != "status" {
return 0, fmt.Errorf("Tweet format isn't right")
}
id_str = parts[5]
}
id, err := strconv.Atoi(id_str)
if err != nil {
return 0, err
}
return scraper.TweetID(id), nil
} }