Skip to content

Commit

Permalink
fix: because the file was unintentionally closed
Browse files Browse the repository at this point in the history
  • Loading branch information
hirokisan committed Mar 12, 2023
1 parent eabf1f0 commit d3050f1
Showing 1 changed file with 17 additions and 20 deletions.
37 changes: 17 additions & 20 deletions main.go
Original file line number Diff line number Diff line change
Expand Up @@ -12,12 +12,6 @@ import (
"github.com/spf13/cobra"
)

func main() {
if err := cmd.Execute(); err != nil {
log.Fatal(err)
}
}

var cmd = &cobra.Command{
Use: "chatty",
Short: "Would you like to have a little chat with us between jobs?",
Expand All @@ -30,12 +24,23 @@ var cmd = &cobra.Command{
}
client := openai.NewClient(key)

messenger, err := getMessenger()
if err != nil {
return fmt.Errorf("get messenger: %w", err)
var messenger message.Messenger
{
messagesPath, acceptMessageHistory := os.LookupEnv("CHATTY_MESSAGES_PATH")
if acceptMessageHistory {
file, err := os.OpenFile(messagesPath, os.O_RDWR|os.O_CREATE, 0666)
if err != nil {
return fmt.Errorf("create or open file: %w", err)
}
defer file.Close()
messenger = message.NewHistoricalMessenger(file)
} else {
messenger = message.NewIndependentMessenger()
}
}

msg := strings.Join(args, " ")

reply, err := messenger.GetReply(ctx, client, msg)
if err != nil {
return fmt.Errorf("get reply: %w", err)
Expand All @@ -46,16 +51,8 @@ var cmd = &cobra.Command{
},
}

func getMessenger() (message.Messenger, error) {
messagesPath, acceptMessageHistory := os.LookupEnv("CHATTY_MESSAGES_PATH")
if acceptMessageHistory {
file, err := os.OpenFile(messagesPath, os.O_RDWR|os.O_CREATE, 0666)
if err != nil {
return nil, fmt.Errorf("create or open file: %w", err)
}
defer file.Close()

return message.NewHistoricalMessenger(file), nil
func main() {
if err := cmd.Execute(); err != nil {
log.Fatal(err)
}
return message.NewIndependentMessenger(), nil
}

0 comments on commit d3050f1

Please sign in to comment.