feat: Support for C programs

Author: ErlanRG

cmd/main.go

@@ -8,7 +8,8 @@ import (
 )
 
 func main() {
-	if err := pro.CheckArgs(); err != nil {
+	language, projectName, err := pro.CheckArgs()
+	if err != nil {
 		log.Fatal(err)
 	}
 
@@ -19,13 +20,14 @@ func main() {
 	}
 
 	p := pro.Project{
-		ProjectName:  os.Args[1],
 		AbsolutePath: currentWorkingDir,
+		Language:     language,
+		ProjectName:  projectName,
 	}
 
 	if err := p.CreateProjectStructure(); err != nil {
 		log.Fatalf("Error creating project structure: %v", err)
 	}
 
-    log.Printf("Project %s created successfully", p.ProjectName)
+	log.Printf("Project %s created successfully", p.ProjectName)
 }

internal/program.go

@@ -3,6 +3,8 @@ package internal
 import (
 	_ "embed"
 	"errors"
+	"flag"
+	"fmt"
 	"log"
 	"os"
 	"os/exec"
@@ -20,7 +22,7 @@ const (
 //go:embed templates/.gitignore.tmpl
 var gitignoreTmpl string
 
-//go:embed templates/main.cpp.tmpl
+//go:embed templates/main.tmpl
 var mainTmpl string
 
 //go:embed templates/Makefile.tmpl
@@ -35,6 +37,7 @@ var readmeTmpl string
 type Project struct {
 	ProjectName  string
 	AbsolutePath string
+	Language     string
 }
 
 func (p *Project) CreateProjectStructure() error {
@@ -86,15 +89,15 @@ func (p *Project) CreateProjectStructure() error {
 		return err
 	}
 
-	// Create main.cpp file
+	// Create main file
 	mainPath := filepath.Join(projectPath, srcPath)
-	if err := p.CreateFileFromTemplate(mainTmpl, mainPath, "main.cpp", nil); err != nil {
-		log.Printf("Error creating main.cpp file: %v\n", err)
+	if err := p.CreateFileFromTemplate(mainTmpl, mainPath, "main."+p.Language, map[string]string{"Language": p.Language}); err != nil {
+		log.Printf("Error creating main.%s file: %v\n", p.Language, err)
 		return err
 	}
 
 	// Create Makefile
-	if err := p.CreateFileFromTemplate(makefileTmpl, projectPath, "Makefile", nil); err != nil {
+	if err := p.CreateFileFromTemplate(makefileTmpl, projectPath, "Makefile", map[string]string{"Language": p.Language}); err != nil {
 		log.Printf("Error creating Makefile: %v\n", err)
 		return err
 	}
@@ -173,11 +176,43 @@ func ExecuteCmd(name string, args []string, dir string) error {
 	return nil
 }
 
-func CheckArgs() error {
-	// Check if there are enough arguments
-	if len(os.Args) != 2 {
-		return errors.New("Usage: bluecpprint <project_name>")
+func usage() {
+	fmt.Fprintf(os.Stderr, "Usage: bluecpprint --language=[LANGUAGE] [PROJECT_NAME]\n")
+    fmt.Fprintf(os.Stderr, "Options:")
+	fmt.Fprintf(os.Stderr, "\n  --language=[LANG]")
+	fmt.Fprintf(os.Stderr, "       Specify the programming language. Choose between 'c' for C or 'cpp' for C++. The project files will change accordingly.\n")
+}
+
+func valudateLanguage(lang string) error {
+	if lang == "" {
+		return errors.New("Language is required")
+	}
+
+	if lang != "c" && lang != "cpp" {
+		return errors.New("Invalid language. Allowed values are: c, cpp")
 	}
 
 	return nil
 }
+
+func printUsageAndError(msg string) (string, string, error) {
+	flag.Usage()
+	return "", "", errors.New(msg)
+}
+
+func CheckArgs() (string, string, error) {
+	language := flag.String("language", "", "")
+	flag.Usage = usage
+	flag.Parse()
+
+	if len(flag.Args()) < 1 {
+		return printUsageAndError("Language and Project Name are required")
+	}
+	projectName := flag.Arg(0)
+
+	if err := valudateLanguage(*language); err != nil {
+		return printUsageAndError(err.Error())
+	}
+
+	return *language, projectName, nil
+}

internal/templates/Makefile.tmpl

@@ -1,4 +1,4 @@
-CC=g++
+{{ if eq .Language "cpp" -}}CC=g++{{ else -}}CC=gcc{{ end }}
 FLAGS=-O2 -Wall -Wextra -Wfatal-errors -pedantic
 BIN_DIR=bin
 SRC_DIR=src
@@ -7,10 +7,11 @@ INCLUDE_DIR=include
 all: build
 
 build:
-	@$(CC) $(FLAGS) -I$(INCLUDE_DIR) $(SRC_DIR)/*.cpp -o $(BIN_DIR)/main
+	@if [ ! -d $(BIN_DIR) ]; then mkdir $(BIN_DIR); fi
+	@$(CC) $(FLAGS) -I$(INCLUDE_DIR) $(SRC_DIR)/*.{{.Language}} -o $(BIN_DIR)/main
 
 clean:
-	@rm -f $(BIN_DIR)/*
+	@rm -rf $(BIN_DIR)
 
 run:
 	@./$(BIN_DIR)/main

internal/templates/main.cpp.tmpl

@@ -0,0 +1,14 @@
+{{ if eq .Language "cpp" -}}
+#include <iostream>
+{{ else -}}
+#include <stdio.h>
+{{ end }}
+int main (int argc, char *argv[]) 
+{
+    {{- if eq .Language "cpp" }}
+    std::cout << "Hello World" << std::endl;
+    {{- else }}
+    printf("Hello World\n");
+    {{- end }}
+    return 0;
+}