youtube.go

package main

import (
	"fmt"
	"github.com/bwmarrin/discordgo"
	"github.com/google/google-api-go-client/googleapi/transport"
	"github.com/rylio/ytdl"
	"google.golang.org/api/youtube/v3"
	"log"
	"net/http"
	"strconv"
	"strings"
	"time"
)

func getDuration(stringRawFull, stringRawOffset string) (stringRemain string) {
	// stringRawFull format: P1DT3H45M2S or PT3H45M2S
	// stringRawOffset format: 4325s (at seconds)

	var stringFull string
	var duration TimeDuration
	var partial time.Duration

	stringFull = strings.Replace(stringRawFull, "P", "", 1)
	stringFull = strings.Replace(stringFull, "T", "", 1)
	stringFull = strings.ToLower(stringFull)

	var secondsFull, secondsOffset int
	value := strings.Split(stringFull, "d")
	if len(value) == 2 {
		secondsFull, _ = strconv.Atoi(value[0])
		// get the days in seconds
		secondsFull = secondsFull * 86400
		// get the format 1h1m1s in seconds
		partial, _ = time.ParseDuration(value[1])
		secondsFull = secondsFull + int(partial.Seconds())
	} else {
		partial, _ = time.ParseDuration(stringFull)
		secondsFull = int(partial.Seconds())
	}

	if stringRawOffset != "" {
		value = strings.Split(stringRawOffset, "s")
		if len(value) == 2 {
			secondsOffset, _ = strconv.Atoi(value[0])
		}
	}
	// substact the time offset
	duration.Second = secondsFull - secondsOffset

	if duration.Second <= 0 {
		return "0:00"
	}

	// print the time
	t := AddTimeDuration(duration)
	if t.Day == 0 && t.Hour == 0 {
		return fmt.Sprintf("%02d:%02d", t.Minute, t.Second)
	}
	if t.Day == 0 {
		return fmt.Sprintf("%02d:%02d:%02d", t.Hour, t.Minute, t.Second)
	}
	return fmt.Sprintf("%d:%02d:%02d:%02d", t.Day, t.Hour, t.Minute, t.Second)
}

func YoutubeFind(searchString string, v *VoiceInstance, m *discordgo.MessageCreate) (song_struct PkgSong, err error) { //(url, title, time string, err error)

	client := &http.Client{
		Transport: &transport.APIKey{Key: o.YoutubeToken},
	}

	service, err := youtube.New(client)
	if err != nil {
		//log.Fatalf("Error creating new YouTube client: %v", err)
		return
	}

	var timeOffset string
	if strings.Contains(searchString, "?t=") || strings.Contains(searchString, "&feature=youtu.be&t=") {
		var split []string
		switch {
		case strings.Contains(searchString, "?t="):
			split = strings.Split(searchString, "?t=")
			break

		case strings.Contains(searchString, "&feature=youtu.be&t="):
			split = strings.Split(searchString, "&feature=youtu.be&t=")
			break
		}
		searchString = split[0]
		timeOffset = split[1]

		if !strings.ContainsAny(timeOffset, "h | m | s") {
			timeOffset = timeOffset + "s" // secons
		}
	}

	call := service.Search.List("id,snippet").Q(searchString).MaxResults(1)
	response, err := call.Do()
	if err != nil {
		//log.Fatalf("Error making search API call: %v", err)
		return
	}

	var (
		audioId, audioTitle string //, fileVideoID string
	)

	for _, item := range response.Items {
		audioId = item.Id.VideoId
		audioTitle = item.Snippet.Title
	}
	if audioId == "" {
		ChMessageSend(m.ChannelID, "Sorry, I can't found this song.")
		return
	}

	vid, err := ytdl.GetVideoInfo("https://www.youtube.com/watch?v=" + audioId)
	if err != nil {
		//ChMessageSend(textChannelID, "Sorry, nothing found for query: "+strings.Trim(searchString, " "))
		return
	}
	format := vid.Formats.Extremes(ytdl.FormatAudioBitrateKey, true)[0]
	videoURL, err := vid.GetDownloadURL(format)
	//log.Println(err)

	videos := service.Videos.List("contentDetails").Id(vid.ID)
	resp, err := videos.Do()

	duration := resp.Items[0].ContentDetails.Duration
	durationString := getDuration(duration, timeOffset)

	var videoURLString string
	if videoURL != nil {
		if timeOffset != "" {
			offset, _ := time.ParseDuration(timeOffset)
			query := videoURL.Query()
			query.Set("begin", fmt.Sprint(int64(offset/time.Millisecond)))
			videoURL.RawQuery = query.Encode()
		}
		videoURLString = videoURL.String()
	} else {
		log.Println("Video URL not found")
	}

	guildID := SearchGuild(m.ChannelID)
	member, _ := v.session.GuildMember(guildID, m.Author.ID)
	name := ""
	if member.Nick == "" {
		name = m.Author.Username
	} else {
		name = member.Nick
	}

	song := Song{
		m.ChannelID,
		name,
		m.Author.ID,
		vid.ID,
		audioTitle,
		durationString,
		videoURLString,
	}

	song_struct.data = song
	song_struct.v = v

	return
}