mirror of
https://github.com/stashapp/stash.git
synced 2025-12-17 20:34:37 +03:00
214a15bc40
* Add a space after // comments
For consistency, the commentFormatting lint checker suggests a space
after each // comment block. This commit handles all the spots in
the code where that is needed.
* Rewrite documentation on functions
Use the Go idiom of commenting:
* First sentence declares the purpose.
* First word is the name being declared
The reason this style is preferred is such that grep is able to find
names the user might be interested in. Consider e.g.,
go doc -all pkg/ffmpeg | grep -i transcode
in which case a match will tell you the name of the function you are
interested in.
* Remove old code comment-blocks
There are some commented out old code blocks in the code base. These are
either 3 years old, or 2 years old. By now, I don't think their use is
going to come back any time soon, and Git will track old pieces of
deleted code anyway.
Opt for deletion.
* Reorder imports
Split stdlib imports from non-stdlib imports in files we are touching.
* Use a range over an iteration variable
Probably more go-idiomatic, and the code needed comment-fixing anyway.
* Use time.After rather than rolling our own
The idiom here is common enough that the stdlib contains a function for
it. Use the stdlib function over our own variant.
* Enable the commentFormatting linter
69 lines
1.5 KiB
Go
69 lines
1.5 KiB
Go
package ffmpeg
|
|
|
|
import (
|
|
"bytes"
|
|
"os"
|
|
|
|
"github.com/stashapp/stash/pkg/logger"
|
|
)
|
|
|
|
// detect file format from magic file number
|
|
// https://github.com/lex-r/filetype/blob/73c10ad714e3b8ecf5cd1564c882ed6d440d5c2d/matchers/video.go
|
|
|
|
func mkv(buf []byte) bool {
|
|
return len(buf) > 3 &&
|
|
buf[0] == 0x1A && buf[1] == 0x45 &&
|
|
buf[2] == 0xDF && buf[3] == 0xA3 &&
|
|
containsMatroskaSignature(buf, []byte{'m', 'a', 't', 'r', 'o', 's', 'k', 'a'})
|
|
}
|
|
|
|
func webm(buf []byte) bool {
|
|
return len(buf) > 3 &&
|
|
buf[0] == 0x1A && buf[1] == 0x45 &&
|
|
buf[2] == 0xDF && buf[3] == 0xA3 &&
|
|
containsMatroskaSignature(buf, []byte{'w', 'e', 'b', 'm'})
|
|
}
|
|
|
|
func containsMatroskaSignature(buf, subType []byte) bool {
|
|
limit := 4096
|
|
if len(buf) < limit {
|
|
limit = len(buf)
|
|
}
|
|
|
|
index := bytes.Index(buf[:limit], subType)
|
|
if index < 3 {
|
|
return false
|
|
}
|
|
|
|
return buf[index-3] == 0x42 && buf[index-2] == 0x82
|
|
}
|
|
|
|
// MagicContainer returns the container type of a file path.
|
|
// Returns the zero-value on errors or no-match. Implements mkv or
|
|
// webm only, as ffprobe can't distinguish between them and not all
|
|
// browsers support mkv
|
|
func MagicContainer(filePath string) Container {
|
|
file, err := os.Open(filePath)
|
|
if err != nil {
|
|
logger.Errorf("[magicfile] %v", err)
|
|
return ""
|
|
}
|
|
|
|
defer file.Close()
|
|
|
|
buf := make([]byte, 4096)
|
|
_, err = file.Read(buf)
|
|
if err != nil {
|
|
logger.Errorf("[magicfile] %v", err)
|
|
return ""
|
|
}
|
|
|
|
if webm(buf) {
|
|
return Webm
|
|
}
|
|
if mkv(buf) {
|
|
return Matroska
|
|
}
|
|
return ""
|
|
}
|