rtmpauth/cmd/simple.go

package cmd

import (
	"fmt"
	"net/http"
	"os"

	"code.icod.de/dalu/sux"
	"github.com/dchest/uniuri"
	"github.com/spf13/cobra"
)

var (
	ip   = "127.0.0.1"
	port = "31337"
)

// simpleCmd represents the simple command
var simpleCmd = &cobra.Command{
	Use:   "simple",
	Short: "simple password comparison",
	RunE: func(cmd *cobra.Command, args []string) error {
		var pass = os.Getenv("RTMPAUTH_PASS")
		if pass == "" {
			pass = uniuri.New()
		}
		fmt.Println("Stream password:", pass)
		fmt.Println("Stream port:", port)
		r := sux.New()
		r.POST("/auth", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			if r.FormValue("name") == pass {
				w.WriteHeader(http.StatusOK)
			} else {
				w.WriteHeader(http.StatusForbidden)
			}
		}))
		addr := fmt.Sprintf("%s:%s", ip, port)
		fmt.Println("Listening on addr:", addr)
		return http.ListenAndServe(addr, r)
	},
}

func init() {
	rootCmd.AddCommand(simpleCmd)

	// Here you will define your flags and configuration settings.

	// Cobra supports Persistent Flags which will work for this command
	// and all subcommands, e.g.:
	// simpleCmd.PersistentFlags().String("foo", "", "A help for foo")

	// Cobra supports local flags which will only run when this command
	// is called directly, e.g.:
	// simpleCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
	simpleCmd.Flags().StringVarP(&ip, "addr", "a", "127.0.0.1", "IP Address to listen on")
	simpleCmd.Flags().StringVarP(&port, "port", "p", "31337", "Port to listen on")
}