Praromvik · ArnobKumarSaha · Jun 9, 2024
diff --git a/index.html b/index.html
@@ -0,0 +1,28 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>HLS Video Streaming</title>
+    <!-- Include Video.js CSS -->
+    <link href="https://vjs.zencdn.net/7.18.1/video-js.css" rel="stylesheet">
+</head>
+<body>
+<h1>HLS Video Streaming with Video.js</h1>
+<!-- Video.js Player -->
+<video id="my-video" class="video-js vjs-default-skin" controls preload="auto" width="640" height="264">
+    <source src="modified.m3u8" type="application/x-mpegURL">
+    <!-- Provide a fallback for non-HLS-compatible browsers -->
+    <p class="vjs-no-js">
+        To view this video please enable JavaScript, and consider upgrading to a web browser that
+        <a href="https://videojs.com/html5-video-support/" target="_blank">supports HTML5 video</a>
+    </p>
+</video>
+
+<!-- Include Video.js and the Video.js HLS plugin -->
+<script src="https://vjs.zencdn.net/7.18.1/video.min.js"></script>
+<script>
+    var player = videojs('my-video');
+</script>
+</body>
+</html>
diff --git a/main.go b/main.go
@@ -7,10 +7,12 @@ import (
 	"github.com/aws/aws-sdk-go/aws/session"
 	"github.com/aws/aws-sdk-go/service/s3"
 	"html/template"
+	"io"
 	"log"
 	"net/http"
 	"net/url"
 	"os"
+	"strings"
 	"time"
 )
 
@@ -20,6 +22,7 @@ var (
 	region    = "us-east-1"
 	bucket    = "arnob"
 	svc       *s3.S3
+	expiry    = time.Hour
 )
 
 func init() {
@@ -51,6 +54,7 @@ func main() {
 	const port = 8080
 	http.Handle("/", addHeaders(http.FileServer(http.Dir(videoDir))))
 	http.HandleFunc("/video", objectStorage)
+	http.HandleFunc("/test", videoHandler)
 	http.HandleFunc("/stream", stream)
 	http.Handle("/hello", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		w.Header().Set("Content-Type", "text/plain")
@@ -69,6 +73,117 @@ func addHeaders(h http.Handler) http.HandlerFunc {
 	}
 }
 
+func generatePresignedURL(key string) (string, error) {
+	req, _ := svc.GetObjectRequest(&s3.GetObjectInput{
+		Bucket: aws.String(bucket),
+		Key:    aws.String(key),
+	})
+
+	preStr, err := req.Presign(expiry)
+	if err != nil {
+		return "", fmt.Errorf("failed to sign request, %v", err)
+	}
+
+	return preStr, nil
+}
+
+func listObjects(prefix string) ([]string, error) {
+	fmt.Println("LISTing v2")
+	resp, err := svc.ListObjectsV2(&s3.ListObjectsV2Input{
+		Bucket: aws.String(bucket),
+		Prefix: aws.String(prefix),
+	})
+	if err != nil {
+		return nil, fmt.Errorf("failed to list objects v2, %v", err)
+	}
+
+	var keys []string
+	var m3u8 string
+	for _, item := range resp.Contents {
+		if strings.HasSuffix(*item.Key, ".m3u8") {
+			m3u8 = *item.Key
+		}
+		keys = append(keys, *item.Key)
+	}
+	keys = append(keys, m3u8)
+	//if keys != nil {
+	//	return keys, nil
+	//}
+	//fmt.Println("LISTing v1")
+	//res, err := svc.ListObjects(&s3.ListObjectsInput{
+	//	Bucket: aws.String(bucket),
+	//	Prefix: aws.String(prefix),
+	//})
+	//if err != nil {
+	//	return nil, fmt.Errorf("failed to list objects, %v", err)
+	//}
+	//for _, item := range res.Contents {
+	//	keys = append(keys, *item.Key)
+	//}
+
+	return keys, nil
+}
+
+func videoHandler(w http.ResponseWriter, r *http.Request) {
+	videoName := r.URL.Query().Get("name")
+	if videoName == "" {
+		http.Error(w, "Missing video name", http.StatusBadRequest)
+		return
+	}
+
+	prefix := fmt.Sprintf("praromvik/%s/", videoName) // Adjust the prefix as necessary
+	keys, err := listObjects(prefix)
+	if err != nil {
+		http.Error(w, fmt.Sprintf("failed to list objects: %v", err), http.StatusInternalServerError)
+		return
+	}
+	fmt.Println(keys)
+
+	var m3u8Content string
+	presignedURL, err := generatePresignedURL(keys[len(keys)-1])
+	if err != nil {
+		http.Error(w, fmt.Sprintf("failed to generate presigned URL: %v", err), http.StatusInternalServerError)
+		return
+	}
+
+	resp, err := http.Get(presignedURL)
+	if err != nil {
+		http.Error(w, fmt.Sprintf("failed to get m3u8 file: %v", err), http.StatusInternalServerError)
+		return
+	}
+	defer resp.Body.Close()
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		http.Error(w, fmt.Sprintf("failed to read m3u8 file: %v", err), http.StatusInternalServerError)
+		return
+	}
+	m3u8Content = string(body)
+
+	for i := 0; i < len(keys)-1; i++ {
+		key := keys[i]
+		presignedURL, err = generatePresignedURL(key)
+		if err != nil {
+			http.Error(w, fmt.Sprintf("failed to generate presigned URL: %v", err), http.StatusInternalServerError)
+			return
+		}
+
+		fileName := key[strings.LastIndex(key, "/")+1:]
+		fmt.Printf("===> %s %s \n", fileName, presignedURL)
+		//segmentName := presignedURL[strings.LastIndex(presignedURL, "/")+1:]
+		//fmt.Printf("--- fileName=%v, segmentName=%v \n", fileName, segmentName)
+		m3u8Content = strings.Replace(m3u8Content, fileName, presignedURL, -1)
+	}
+	fmt.Printf("After m3u8Content: %v\n", m3u8Content)
+
+	// Serve the modified .m3u8 content
+	w.Header().Set("Content-Type", "application/vnd.apple.mpegurl")
+	w.WriteHeader(http.StatusOK)
+	_, err = w.Write([]byte(m3u8Content))
+	if err != nil {
+		http.Error(w, fmt.Sprintf("failed to write m3u8 content: %v", err), http.StatusInternalServerError)
+	}
+}
+
 func objectStorage(w http.ResponseWriter, r *http.Request) {
 	videoName := r.URL.Query().Get("name")
 
@@ -120,3 +235,7 @@ func stream(w http.ResponseWriter, r *http.Request) {
 	tmpl := template.Must(template.New("index").Parse(text))
 	tmpl.Execute(w, nil)
 }
+
+// https://www.youtube.com/watch?v=JTIm3ChI-6w
+// https://cloud.google.com/livestream/docs/quickstarts/quickstart-hls#check_channel_streaming
+// https://aws.amazon.com/blogs/networking-and-content-delivery/secure-and-cost-effective-video-streaming-using-cloudfront-signed-urls/