rm.go 994 B

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. package cli
  2. import (
  3. "errors"
  4. "fmt"
  5. "io"
  6. "net/http"
  7. "strings"
  8. )
  9. type RmCmd struct {
  10. Token string `short:"t" help:"Bearer token (SHORTY_TOKEN)" default:"${token}"`
  11. URL string `short:"u" help:"Shorty URL (SHORTY_URL)" default:"${url}"`
  12. Path string `arg:"" help:"Redirect path"`
  13. }
  14. func (c *RmCmd) Run(ctx *Context) error {
  15. url := strings.Join([]string{c.URL, c.Path}, "")
  16. req, err := http.NewRequest(http.MethodDelete, url, nil)
  17. if err != nil {
  18. return err
  19. }
  20. if c.Token != "" {
  21. req.Header.Add("authorization", fmt.Sprintf("bearer %s", c.Token))
  22. }
  23. res, err := http.DefaultClient.Do(req)
  24. if err != nil {
  25. return err
  26. }
  27. if res.StatusCode != http.StatusOK {
  28. body, err := io.ReadAll(res.Body)
  29. if err != nil {
  30. return err
  31. }
  32. if len(body) > 0 {
  33. return errors.New(string(body))
  34. }
  35. return errors.New(res.Status)
  36. }
  37. return nil
  38. }
  39. func (c *RmCmd) Validate() error {
  40. if len(c.Path) < 1 || c.Path[0] != '/' {
  41. return errors.New("invalid <path>")
  42. }
  43. return nil
  44. }