選択できるのは25トピックまでです。 トピックは、先頭が英数字で、英数字とダッシュ('-')を使用した35文字以内のものにしてください。

ids.go 1.1KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. package api
  2. import (
  3. "regexp"
  4. "os"
  5. "strings"
  6. "log"
  7. "net/url"
  8. )
  9. var matchChannelID = regexp.MustCompile("^([\\w\\-]|(%3[dD]))+$")
  10. func GetChannelID(chanURL string) (string, error) {
  11. if !matchChannelID.MatchString(chanURL) {
  12. // Check if youtube.com domain
  13. _url, err := url.Parse(chanURL)
  14. if err != nil || (_url.Host != "www.youtube.com" && _url.Host != "youtube.com") {
  15. log.Fatal("Not a channel ID:", chanURL)
  16. os.Exit(1)
  17. }
  18. // Check if old /user/ URL
  19. if strings.HasPrefix(_url.Path, "/user/") {
  20. // TODO Implement extraction of channel ID
  21. log.Fatal("New /channel/ link is required!\n" +
  22. "The old /user/ links do not work.")
  23. os.Exit(1)
  24. }
  25. // Remove /channel/ path
  26. channelID := strings.TrimPrefix(_url.Path, "/channel/")
  27. if len(channelID) == len(_url.Path) {
  28. // No such prefix to be removed
  29. log.Fatal("Not a channel ID:", channelID)
  30. os.Exit(1)
  31. }
  32. // Remove rest of path from channel ID
  33. slashIndex := strings.IndexRune(channelID, '/')
  34. if slashIndex != -1 {
  35. channelID = channelID[:slashIndex]
  36. }
  37. return channelID, nil
  38. } else {
  39. // It's already a channel ID
  40. return chanURL, nil
  41. }
  42. }