142 lines
3.6 KiB
Go
142 lines
3.6 KiB
Go
/*
|
|
Copyright © 2021 Zoraiz Hassan <hzoraiz8@gmail.com>
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package aic_package
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"os/exec"
|
|
"path"
|
|
"runtime"
|
|
"strings"
|
|
|
|
imgManip "github.com/TheZoraiz/ascii-image-converter/image_manipulation"
|
|
)
|
|
|
|
func saveAsciiArt(asciiSet [][]imgManip.AsciiChar, imagePath, savePath, urlImgName string) error {
|
|
// To make sure uncolored ascii art is the one saved as .txt
|
|
saveAscii := flattenAscii(asciiSet, false)
|
|
|
|
saveFileName, err := createSaveFileName(imagePath, urlImgName, "-ascii-art.txt")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
savePathLastChar := string(savePath[len(savePath)-1])
|
|
|
|
// Check if path is closed with appropriate path separator (depending on OS)
|
|
if savePathLastChar != string(os.PathSeparator) {
|
|
savePath += string(os.PathSeparator)
|
|
}
|
|
|
|
// If path exists
|
|
if _, err := os.Stat(savePath); !os.IsNotExist(err) {
|
|
return ioutil.WriteFile(savePath+saveFileName, []byte(strings.Join(saveAscii, "\n")), 0666)
|
|
} else {
|
|
return fmt.Errorf("save path %v does not exist", savePath)
|
|
}
|
|
}
|
|
|
|
// Returns new image file name along with extension
|
|
func createSaveFileName(imagePath, urlImgName, label string) (string, error) {
|
|
if urlImgName != "" {
|
|
currExt := path.Ext(urlImgName)
|
|
newName := urlImgName[:len(urlImgName)-len(currExt)] // e.g. Grabs myImage from myImage.jpeg
|
|
|
|
return newName + label, nil
|
|
}
|
|
|
|
fileInfo, err := os.Stat(imagePath)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
currName := fileInfo.Name()
|
|
currExt := path.Ext(currName)
|
|
newName := currName[:len(currName)-len(currExt)] // e.g. Grabs myImage from myImage.jpeg
|
|
|
|
return newName + label, nil
|
|
}
|
|
|
|
// flattenAscii flattens a two-dimensional grid of ascii characters into a one dimension
|
|
// of lines of ascii
|
|
func flattenAscii(asciiSet [][]imgManip.AsciiChar, colored bool) []string {
|
|
var ascii []string
|
|
|
|
for _, line := range asciiSet {
|
|
var tempAscii []string
|
|
|
|
for i := 0; i < len(line); i++ {
|
|
if colored {
|
|
tempAscii = append(tempAscii, line[i].Colored)
|
|
} else {
|
|
tempAscii = append(tempAscii, line[i].Simple)
|
|
}
|
|
}
|
|
|
|
ascii = append(ascii, strings.Join(tempAscii, ""))
|
|
}
|
|
|
|
return ascii
|
|
}
|
|
|
|
// Returns path with the file name concatenated to it
|
|
func getFullSavePath(imageName, saveFilePath string) (string, error) {
|
|
savePathLastChar := string(saveFilePath[len(saveFilePath)-1])
|
|
|
|
// Check if path is closed with appropriate path separator (depending on OS)
|
|
if savePathLastChar != string(os.PathSeparator) {
|
|
saveFilePath += string(os.PathSeparator)
|
|
}
|
|
|
|
// If path exists
|
|
if _, err := os.Stat(saveFilePath); !os.IsNotExist(err) {
|
|
return saveFilePath + imageName, nil
|
|
} else {
|
|
return "", err
|
|
}
|
|
}
|
|
|
|
// Following is for clearing screen when showing gif
|
|
var clear map[string]func()
|
|
|
|
func init() {
|
|
clear = make(map[string]func())
|
|
clear["linux"] = func() {
|
|
cmd := exec.Command("clear")
|
|
cmd.Stdout = os.Stdout
|
|
cmd.Run()
|
|
}
|
|
clear["windows"] = func() {
|
|
cmd := exec.Command("cmd", "/c", "cls")
|
|
cmd.Stdout = os.Stdout
|
|
cmd.Run()
|
|
}
|
|
clear["darwin"] = clear["linux"]
|
|
}
|
|
|
|
func clearScreen() {
|
|
value, ok := clear[runtime.GOOS]
|
|
if ok {
|
|
value()
|
|
} else {
|
|
fmt.Println("Error: your platform is unsupported, terminal can't be cleared")
|
|
os.Exit(0)
|
|
}
|
|
}
|