diji.go

package main

import (
	"errors"
	"fmt"
	"io"
	"log"
	"os"
	"os/exec"
	"path/filepath"
	"strings"
)

func check(e error) {
	if e != nil {
		panic(e)
	}
}

func copy(src, dst string) (int64, error) {
	sourceFileStat, err := os.Stat(src)
	if err != nil {
		return 0, err
	}

	if !sourceFileStat.Mode().IsRegular() {
		return 0, fmt.Errorf("%s is not a regular file", src)
	}

	source, err := os.Open(src)
	if err != nil {
		return 0, err
	}
	defer source.Close()

	destination, err := os.Create(dst)
	if err != nil {
		return 0, err
	}
	defer destination.Close()
	nBytes, err := io.Copy(destination, source)
	return nBytes, err
}

func main() {
	var ver string = "1.3"
	//	var version int = 4

	argsar := os.Args[1:]
	var quickelement string = "-q"
	var buildelement string = "--buildversion"
	var addfavelement string = "-f"
	var commitelement string = "-c"
	var quickargument bool = false
	var buildargument bool = false
	var addfavargument bool = false
	var commitargument bool = false

	for i := 0; i < len(argsar); i++ {
		// checking if the array contains the given value
		if argsar[i] == quickelement {
			// changing the boolean variable
			quickargument = true
			break
		}
	}

	for i := 0; i < len(argsar); i++ {
		// checking if the array contains the given value
		if argsar[i] == buildelement {
			// changing the boolean variable
			buildargument = true
			break
		}
	}

	for i := 0; i < len(argsar); i++ {
		// checking if the array contains the given value
		if argsar[i] == addfavelement {
			// changing the boolean variable
			addfavargument = true
			break
		}
	}

	if buildargument == true {
		fmt.Print(ver)
		os.Exit(0)
	}

	for i := 0; i < len(argsar); i++ {
		// checking if the array contains the given value
		if argsar[i] == commitelement {
			// changing the boolean variable
			commitargument = true
			break
		}
	}

	var projname string

	fmt.Println("地基")
	fmt.Println("diji " + ver)
	fmt.Println("by KZacharski")
	fmt.Print("Project name: ")
	fmt.Scanln(&projname)
	var createcss bool = true
	var createjs bool = true
	var createsample bool = false
	var initgit bool = true
	var creategitignore bool = true

	var configlocation string
	var configfile string
	var exPath string
	if _, err := os.Stat("diji-config"); err == nil {
		configlocation = "diji-config"

	} else if errors.Is(err, os.ErrNotExist) {
		ex, err := os.Executable()
		if err != nil {
			panic(err)
		}
		exPath = filepath.Dir(ex)
		configlocation = exPath + "/diji-config"

	}
	configfile = configlocation + "/config.txt"
	configbytes, err := os.ReadFile(configfile)
	if err != nil {
		fmt.Print(err)
	}
	configtext := string(configbytes)

	var gitemplatelocation string

	gitemplatelocation = configlocation + "/gitignore.txt"
	gibytes, err := os.ReadFile(gitemplatelocation)
	if err != nil {
		fmt.Print(err)
	}
	gitemplatetext := string(gibytes)

	var quickmode bool = strings.Contains(configtext, "quick-mode = true")
	var addfav bool = strings.Contains(configtext, "favicon = true")
	var commitinit bool = strings.Contains(configtext, "commit = true")

	if quickargument == true {
		if quickmode == true {
			quickmode = false
		} else if quickmode == false {
			quickmode = true
		}
	}

	if addfavargument == true {
		if addfav == true {
			addfav = false
		} else if addfav == false {
			addfav = true
		}
	}

	if commitargument == true {
		if commitinit == true {
			commitinit = false
		} else if commitinit == false {
			commitinit = true
		}
	}

	var cssstr string
	var jsstr string
	var samplestr string
	var langstr string = "en"
	var gitstr string
	var gitignorestr string
	var gifiles string

	if quickmode == false {

		fmt.Print("Create a css file (y/n, default y): ")
		fmt.Scanln(&cssstr)
		fmt.Print("Create a js file (y/n, default y): ")
		fmt.Scanln(&jsstr)
		fmt.Print("Website language (default en): ")
		fmt.Scanln(&langstr)
		fmt.Print("Insert sample content (y/n, default n): ")
		fmt.Scanln(&samplestr)
		fmt.Print("Initialize a git repo (y/n, default y): ")
		fmt.Scanln(&gitstr)
		if gitstr == "n" {
			initgit = false
		}
		if initgit == true {
			fmt.Print("Create .gitignore (y/n, default y): ")
			fmt.Scanln(&gitignorestr)
			if gitignorestr == "n" {
				creategitignore = false
			}
			if creategitignore == true {
				fmt.Print("Add files/file types to .gitignore: ")
				fmt.Scanln(&gifiles)
			}

		}
		if cssstr == "n" {
			createcss = false
		}
		if jsstr == "n" {
			createjs = false
		}
		if samplestr == "y" {
			createsample = true
		}
	}

	if initgit == false {
		creategitignore = false
	}

	if err := os.Mkdir(projname, os.ModePerm); err != nil {
		log.Fatal(err)
	}
	var indname string = projname + "/index.html"
	index, err := os.Create(indname)
	if err != nil {
		log.Fatal(err)
	}

	if addfav == true {
		var assetspath string = projname + "/assets"
		if err := os.Mkdir(assetspath, os.ModePerm); err != nil {
			log.Fatal(err)
		}
		var favpath string = assetspath + "/favicon.png"
		copy(configlocation+"/defaultfav.png", favpath)
		fmt.Println(favpath + " created.")
	}

	var indexcontent1 string = `<!DOCTYPE html>
<html lang="` + langstr + `">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<meta http-equiv="X-UA-Compatible" content="ie=edge">
<title>` + projname + `</title>
`
	var csstag string = `<link rel="stylesheet" type="text/css" href="style.css">
`
	var jstag string = `<script src="script.js"></script>
`
	var indexcontent2 string = `</head>
<body>
</body>
</html>`
	var samplecontent string = `<h1>` + projname + `</h1>
<h3>Generated with diji ` + ver + `</h3>
<p>Website content</p>`
	if createsample == true {
		indexcontent2 = `</head>
<body>
` + samplecontent + `
</body>`
	}
	var favtag string = `<link rel="icon" type="image" href="./assets/favicon.png">
`
	if addfav == true {
		indexcontent1 = indexcontent1 + favtag
	}
	if createcss == true {
		indexcontent1 = indexcontent1 + csstag
	}
	if createjs == true {
		indexcontent1 = indexcontent1 + jstag
	}
	var indexcontent string = indexcontent1 + indexcontent2
	_, err2 := index.WriteString(indexcontent)
	check(err2)
	defer index.Close()
	fmt.Println(indname + " created.")

	var cssname string
	var csscontent string

	if createcss == true {
		cssname = projname + "/style.css"
		csscontent = `body {
font-family: sans-serif;
}`
		style, err := os.Create(cssname)
		if err != nil {
			log.Fatal(err)
		}
		_, err2 := style.WriteString(csscontent)
		check(err2)
		defer style.Close()
		fmt.Println(cssname + " created.")
	}

	var jsname string
	if createjs == true {
		jsname = projname + "/script.js"
		script, err := os.Create(jsname)
		if err != nil {
			log.Fatal(err)
		}
		defer script.Close()
		fmt.Println(jsname + " created.")
	}

	var giname string
	var gicontent string

	if creategitignore == true {
		giname = projname + "/.gitignore"
		gicontent = gitemplatetext + gifiles
		gignore, err := os.Create(giname)
		if err != nil {
			log.Fatal(err)
		}
		_, err2 := gignore.WriteString(gicontent)
		check(err2)
		defer gignore.Close()

		fmt.Println(giname + " created.")
	}

	if initgit == true {
		cmd := exec.Command("git", "init", projname)
		if err := cmd.Run(); err != nil {
			log.Fatal(err)
		}

		if commitinit == true {

			cmd1 := exec.Command("git", "add", ".")
			cmd1.Dir = "./" + projname
			if err := cmd1.Run(); err != nil {
				log.Fatal(err)
			}

			cmd2 := exec.Command("git", "commit", "-a", "-m", `"Initial commit"`)
			cmd2.Dir = "./" + projname
			if err := cmd2.Run(); err != nil {
				log.Fatal(err)
			}
		}
		fmt.Println("Git repo initialized.")
	}

	fmt.Println("Project created in ./" + projname + ".")
	fmt.Println("Thanks for using diji.")

}