commit c6d2306ab3b62a466dcbce7dab2eec18183c584d · hailey.at/peruse

+6

cmd/peruse/main.go

···

       84
       84
        
       				EnvVars:  []string{"PERUSE_NERVANA_API_KEY"},

     

       85
       85
        
       				Required: true,

     

       86
       86
        
       			},

     

       87
       87
       +
       			&cli.StringFlag{

     

       88
       88
       +
       				Name:    "relay-host",

     

       89
       89
       +
       				EnvVars: []string{"PERUSE_RELAY_HOST"},

     

       90
       90
       +
       				Value:   "wss://bsky.network",

     

       91
       91
       +
       			},

     

       87
       92
        
       		},

     

       88
       93
        
       		Action: run,

     

       89
       94
        
       	}

     
···

       114
       119
        
       		SuggestedFollowsRkey: cmd.String("suggested-follows-rkey"),

     

       115
       120
        
       		NervanaEndpoint:      cmd.String("nervana-endpoint"),

     

       116
       121
        
       		NervanaApiKey:        cmd.String("nervana-api-key"),

     

       122
       122
       +
       		RelayHost:            cmd.String("relay-host"),

     

       117
       123
        
       	})

     

       118
       124
        
       	if err != nil {

     

       119
       125
        
       		logger.Error("error creating server", "error", err)

+1 -1

peruse/consumer.go

···

       65
       65
        
       	s.logger.Info("connecting to relay", "url", u.String())

     

       66
       66
        
       

     

       67
       67
        
       	con, _, err := d.Dial(u.String(), http.Header{

     

       68
       68
       -
       		"user-agent": []string{"photocopy/0.0.0"},

     

       68
       68
       +
       		"user-agent": []string{"peruse/0.0.0"},

     

       69
       69
        
       	})

     

       70
       70
        
       	if err != nil {

     

       71
       71
        
       		return fmt.Errorf("failed to connect to relay: %w", err)

+6

peruse/peruse.go

···

       142
       142
        
       		}

     

       143
       143
        
       	}()

     

       144
       144
        
       

     

       145
       145
       +
       	go func(ctx context.Context, cancel context.CancelFunc) {

     

       146
       146
       +
       		if err := s.startConsumer(ctx, cancel); err != nil {

     

       147
       147
       +
       			s.logger.Error("error starting consumer", "error", err)

     

       148
       148
       +
       		}

     

       149
       149
       +
       	}(ctx, cancel)

     

       150
       150
       +
       

     

       145
       151
        
       	<-ctx.Done()

     

       146
       152
        
       

     

       147
       153
        
       	s.logger.Info("shutting down server...")