now filling the form with the fucking data. currently, unidentified
This commit is contained in:
benny
179
packages.go
179
packages.go
@@ -1,15 +1,32 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"crypto/sha256"
|
||||
"database/sql"
|
||||
"fmt"
|
||||
"log"
|
||||
"log/slog"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/exec"
|
||||
"encoding/hex"
|
||||
"regexp"
|
||||
"runtime"
|
||||
"strconv"
|
||||
"strings"
|
||||
"html/template"
|
||||
|
||||
_ "modernc.org/sqlite"
|
||||
)
|
||||
|
||||
type jsonStruct struct {
|
||||
Identifier string
|
||||
Version string
|
||||
Type string
|
||||
Location string
|
||||
Checksum string
|
||||
Url string
|
||||
}
|
||||
|
||||
// packages struct represents a package with its details.
|
||||
type packages struct {
|
||||
Id string
|
||||
@@ -66,8 +83,8 @@ type DownloadElements struct {
|
||||
// Printer struct is an empty struct. Its purpose is not clear from the provided code.
|
||||
type Printer struct{}
|
||||
|
||||
func getDataFromFile(path string) map[string]string {
|
||||
log.Println(regexp.MustCompile(`\.\w+$`).FindString(path))
|
||||
func getDataFromFile(path string) jsonStruct {
|
||||
slog.Debug(regexp.MustCompile(`\.\w+$`).FindString(path))
|
||||
switch extention := regexp.MustCompile(`\.\w+$`).FindString(path); extention {
|
||||
case ".exe":
|
||||
return getParamsFromExif(path)
|
||||
@@ -75,32 +92,35 @@ func getDataFromFile(path string) map[string]string {
|
||||
case ".msi":
|
||||
return getParamsFromMsi(path)
|
||||
}
|
||||
return make(map[string]string)
|
||||
return jsonStruct{}
|
||||
}
|
||||
|
||||
// getParamsFromExif extracts EXIF parameters from a file at the given path using the exiftool command.
|
||||
// It returns a map where the keys are the EXIF parameter names and the values are the corresponding EXIF parameter values.
|
||||
func getParamsFromExif(path string) map[string]string {
|
||||
func getParamsFromExif(path string) jsonStruct {
|
||||
exiftool := os.Getenv("exiftool")
|
||||
e, _ := exec.Command(exiftool, "-f", path).Output()
|
||||
return splitData(e)
|
||||
data := splitData(e)
|
||||
data["Extention"] = data["File Type Extension"]
|
||||
data["Path"] = path
|
||||
return createObject(data)
|
||||
}
|
||||
func splitData(commandOutput []byte) map[string]string {
|
||||
tags := make(map[string]string)
|
||||
spliter := regexp.MustCompile(`\r?\n`)
|
||||
crop := spliter.Split(string(commandOutput), -1)
|
||||
head := regexp.MustCompile(`\s+(:\s)?`)
|
||||
head := regexp.MustCompile(`\t|(\s+:\s?)`)
|
||||
for _, c := range crop {
|
||||
if len(c) > 1 {
|
||||
data := head.Split(c, -1)
|
||||
log.Println("My Raw Data: ", c)
|
||||
slog.Debug("My Raw Data: ", c)
|
||||
tags[data[0]] = data[1]
|
||||
}
|
||||
}
|
||||
log.Println(tags)
|
||||
slog.Debug(fmt.Sprint(tags))
|
||||
return tags
|
||||
}
|
||||
func getParamsFromMsi(path string) map[string]string {
|
||||
func getParamsFromMsi(path string) jsonStruct {
|
||||
if runtime.GOOS == "windows" {
|
||||
|
||||
// orca :=os.Getenv("orca")
|
||||
@@ -113,8 +133,143 @@ func getParamsFromMsi(path string) map[string]string {
|
||||
} else {
|
||||
msiinfo := os.Getenv("msiinfo")
|
||||
e, _ := exec.Command(msiinfo, "export", path, "Property").Output()
|
||||
return splitData(e)
|
||||
data := splitData(e)
|
||||
data["Extention"] = "msi"
|
||||
data["Path"] = path
|
||||
return createObject(data)
|
||||
}
|
||||
return make(map[string]string)
|
||||
return jsonStruct{}
|
||||
}
|
||||
func getVendor(data map[string]string) string {
|
||||
for key, val := range data {
|
||||
switch key {
|
||||
case "Author":
|
||||
return strings.ToLower(strings.Replace(val," ","",-1))
|
||||
|
||||
case "Company Name":
|
||||
return strings.ToLower(strings.Replace(val," ","",-1))
|
||||
|
||||
case "Publisher":
|
||||
return strings.ToLower(strings.Replace(val," ","",-1))
|
||||
|
||||
}
|
||||
}
|
||||
return ""
|
||||
}
|
||||
func getProgram(data map[string]string) string {
|
||||
for key, val := range data {
|
||||
switch key {
|
||||
case "Title":
|
||||
return strings.ToLower(strings.Replace(val," ","",-1))
|
||||
|
||||
case "Product Name":
|
||||
return strings.ToLower(strings.Replace(val," ","",-1))
|
||||
|
||||
case "Subject":
|
||||
return strings.ToLower(strings.Replace(val," ","",-1))
|
||||
}
|
||||
}
|
||||
return ""
|
||||
}
|
||||
func getVersion(data map[string]string) string {
|
||||
replacer:=regexp.MustCompile(`[, ]+`)
|
||||
for key, val := range data {
|
||||
switch key {
|
||||
case "Version":
|
||||
return replacer.ReplaceAllString(val,".")
|
||||
case "Product Version":
|
||||
return replacer.ReplaceAllString(val,".")
|
||||
case "ProductVersion":
|
||||
return replacer.ReplaceAllString(val,".")
|
||||
case "File Version":
|
||||
return replacer.ReplaceAllString(val,".")
|
||||
case "FileVersion":
|
||||
return replacer.ReplaceAllString(val,".")
|
||||
}
|
||||
}
|
||||
return ""
|
||||
}
|
||||
func detectProblems(data map[string]string) bool {
|
||||
exe, err := data["Extention"]
|
||||
mime, _ := data["MIME Type"]
|
||||
if !err || (exe == "exe" && !(mime == "application/octet-stream")) {
|
||||
return false
|
||||
}
|
||||
if (exe != "msi" && exe != "exe" && exe != "zip") {
|
||||
return false
|
||||
}
|
||||
return true
|
||||
}
|
||||
func getInstallerType(data map[string]string) string {
|
||||
for _, val := range data {
|
||||
if val == "msi" {
|
||||
return "msi"
|
||||
}
|
||||
if strings.Contains(strings.ToLower(val), "nullsoft") {
|
||||
return "nullsoft"
|
||||
}
|
||||
if strings.Contains(strings.ToLower(val), "inno") {
|
||||
return "inno"
|
||||
}
|
||||
}
|
||||
return "exe"
|
||||
}
|
||||
func moveToNewLocationAndName(data map[string]string) string {
|
||||
newLoc := os.Getenv("installationDirectory") + "/"
|
||||
newFileLoc := getVendor(data) + "." + getProgram(data) + ".v."
|
||||
newFileLoc += getVersion(data) + "."
|
||||
newFileLoc += data["Extention"]
|
||||
newLoc += newFileLoc
|
||||
file, _ := os.OpenFile(newLoc, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0666)
|
||||
defer file.Close()
|
||||
oldFile, _ := os.ReadFile(data["Path"])
|
||||
file.Write(oldFile)
|
||||
return newFileLoc
|
||||
}
|
||||
func createObject(object map[string]string) jsonStruct {
|
||||
if detectProblems(object) {
|
||||
prop := jsonStruct{
|
||||
Identifier: getVendor(object) + "." + getProgram(object),
|
||||
Version: getVersion(object),
|
||||
Type: getInstallerType(object),
|
||||
}
|
||||
|
||||
loc, _ := os.ReadFile(object["Directory"] + "/" + object["File Name"])
|
||||
chksum:=sha256.Sum256(loc)
|
||||
prop.Checksum = hex.EncodeToString(chksum[:])
|
||||
lc := moveToNewLocationAndName(object)
|
||||
prop.Url = os.Getenv("uri") + lc
|
||||
prop.Location = os.Getenv("installationDirectory") + "/" + lc
|
||||
newJson(prop)
|
||||
return prop
|
||||
}
|
||||
return jsonStruct{}
|
||||
}
|
||||
func newJson(object jsonStruct) {
|
||||
|
||||
sqldb, err := sql.Open("sqlite", "file:progs.db")
|
||||
if err != nil {
|
||||
slog.Error(err.Error())
|
||||
}
|
||||
sqldb.Exec("INSERT OR IGNORE INTO newapps(identifier, version, type, location, checksum, url) values (?,?,?,?,?,?)", object.Identifier, object.Version, object.Type, object.Location, object.Checksum, object.Url)
|
||||
|
||||
}
|
||||
func getJson(sqldb *sql.DB) http.Handler {
|
||||
return http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
|
||||
quer := generateSQL("newapps", []string{"identifier", "version", "type", "location", "checksum", "url"}) + " ORDER BY ROWID DESC LIMIT 1"
|
||||
data := sqldb.QueryRow(quer)
|
||||
jsonelement := jsonStruct{}
|
||||
err := data.Scan(&jsonelement.Identifier, &jsonelement.Version, &jsonelement.Type, &jsonelement.Location, &jsonelement.Checksum, &jsonelement.Url)
|
||||
if err != nil {
|
||||
slog.Error(err.Error())
|
||||
}
|
||||
temp,err:=template.ParseFiles("html/lastob.json")
|
||||
if err != nil {
|
||||
slog.Error(err.Error())
|
||||
}
|
||||
err=temp.ExecuteTemplate(res, "main", jsonelement)
|
||||
if err != nil {
|
||||
slog.Error(err.Error())
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user