mirrors/quickpassthrough
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

move over to fileio for checking if file exists

Browse source
This commit is contained in:
HikariKnight 2023-04-09 17:17:04 +02:00
parent f3032966b9
commit 855f197650
1 changed files with 4 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

10
internal/ls_iommu_downloader/ls_iommu_downloader.go
View file

@ -4,7 +4,6 @@ import (
"crypto/sha256" "crypto/sha256"
"encoding/hex" "encoding/hex"
"encoding/json" "encoding/json"
"errors"
"fmt" "fmt"
"io" "io"
"log" "log"
@ -14,6 +13,7 @@ import (
"time" "time"
"github.com/HikariKnight/ls-iommu/pkg/errorcheck" "github.com/HikariKnight/ls-iommu/pkg/errorcheck"
"github.com/HikariKnight/quickpassthrough/pkg/fileio"
"github.com/HikariKnight/quickpassthrough/pkg/untar" "github.com/HikariKnight/quickpassthrough/pkg/untar"
"github.com/cavaliergopher/grab/v3" "github.com/cavaliergopher/grab/v3"
) )
@ -111,7 +111,7 @@ func CheckLsIOMMU() {
// Make the directory for ls-iommu if it does not exist // Make the directory for ls-iommu if it does not exist
path := "utils" path := "utils"
if _, err := os.Stat(path); errors.Is(err, os.ErrNotExist) { if !fileio.FileExist(path) {
err := os.Mkdir(path, os.ModePerm) err := os.Mkdir(path, os.ModePerm)
errorcheck.ErrorCheck(err) errorcheck.ErrorCheck(err)
} }
@ -139,9 +139,7 @@ func CheckLsIOMMU() {
errorcheck.ErrorCheck(err) errorcheck.ErrorCheck(err)
// Check if the tar.gz exists // Check if the tar.gz exists
_, err = os.Stat(fileName) if !fileio.FileExist(fileName) {
if errors.Is(err, os.ErrNotExist) {
downloadNewVersion(path, fileName, downloadUrl) downloadNewVersion(path, fileName, downloadUrl)
if checkSum(string(checksums_txt), fileName) { if checkSum(string(checksums_txt), fileName) {
err = untar.Untar(fmt.Sprintf("%s/", path), fileName) err = untar.Untar(fmt.Sprintf("%s/", path), fileName)
@ -184,7 +182,7 @@ func downloadNewVersion(path, fileName, downloadUrl string) {
// check for errors // check for errors
if err := download.Err(); err != nil { if err := download.Err(); err != nil {
fmt.Fprintf(os.Stderr, "Download failed: %v\n", err) fmt.Fprintf(os.Stderr, "Download failed: %v\n", err)
if _, err := os.Stat("utils/ls-iommu"); errors.Is(err, os.ErrNotExist) { if !fileio.FileExist("utils/ls-iommu") {
log.Fatal("If the above error is 404, then we could not communicate with the GitHub API\n Please manually download and extract ls-iommu to: utils/\nYou can download it from: https://github.com/HikariKnight/ls-iommu/releases") log.Fatal("If the above error is 404, then we could not communicate with the GitHub API\n Please manually download and extract ls-iommu to: utils/\nYou can download it from: https://github.com/HikariKnight/ls-iommu/releases")
} else { } else {
fmt.Println("Existing ls-iommu binary detected in \"utils/\", will use that instead as the GitHub API did not respond.") fmt.Println("Existing ls-iommu binary detected in \"utils/\", will use that instead as the GitHub API did not respond.")