100 lines
3.7 KiB
Go
100 lines
3.7 KiB
Go
package main
|
|
|
|
import (
|
|
"slices"
|
|
"mime"
|
|
"strings"
|
|
"path/filepath"
|
|
)
|
|
|
|
type GopherTypeCharParams struct {
|
|
GophermapFiles []string
|
|
GophermapExtensions []string
|
|
PromptFileExtensions []string
|
|
}
|
|
|
|
var plainTextExtensions = []string{
|
|
".txt", ".md", ".markdown", ".rst", ".adoc", ".text", ".asc", ".utxt",
|
|
".json", ".jsonl", ".yaml", ".yml", ".toml", ".xml", ".xsd", ".xsl", ".xslt",
|
|
".ini", ".conf", ".cfg", ".cnf", ".env", ".properties", ".prefs", ".opts",
|
|
".csv", ".tsv", ".tab", ".log", ".sql", ".ddl", ".dump",
|
|
".go", ".c", ".cpp", ".h", ".hpp", ".cc", ".hh", ".cxx", ".hxx", ".h++",
|
|
".cs", ".fs", ".fsx", ".fsi", ".java", ".kt", ".kts", ".rs", ".swift",
|
|
".py", ".pyw", ".rb", ".rbw", ".pl", ".pm", ".tcl", ".lua", ".php", ".phtml",
|
|
".js", ".mjs", ".cjs", ".jsx", ".ts", ".tsx", ".vue", ".svelte", ".dart",
|
|
".sh", ".bash", ".zsh", ".fish", ".bat", ".cmd", ".ps1", ".psm1", ".psd1",
|
|
".css", ".scss", ".sass", ".less", ".styl", ".html", ".htm", ".shtml", ".xhtml",
|
|
".dockerfile", ".makefile", ".cmake", ".gradle", ".gitignore", ".gitconfig",
|
|
".gitattributes", ".editorconfig", ".npmrc", ".babelrc", ".eslintrc",
|
|
".diff", ".patch", ".latex", ".tex", ".bib", ".cls", ".sty",
|
|
".erl", ".hrl", ".ex", ".exs", ".hs", ".lhs", ".scala", ".sc", ".clj", ".cljs",
|
|
".edn", ".lisp", ".lsp", ".scm", ".ss", ".r", ".rmd", ".f", ".f90", ".f95",
|
|
".m", ".matlab", ".jl", ".v", ".vhdl", ".sv", ".proto", ".thrift",
|
|
".tpl", ".tmpl", ".twig", ".blade", ".erb", ".haml", ".pug", ".jade",
|
|
".srt", ".vtt", ".sub",
|
|
}
|
|
var areAdditionalMimeTypesRegistered = false
|
|
|
|
var archives = []string{
|
|
"application/x-archive", "application/x-cpio", "application/x-shar",
|
|
"application/x-iso9660-image", "application/x-sbx", "application/x-tar",
|
|
"application/x-brotli", "application/x-bzip2", "application/vnd.genozip",
|
|
"application/gzip", "application/lzip", "application/x-lzma",
|
|
"application/x-lzop", "application/x-snappy-framed", "application/x-xz",
|
|
"application/x-compress", "application/x-compress", "application/zstd",
|
|
"application/x-7z-compressed", "application/x-7z-compressed",
|
|
"application/x-ace-compressed", "application/x-astrotite-afa",
|
|
"application/x-alz-compressed", "application/x-freearc", "application/x-arj",
|
|
"application/x-b1", "application/vnd.ms-cab-compressed",
|
|
"application/x-cfs-compressed", "application/x-dar", "application/x-dgc-compressed",
|
|
"application/x-apple-diskimage", "application/x-gca-compressed",
|
|
"application/x-lzh", "application/x-lzx", "application/x-rar-compressed",
|
|
"application/x-stuffit", "application/x-stuffitx", "application/x-gtar",
|
|
"application/x-ms-wim", "application/x-xar", "application/zip", "application/x-zoo",
|
|
}
|
|
|
|
func (params *GopherTypeCharParams) GopherTypeChar(path string, isDir bool) string {
|
|
switch {
|
|
case strings.HasPrefix(path, "http://") || strings.HasPrefix(path, "https://"):
|
|
return "h"
|
|
case strings.HasPrefix(path, "telnet://"):
|
|
return "8"
|
|
}
|
|
|
|
|
|
extension := filepath.Ext(path)
|
|
if isDir || slices.Contains(params.GophermapFiles, filepath.Base(path)) || slices.Contains(params.GophermapExtensions, extension) {
|
|
return "1"
|
|
}
|
|
if slices.Contains(params.PromptFileExtensions, extension) {
|
|
return "7"
|
|
}
|
|
|
|
if !areAdditionalMimeTypesRegistered {
|
|
for _, ext := range plainTextExtensions {
|
|
mime.AddExtensionType(ext, "text/plain")
|
|
}
|
|
areAdditionalMimeTypesRegistered = true
|
|
}
|
|
|
|
mimeType := mime.TypeByExtension(extension)
|
|
|
|
switch {
|
|
case strings.HasPrefix(mimeType, "text/plain"):
|
|
return "0"
|
|
case slices.Contains(archives, mimeType):
|
|
return "5"
|
|
case mimeType == "image/gif":
|
|
return "g"
|
|
case strings.HasPrefix(mimeType, "image/"):
|
|
return "I"
|
|
case strings.HasPrefix(mimeType, "audio/"):
|
|
return "s"
|
|
case strings.HasPrefix(mimeType, "text/html"):
|
|
return "h"
|
|
default:
|
|
return "9"
|
|
}
|
|
}
|
|
|