2023-10-22 11:46:19 CST W43D0

examples/avio-list-dir/main.go

@@ -1,5 +1,113 @@
 package main
 
-func main() {
+import (
+	"fmt"
+	"os"
 
+	ffmpeg "github.com/qrtc/ffmpeg-dev-go"
+)
+
+func typeString(_type int32) string {
+	switch _type {
+	case ffmpeg.AVIO_ENTRY_DIRECTORY:
+		return "<DIR>"
+	case ffmpeg.AVIO_ENTRY_FILE:
+		return "<FILE>"
+	case ffmpeg.AVIO_ENTRY_BLOCK_DEVICE:
+		return "<BLOCK DEVICE>"
+	case ffmpeg.AVIO_ENTRY_CHARACTER_DEVICE:
+		return "<CHARACTER DEVICE>"
+	case ffmpeg.AVIO_ENTRY_NAMED_PIPE:
+		return "<PIPE>"
+	case ffmpeg.AVIO_ENTRY_SYMBOLIC_LINK:
+		return "<LINK>"
+	case ffmpeg.AVIO_ENTRY_SOCKET:
+		return "<SOCKET>"
+	case ffmpeg.AVIO_ENTRY_SERVER:
+		return "<SERVER>"
+	case ffmpeg.AVIO_ENTRY_SHARE:
+		return "<SHARE>"
+	case ffmpeg.AVIO_ENTRY_WORKGROUP:
+		return "<WORKGROUP>"
+	case ffmpeg.AVIO_ENTRY_UNKNOWN:
+	default:
+		break
+	}
+	return "<UNKNOWN>"
+}
+
+func listOp(inputDir string) int32 {
+	var entry *ffmpeg.AVIODirEntry
+	var ctx *ffmpeg.AVIODirContext
+	var cnt, ret int32
+	var filemode, uidAndGid string
+
+	if ret = ffmpeg.AvIOOpenDir(&ctx, inputDir, nil); ret < 0 {
+		ffmpeg.AvLog(nil, ffmpeg.AV_LOG_ERROR, "Cannot open directory: %s.\n", ffmpeg.AvErr2str(ret))
+		goto fail
+	}
+
+	for {
+		if ret = ffmpeg.AvIOReadDir(ctx, &entry); ret < 0 {
+			ffmpeg.AvLog(nil, ffmpeg.AV_LOG_ERROR, "Cannot list directory: %s.\n", ffmpeg.AvErr2str(ret))
+			goto fail
+		}
+		if entry == nil {
+			break
+		}
+		if entry.GetFilemode() == -1 {
+			filemode = "???"
+		} else {
+			filemode = fmt.Sprintf("%3d", entry.GetFilemode())
+		}
+		uidAndGid = fmt.Sprintf("%d(%d)", entry.GetUserId(), entry.GetGroupId())
+		if cnt == 0 {
+			ffmpeg.AvLog(nil, ffmpeg.AV_LOG_INFO, "%-9s %12s %30s %10s %s %16s %16s %16s\n",
+				"TYPE", "SIZE", "NAME", "UID(GID)", "UGO", "MODIFIED",
+				"ACCESSED", "STATUS_CHANGED")
+		}
+		ffmpeg.AvLog(nil, ffmpeg.AV_LOG_INFO, "%-9s %12d %30s %10s %s %d %d %d\n",
+			typeString(entry.GetType()),
+			entry.GetSize(),
+			entry.GetName(),
+			uidAndGid,
+			filemode,
+			entry.GetModificationTimestamp(),
+			entry.GetAccessTimestamp(),
+			entry.GetStatusChangeTimestamp())
+		ffmpeg.AvIOFreeDirectoryEntry(&entry)
+		cnt++
+	}
+
+fail:
+	ffmpeg.AvIOCloseDir(&ctx)
+	return ret
+}
+
+func usage(programName string) {
+	fmt.Fprintf(os.Stderr, "usage: %s input_dir\n"+
+		"API example program to show how to list files in directory "+
+		"accessed through AVIOContext.\n", programName)
+	os.Exit(1)
+}
+
+func main() {
+	var ret int32
+
+	ffmpeg.AvLogSetLevel(ffmpeg.AV_LOG_DEBUG)
+
+	if len(os.Args) < 2 {
+		usage(os.Args[0])
+		os.Exit(1)
+	}
+
+	ffmpeg.AvFormatNetworkInit()
+
+	ret = listOp(os.Args[1])
+
+	ffmpeg.AvFormatNetworkDeinit()
+
+	if ret < 0 {
+		os.Exit(1)
+	}
 }