raven.go: add arg length check to twt command
This commit is contained in:
parent
057771467d
commit
c274e63e82
4 changed files with 20 additions and 15 deletions
|
@ -1,4 +1,4 @@
|
||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
@ -138,7 +138,7 @@ func parseConfig(configFile string) *Config {
|
||||||
f, err := os.ReadFile(configFile)
|
f, err := os.ReadFile(configFile)
|
||||||
cfg := &Config{}
|
cfg := &Config{}
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Println(err.Error())
|
fmt.Println(err.Error())
|
||||||
return cfg
|
return cfg
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -173,4 +173,4 @@ func parseConfig(configFile string) *Config {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return cfg
|
return cfg
|
||||||
}
|
}
|
||||||
|
|
12
feed.go
12
feed.go
|
@ -80,13 +80,13 @@ func buildFeed(feed *Feed, nick, data string, max int64) error {
|
||||||
lines := strings.Split(data, "\n")
|
lines := strings.Split(data, "\n")
|
||||||
|
|
||||||
for i, l := range lines {
|
for i, l := range lines {
|
||||||
if int64(i) > max {
|
if int64(i) > max {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
parts := strings.Split(l, "\t")
|
parts := strings.Split(l, "\t")
|
||||||
t, err := time.Parse(time.RFC3339, parts[0])
|
t, err := time.Parse(time.RFC3339, parts[0])
|
||||||
if err != nil {
|
if err != nil {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
*feed = append(*feed, FeedEntry{
|
*feed = append(*feed, FeedEntry{
|
||||||
Nick: nick,
|
Nick: nick,
|
||||||
|
@ -134,7 +134,7 @@ func GetFeed(friend, friendsFile string, max int64) error {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
if len(friend) > 0 {
|
if len(friend) > 0 {
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -150,4 +150,4 @@ func GetOwnFeed(nick, feedFile string, max int64) error {
|
||||||
}
|
}
|
||||||
printFeed(feed)
|
printFeed(feed)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
7
raven.go
7
raven.go
|
@ -28,7 +28,12 @@ func run(args []string) error {
|
||||||
} else {
|
} else {
|
||||||
switch args[1] {
|
switch args[1] {
|
||||||
case "twt":
|
case "twt":
|
||||||
return Post(args[2], cfg.FeedFile, cfg.FeedAscend)
|
if len(args) > 2 {
|
||||||
|
return Post(args[2], cfg.FeedFile, cfg.FeedAscend)
|
||||||
|
} else {
|
||||||
|
helpMe(args[0])
|
||||||
|
return errors.New("ARG is required with the twt command")
|
||||||
|
}
|
||||||
case "feed":
|
case "feed":
|
||||||
if len(args) > 2 {
|
if len(args) > 2 {
|
||||||
return GetFeed(args[2], cfg.FriendsFile, cfg.MaxPosts)
|
return GetFeed(args[2], cfg.FriendsFile, cfg.MaxPosts)
|
||||||
|
|
10
types.go
10
types.go
|
@ -5,11 +5,11 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
type Config struct {
|
type Config struct {
|
||||||
FeedFile string
|
FeedFile string
|
||||||
FeedAscend bool
|
FeedAscend bool
|
||||||
FriendsFile string
|
FriendsFile string
|
||||||
MaxPosts int64
|
MaxPosts int64
|
||||||
Nick string
|
Nick string
|
||||||
}
|
}
|
||||||
|
|
||||||
type FeedEntry struct {
|
type FeedEntry struct {
|
||||||
|
|
Loading…
Reference in a new issue