235 lines
7.7 KiB
Python
Executable File
235 lines
7.7 KiB
Python
Executable File
#!/usr/bin/python3
|
|
|
|
# This script is used for initializing a CLion C++ Project
|
|
|
|
import requests
|
|
import json
|
|
import argparse
|
|
import os
|
|
import subprocess
|
|
import create_git_repo as repo
|
|
import util.color_io as io
|
|
|
|
scripts_dir = "/home/brett/Documents/code/scripts"
|
|
dir_path = os.path.dirname(os.path.realpath(__file__))
|
|
if not dir_path.endswith("/"):
|
|
dir_path += "/"
|
|
github_url = "https://github.com/Tri11Paragon/"
|
|
|
|
git_ignore = """cmake-build*/
|
|
build/
|
|
out/
|
|
./cmake-build*/
|
|
./build/
|
|
./out/
|
|
"""
|
|
|
|
cmake_exec_default_text = """cmake_minimum_required(VERSION ${CMAKE_MAJOR_VERSION}.${CMAKE_MINOR_VERSION})
|
|
project(${PROJECT_NAME} VERSION 0.0.1)
|
|
|
|
option(ENABLE_ADDRSAN "Enable the address sanitizer" OFF)
|
|
option(ENABLE_UBSAN "Enable the ub sanitizer" OFF)
|
|
option(ENABLE_TSAN "Enable the thread data race sanitizer" OFF)
|
|
|
|
set(CMAKE_CXX_STANDARD ${CMAKE_LANGUAGE_VERSION})
|
|
|
|
${SUB_DIRS}
|
|
|
|
include_directories(include/)
|
|
file(GLOB_RECURSE PROJECT_BUILD_FILES "${CMAKE_CURRENT_SOURCE_DIR}/src/*.cpp")
|
|
|
|
add_executable(${PROJECT_NAME} ${PROJECT_BUILD_FILES})
|
|
|
|
target_compile_options(${PROJECT_NAME} PRIVATE -Wall -Wextra -Weverything -Wpedantic -Wno-comment)
|
|
target_link_options(${PROJECT_NAME} PRIVATE -Wall -Wextra -Weverything -Wpedantic -Wno-comment)
|
|
|
|
${LINKS}
|
|
|
|
if (${ENABLE_ADDRSAN} MATCHES ON)
|
|
target_compile_options(${PROJECT_NAME} PRIVATE -fsanitize=address)
|
|
target_link_options(${PROJECT_NAME} PRIVATE -fsanitize=address)
|
|
endif ()
|
|
|
|
if (${ENABLE_UBSAN} MATCHES ON)
|
|
target_compile_options(${PROJECT_NAME} PRIVATE -fsanitize=undefined)
|
|
target_link_options(${PROJECT_NAME} PRIVATE -fsanitize=undefined)
|
|
endif ()
|
|
|
|
if (${ENABLE_TSAN} MATCHES ON)
|
|
target_compile_options(${PROJECT_NAME} PRIVATE -fsanitize=thread)
|
|
target_link_options(${PROJECT_NAME} PRIVATE -fsanitize=thread)
|
|
endif ()
|
|
"""
|
|
|
|
cmake_lib_default_text = """cmake_minimum_required(VERSION ${CMAKE_MAJOR_VERSION}.${CMAKE_MINOR_VERSION})
|
|
project(${PROJECT_NAME})
|
|
|
|
option(ENABLE_ADDRSAN "Enable the address sanitizer" OFF)
|
|
option(ENABLE_UBSAN "Enable the ub sanitizer" OFF)
|
|
option(ENABLE_TSAN "Enable the thread data race sanitizer" OFF)
|
|
|
|
set(CMAKE_CXX_STANDARD ${CMAKE_LANGUAGE_VERSION})
|
|
|
|
${SUB_DIRS}
|
|
|
|
include_directories(include/)
|
|
file(GLOB_RECURSE PROJECT_BUILD_FILES "${CMAKE_CURRENT_SOURCE_DIR}/src/*.cpp")
|
|
|
|
add_library(${PROJECT_NAME}${CMAKE_LIBRARY_TYPE} ${PROJECT_BUILD_FILES})
|
|
|
|
target_compile_options(${PROJECT_NAME} PRIVATE -Wall -Wextra -Weverything -Wpedantic -Wno-comment)
|
|
target_link_options(${PROJECT_NAME} PRIVATE -Wall -Wextra -Weverything -Wpedantic -Wno-comment)
|
|
|
|
${LINKS}
|
|
|
|
if (${ENABLE_ADDRSAN} MATCHES ON)
|
|
target_compile_options(${PROJECT_NAME} PRIVATE -fsanitize=address)
|
|
target_link_options(${PROJECT_NAME} PRIVATE -fsanitize=address)
|
|
endif ()
|
|
|
|
if (${ENABLE_UBSAN} MATCHES ON)
|
|
target_compile_options(${PROJECT_NAME} PRIVATE -fsanitize=undefined)
|
|
target_link_options(${PROJECT_NAME} PRIVATE -fsanitize=undefined)
|
|
endif ()
|
|
|
|
if (${ENABLE_TSAN} MATCHES ON)
|
|
target_compile_options(${PROJECT_NAME} PRIVATE -fsanitize=thread)
|
|
target_link_options(${PROJECT_NAME} PRIVATE -fsanitize=thread)
|
|
endif ()
|
|
"""
|
|
|
|
default_main_file = """#include <iostream>
|
|
|
|
int main()
|
|
{
|
|
std::cout << "Hello World!" << std::endl;
|
|
}
|
|
"""
|
|
|
|
parser = argparse.ArgumentParser(prog='Bargo', description='Cargo but bad, for C++', epilog='Meow :3')
|
|
parser.add_argument("--cmake", default=None, help="Specify CMake version, defaults to using version installed on the system")
|
|
parser.add_argument("--cpp", "-p", default="17", help="C++ Version, defaults to C++17")
|
|
parser.add_argument("--no_git", default=True, action="store_false", help="Disables creating a git repo")
|
|
parser.add_argument("--create_git", "-c", nargs='?', const=True, default=False, help="Create the associated git repo", metavar="DESCRIPTION")
|
|
parser.add_argument("--no_blt", default=True, action="store_false", help="Disables init with BLT")
|
|
parser.add_argument("--lib", action="store_true", help="Create a lib instead of an exec")
|
|
parser.add_argument("--graphics", "-g", default=False, action="store_true", help="Init with graphics in mind, ie use BLT With Graphics")
|
|
|
|
parser.add_argument("action", help="Type of action to take, currently only 'init' is supported.")
|
|
parser.add_argument("name", help="Project Name")
|
|
|
|
args = parser.parse_args()
|
|
|
|
def open_process(command, print_out = True):
|
|
process = subprocess.Popen(command, stderr=subprocess.PIPE, stdout=subprocess.PIPE)
|
|
stdout, stderr = process.communicate()
|
|
exit_code = process.wait()
|
|
str_out = stdout.decode('utf8')
|
|
str_err = stderr.decode('utf8')
|
|
if print_out and len(str_out) > 0:
|
|
print(str_out, end='')
|
|
if print_out and len(str_err) > 0:
|
|
print(str_err, end='')
|
|
#print(stdout, stderr, exit_code)
|
|
return (stdout, stderr, exit_code)
|
|
|
|
def get_cmake_version():
|
|
cmake_output, _, _ = open_process(["cmake", "--version"])
|
|
|
|
cmake_lines = cmake_output.decode('utf8').splitlines()
|
|
cmake_version = str(cmake_lines[0]).split("version ")[1].split('.')
|
|
|
|
cmake_major = cmake_version[0]
|
|
cmake_minor = cmake_version[1]
|
|
|
|
return (cmake_major, cmake_minor)
|
|
|
|
def setup_dirs():
|
|
open_process(["mkdir", "include", "src", "lib"])
|
|
|
|
def setup_blt(use_git, blt_url, blt_path):
|
|
if use_git:
|
|
open_process(["git", "submodule", "add", blt_url, "lib/" + blt_path])
|
|
open_process(["git", "submodule", "update", "--remote", "--init", "--recursive"])
|
|
else:
|
|
open_process(["git", "clone", "--recursive", blt_url, "lib/" + blt_path])
|
|
|
|
def create_git_ignore():
|
|
with open(".gitignore", "w") as f:
|
|
f.write(git_ignore)
|
|
|
|
cmake_major, cmake_minor = get_cmake_version()
|
|
cpp_version = args.cpp
|
|
use_blt = args.no_blt
|
|
use_git = args.no_git
|
|
blt_url = "https://github.com/Tri11Paragon/BLT.git"
|
|
blt_path = "blt"
|
|
blt_lib = "BLT"
|
|
cmake_text = cmake_exec_default_text
|
|
project_name = args.name.replace(" ", "-").replace("_", "-")
|
|
sub_dirs = ""
|
|
links = ""
|
|
|
|
open_process(["mkdir", project_name])
|
|
wd = os.getcwd()
|
|
if not wd.endswith('/'):
|
|
wd += "/"
|
|
wd += project_name
|
|
os.chdir(wd);
|
|
|
|
if args.lib:
|
|
cmake_text = cmake_lib_default_text
|
|
|
|
if args.cmake:
|
|
version_arr = args.cmake.split(".")
|
|
if len(version_arr) < 2:
|
|
io.red_print("Must provide at least major.minor CMake version!")
|
|
exit()
|
|
cmake_major = version_arr[0]
|
|
cmake_minor = version_arr[1]
|
|
|
|
if args.graphics:
|
|
blt_url = "https://git.tpgc.me/tri11paragon/BLT-With-Graphics-Template"
|
|
blt_path = "blt-with-graphics"
|
|
blt_lib = "BLT_WITH_GRAPHICS"
|
|
|
|
open_process(["cp", scripts_dir + "/commit.py", "./"]);
|
|
|
|
setup_dirs()
|
|
|
|
if use_git:
|
|
open_process(["git", "init"])
|
|
open_process(["git", "branch", "-M", "main"])
|
|
create_git_ignore()
|
|
|
|
if use_blt:
|
|
setup_blt(use_git=use_git, blt_url=blt_url, blt_path=blt_path)
|
|
sub_dirs += "add_subdirectory(lib/" + blt_path + ")"
|
|
links += "target_link_libraries(${PROJECT_NAME} PRIVATE " + blt_lib + ")"
|
|
|
|
if args.create_git:
|
|
desc = ""
|
|
if isinstance(args.create_git, str):
|
|
desc = args.create_git
|
|
open_process(["python3", dir_path + "create_git_repo.py", "-d", desc, project_name])
|
|
if not github_url.endswith("/"):
|
|
github_url += "/"
|
|
open_process(["git", "remote", "add", "origin", github_url + project_name])
|
|
open_process(["git", "branch", "--set-upstream-to=origin/main", "main"])
|
|
|
|
cmake_text = cmake_text.replace("${SUB_DIRS}", sub_dirs)
|
|
cmake_text = cmake_text.replace("${LINKS}", links)
|
|
cmake_text = cmake_text.replace("${CMAKE_MAJOR_VERSION}", cmake_major)
|
|
cmake_text = cmake_text.replace("${CMAKE_MINOR_VERSION}", cmake_minor)
|
|
cmake_text = cmake_text.replace("${PROJECT_NAME}", project_name)
|
|
cmake_text = cmake_text.replace("${CMAKE_LANGUAGE_VERSION}", cpp_version)
|
|
|
|
with open("CMakeLists.txt", "w") as f:
|
|
f.write(cmake_text)
|
|
|
|
with open("src/main.cpp", "w") as f:
|
|
f.write(default_main_file)
|
|
|
|
print("Created " + project_name + "!")
|
|
|