vue-movie/server/main.go

package main

import (
	"bytes"
	"fmt"
	"io/fs"
	"log"
	"os/exec"
	"path/filepath"
	"strings"

	"github.com/gin-gonic/gin"
)

// Movie 电影结构
type Movie struct {
	Name  string `json:"filename"`
	Image string `json:"image"`
}

var movies []Movie

func initMovie() {
	var movieDict map[string]string = make(map[string]string)

	matches, err := filepath.Glob("movie/*")
	if err != nil {
		log.Println(err)
	}

	for _, filename := range matches {
		base := filepath.Base(filename)

		ext := filepath.Ext(base)
		base = base[:strings.IndexByte(base, '.')]

		if _, ok := movieDict[base]; ok {
			delete(movieDict, base)
		} else {
			movieDict[base] = filename
		}

		log.Println(base, ext)
	}

	for key, filename := range movieDict {
		width := 160
		height := 120
		log.Println(filename)
		cmd := exec.Command("ffmpeg", "-i", filename, "-vframes", "1", "-s", fmt.Sprintf("%dx%d", width, height), "movie/"+key+".png")
		var buffer bytes.Buffer
		cmd.Stdout = &buffer
		if cmd.Run() != nil {
			panic("could not generate frame")
		}
	}

	filepath.Walk("./movie", func(path string, info fs.FileInfo, err error) error {
		if !info.IsDir() && filepath.Ext(info.Name()) != ".png" {
			base := info.Name()
			log.Println(path, info.Name())
			movies = append(movies, Movie{
				Name:  info.Name(),
				Image: base[:strings.IndexByte(base, '.')] + ".png",
			})
		}
		return nil
	})

	log.Printf("%##v", movies)
}

func main() {
	initMovie()

	eg := gin.Default()
	eg.Use(Cors())

	eg.Static("/res", "movie/")

	movie := eg.Group("movie")
	movie.GET("/", MovieList)

	eg.Run(":4444")
}