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

Add flag to auto-open page in web browser

Browse Source
This commit is contained in:
Alessio 2023-12-26 12:20:58 -06:00
parent f3f2550d85
commit 795b7250ed
2 changed files with 24 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
cmd/twitter/main.go
View File

@ -15,16 +15,11 @@ import (
"gitlab.com/offline-twitter/twitter_offline_engine/pkg/scraper"
)
/**
* Global variable referencing the open data profile
*/
// Global variable referencing the open data profile
var profile persistence.Profile
var version_string string
/**
* Main method
*/
func main() {
profile_dir := flag.String("profile", ".", "")
flag.StringVar(profile_dir, "p", ".", "")
@ -159,7 +154,10 @@ func main() {
case "unlike_tweet":
unlike_tweet(target)
case "webserver":
start_webserver(*addr)
fs := flag.NewFlagSet("", flag.ExitOnError)
should_auto_open := fs.Bool("auto-open", false, "")
fs.Parse(args[1:])
start_webserver(*addr, *should_auto_open)
case "fetch_inbox":
fetch_inbox(*how_many)
case "fetch_dm":
@ -184,7 +182,6 @@ func main() {
// args:
// - username: twitter username or email address
// - password: twitter account password
func login(username string, password string) {
// Skip the scraper.the_api variable, just use a local one since no scraping is happening
api := scraper.NewGuestSession()
@ -406,9 +403,9 @@ func list_followed() {
}
}
func start_webserver(addr string) {
func start_webserver(addr string, should_auto_open bool) {
app := webserver.NewApp(profile)
app.Run(addr)
app.Run(addr, should_auto_open)
}
func fetch_inbox(how_many int) {

18
internal/webserver/server.go
View File

@ -8,6 +8,8 @@ import (
"log"
"net/http"
"os"
"os/exec"
"runtime"
"path"
"strconv"
"strings"
@ -130,7 +132,7 @@ func (app *Application) ServeHTTP(w http.ResponseWriter, r *http.Request) {
}
}
func (app *Application) Run(address string) {
func (app *Application) Run(address string, should_auto_open bool) {
srv := &http.Server{
Addr: address,
ErrorLog: app.ErrorLog,
@ -147,10 +149,24 @@ func (app *Application) Run(address string) {
app.start_background()
if should_auto_open {
go openWebPage("http://" + address)
}
err := srv.ListenAndServe()
app.ErrorLog.Fatal(err)
}
func openWebPage(url string) error {
switch runtime.GOOS {
case "darwin": // macOS
return exec.Command("open", url).Run()
case "windows":
return exec.Command("cmd", "/c", "start", url).Run()
default: // Linux and others
return exec.Command("xdg-open", url).Run()
}
}
func parse_cursor_value(c *persistence.Cursor, r *http.Request) error {
cursor_param := r.URL.Query().Get("cursor")
if cursor_param != "" {