package cmd import ( "fmt" "github.com/bdlm/log" "github.com/bdlm/std/logger" "github.com/spf13/cobra" "github.com/o3ma/o3" "dev.sum7.eu/genofire/golang-lib/database" "dev.sum7.eu/genofire/golang-lib/file" // for init database "dev.sum7.eu/genofire/thrempp/models" ) type Config struct { LogLevel logger.Level `toml:"log_level"` Database database.Config `toml:"database"` } var configPath string // serveCmd represents the serve command var serveCmd = &cobra.Command{ Use: "serve", Short: "Run xmpp transport", Example: "yanic serve --config /etc/thrempp.toml", Run: func(cmd *cobra.Command, args []string) { config := &Config{} if err := file.ReadTOML(configPath, config); err != nil { log.Panicf("open config file: %s", err) } log.SetLevel(config.LogLevel) if err := database.Open(config.Database); err != nil { log.Panicf("no database connection: %s", err) } defer database.Close() server := o3.ThreemaRest{} var thrAccount models.AccountThreema if err := database.Read.First(&thrAccount).Error; err != nil { id, _ := server.CreateIdentity() thrAccount.TID = make([]byte, len(id.ID)) thrAccount.LSK = make([]byte, len(id.LSK)) copy(thrAccount.TID, id.ID[:]) copy(thrAccount.LSK, id.LSK[:]) database.Write.Create(&thrAccount) } log.Warnf("%s", thrAccount.TID) var lsk [32]byte copy(lsk[:], thrAccount.LSK[:]) tid, err := o3.NewThreemaID(string(thrAccount.TID), lsk, o3.AddressBook{}) tid.Nick = o3.NewPubNick("xmpp:geno@fireorbit.de") ctx := o3.NewSessionContext(tid) // let the session begin log.Info("Starting session") sendMsgChan, receiveMsgChan, err := ctx.Run() if err != nil { log.Fatal(err) } // handle incoming messages for receivedMessage := range receiveMsgChan { if receivedMessage.Err != nil { log.Errorf("Error Receiving Message: %s\n", receivedMessage.Err) continue } switch msg := receivedMessage.Msg.(type) { case o3.TextMessage: if tid.String() == msg.Sender().String() { continue } qoute := fmt.Sprintf("> %s: %s\n%s", msg.Sender(), msg.Text(), "Exactly!") err = ctx.SendTextMessage(msg.Sender().String(), qoute, sendMsgChan) if err != nil { log.Fatal(err) } } } }, } func init() { RootCmd.AddCommand(serveCmd) serveCmd.Flags().StringVarP(&configPath, "config", "c", "config.toml", "Path to configuration file") }