basic template

cmd/github-site-gen/main.go

@@ -0,0 +1,160 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"log"
+	"os"
+	"path/filepath"
+	"strings"
+	"time"
+
+	"github.com/go-i2p/go-gh-page/pkg/generator"
+	"github.com/go-i2p/go-gh-page/pkg/git"
+	"github.com/go-i2p/go-gh-page/pkg/templates"
+)
+
+func main() {
+	// Define command-line flags
+	repoFlag := flag.String("repo", "", "GitHub repository in format 'owner/repo-name'")
+	outputFlag := flag.String("output", "./output", "Output directory for generated site")
+	branchFlag := flag.String("branch", "main", "Branch to use (default: main)")
+	workDirFlag := flag.String("workdir", "", "Working directory for cloning (default: temporary directory)")
+	githost := flag.String("githost", "github.com", "Git host (default: github.com)")
+	mainTemplateOverride := flag.String("main-template", "", "Path to custom main template")
+	docTemplateOverride := flag.String("doc-template", "", "Path to custom documentation template")
+	styleTemplateOverride := flag.String("style-template", "", "Path to custom style template")
+
+	flag.Parse()
+
+	// Validate repository flag
+	if *repoFlag == "" {
+		fmt.Println("Error: -repo flag is required (format: owner/repo-name)")
+		flag.Usage()
+		os.Exit(1)
+	}
+
+	repoParts := strings.Split(*repoFlag, "/")
+	if len(repoParts) != 2 {
+		fmt.Println("Error: -repo flag must be in format 'owner/repo-name'")
+		flag.Usage()
+		os.Exit(1)
+	}
+	// if mainTemplateOverride is not empty, check if a file exists
+	if *mainTemplateOverride != "" {
+		if _, err := os.Stat(*mainTemplateOverride); os.IsNotExist(err) {
+			fmt.Printf("Error: main template file %s does not exist\n", *mainTemplateOverride)
+			os.Exit(1)
+		} else {
+			fmt.Printf("Using custom main template: %s\n", *mainTemplateOverride)
+			// read the file in and override templates.MainTemplate
+			data, err := os.ReadFile(*mainTemplateOverride)
+			if err != nil {
+				fmt.Printf("Error: failed to read main template file %s: %v\n", *mainTemplateOverride, err)
+				os.Exit(1)
+			}
+			templates.MainTemplate = string(data)
+		}
+	}
+	// if docTemplateOverride is not empty, check if a file exists
+	if *docTemplateOverride != "" {
+		if _, err := os.Stat(*docTemplateOverride); os.IsNotExist(err) {
+			fmt.Printf("Error: doc template file %s does not exist\n", *docTemplateOverride)
+			os.Exit(1)
+		} else {
+			fmt.Printf("Using custom docs template: %s\n", *docTemplateOverride)
+			// read the file in and override templates.MainTemplate
+			data, err := os.ReadFile(*docTemplateOverride)
+			if err != nil {
+				fmt.Printf("Error: failed to read docs template file %s: %v\n", *docTemplateOverride, err)
+				os.Exit(1)
+			}
+			templates.DocTemplate = string(data)
+		}
+	}
+	// if styleTemplateOverride is not empty, check if a file exists
+	if *styleTemplateOverride != "" {
+		if _, err := os.Stat(*styleTemplateOverride); os.IsNotExist(err) {
+			fmt.Printf("Error: style template file %s does not exist\n", *styleTemplateOverride)
+			os.Exit(1)
+		} else {
+			fmt.Printf("Using custom style template: %s\n", *styleTemplateOverride)
+			// read the file in and override templates.MainTemplate
+			data, err := os.ReadFile(*styleTemplateOverride)
+			if err != nil {
+				fmt.Printf("Error: failed to read style template file %s: %v\n", *styleTemplateOverride, err)
+				os.Exit(1)
+			}
+			templates.StyleTemplate = string(data)
+		}
+	}
+
+	owner, repo := repoParts[0], repoParts[1]
+	repoURL := fmt.Sprintf("https://%s/%s/%s.git", *githost, owner, repo)
+
+	// Create output directory if it doesn't exist
+	if err := os.MkdirAll(*outputFlag, 0755); err != nil {
+		log.Fatalf("Failed to create output directory: %v", err)
+	}
+
+	// Determine working directory
+	workDir := *workDirFlag
+	if workDir == "" {
+		// Create temporary directory
+		tempDir, err := os.MkdirTemp("", "github-site-gen-*")
+		if err != nil {
+			log.Fatalf("Failed to create temporary directory: %v", err)
+		}
+		workDir = tempDir
+		defer os.RemoveAll(tempDir) // Clean up when done
+	} else {
+		// Ensure the specified work directory exists
+		if err := os.MkdirAll(workDir, 0755); err != nil {
+			log.Fatalf("Failed to create working directory: %v", err)
+		}
+	}
+
+	cloneDir := filepath.Join(workDir, repo)
+
+	// Clone the repository
+	fmt.Printf("Cloning %s/%s into %s...\n", owner, repo, cloneDir)
+	startTime := time.Now()
+	gitRepo, err := git.CloneRepository(repoURL, cloneDir, *branchFlag)
+	if err != nil {
+		log.Fatalf("Failed to clone repository: %v", err)
+	}
+	fmt.Printf("Repository cloned in %.2f seconds\n", time.Since(startTime).Seconds())
+
+	// Get repository data
+	repoData, err := git.GetRepositoryData(gitRepo, owner, repo, cloneDir)
+	if err != nil {
+		log.Fatalf("Failed to gather repository data: %v", err)
+	}
+
+	// Create generator
+	gen := generator.NewGenerator(repoData, *outputFlag)
+
+	// Generate site
+	fmt.Println("Generating static site...")
+	startGenTime := time.Now()
+	result, err := gen.GenerateSite()
+	if err != nil {
+		log.Fatalf("Failed to generate site: %v", err)
+	}
+
+	// Print summary
+	fmt.Printf("\nRepository site for %s/%s successfully generated in %.2f seconds:\n",
+		owner, repo, time.Since(startGenTime).Seconds())
+	fmt.Printf("- Main page: %s\n", filepath.Join(*outputFlag, "index.html"))
+	fmt.Printf("- Documentation pages: %d markdown files converted\n", result.DocsCount)
+
+	if result.ImagesCount > 0 {
+		fmt.Printf("- Images directory: %s/images/\n", *outputFlag)
+	}
+
+	fmt.Printf("\nSite structure:\n%s\n", result.SiteStructure)
+	fmt.Printf("\nYou can open index.html directly in your browser\n")
+	fmt.Printf("or deploy the entire directory to any static web host.\n")
+
+	fmt.Printf("\nTotal time: %.2f seconds\n", time.Since(startTime).Seconds())
+}