v/cmd/tools/vcreate.v

187 lines
4.1 KiB
V
Raw Normal View History

// Copyright (c) 2019-2021 Alexander Medvednikov. All rights reserved.
// Use of this source code is governed by an MIT license that can be found in the LICENSE file.
2021-03-13 07:43:12 +01:00
module main
2020-03-15 11:20:12 +01:00
// This module follows a similar convention to Rust: `init` makes the
2020-04-07 19:37:15 +02:00
// structure of the program in the _current_ directory, while `new`
2020-03-15 11:20:12 +01:00
// makes the program structure in a _sub_ directory. Besides that, the
// functionality is essentially the same.
2020-04-26 08:32:05 +02:00
import os
2019-11-10 22:48:56 +01:00
struct Create {
mut:
2020-02-13 19:27:13 +01:00
name string
2019-11-10 22:48:56 +01:00
description string
version string
license string
2019-11-10 22:48:56 +01:00
}
fn cerror(e string) {
2019-11-10 22:48:56 +01:00
eprintln('\nerror: $e')
}
fn check_name(name string) string {
2021-05-24 14:17:57 +02:00
if name.trim_space().len == 0 {
cerror('project name cannot be empty')
exit(1)
}
2021-03-13 07:43:12 +01:00
if name.is_title() {
mut cname := name.to_lower()
if cname.contains(' ') {
cname = cname.replace(' ', '_')
}
eprintln('warning: the project name cannot be capitalized, the name will be changed to `$cname`')
return cname
}
if name.contains(' ') {
cname := name.replace(' ', '_')
eprintln('warning: the project name cannot contain spaces, the name will be changed to `$cname`')
return cname
}
return name
}
fn vmod_content(c Create) string {
return [
2020-02-13 19:27:13 +01:00
'Module {',
" name: '$c.name'",
" description: '$c.description'",
" version: '$c.version'",
" license: '$c.license'",
2020-02-13 19:27:13 +01:00
' dependencies: []',
'}',
'',
2020-03-15 11:20:12 +01:00
].join('\n')
2019-11-10 22:48:56 +01:00
}
2020-03-15 11:20:12 +01:00
fn main_content() string {
return [
2020-02-13 19:27:13 +01:00
'module main\n',
'fn main() {',
" println('Hello World!')",
'}',
'',
2020-03-15 11:20:12 +01:00
].join('\n')
2019-11-10 22:48:56 +01:00
}
2020-03-15 11:20:12 +01:00
fn gen_gitignore(name string) string {
return [
'# Binaries for programs and plugins',
2020-03-15 11:20:12 +01:00
'main',
'$name',
'*.exe',
'*.exe~',
2020-03-15 11:20:12 +01:00
'*.so',
'*.dylib',
'*.dll',
'',
2020-03-15 11:20:12 +01:00
].join('\n')
}
fn (c &Create) write_vmod(new bool) {
vmod_path := if new { '$c.name/v.mod' } else { 'v.mod' }
mut vmod := os.create(vmod_path) or {
cerror(err.msg)
2020-03-15 11:20:12 +01:00
exit(1)
}
vmod.write_string(vmod_content(c)) or { panic(err) }
2020-03-15 11:20:12 +01:00
vmod.close()
}
fn (c &Create) write_main(new bool) {
if !new && (os.exists('${c.name}.v') || os.exists('src/${c.name}.v')) {
return
}
main_path := if new { '$c.name/${c.name}.v' } else { '${c.name}.v' }
mut mainfile := os.create(main_path) or {
cerror(err.msg)
2020-03-15 11:20:12 +01:00
exit(2)
}
mainfile.write_string(main_content()) or { panic(err) }
mainfile.close()
2020-03-15 11:20:12 +01:00
}
fn (c &Create) create_git_repo(dir string) {
2020-03-15 11:20:12 +01:00
// Create Git Repo and .gitignore file
if !os.is_dir('$dir/.git') {
res := os.execute('git init $dir')
if res.exit_code != 0 {
2020-03-15 11:20:12 +01:00
cerror('Unable to create git repo')
2020-04-07 19:37:15 +02:00
exit(4)
2020-03-15 11:20:12 +01:00
}
if !os.exists('$dir/.gitignore') {
mut fl := os.create('$dir/.gitignore') or {
2020-03-15 11:20:12 +01:00
// We don't really need a .gitignore, it's just a nice-to-have
return
}
fl.write_string(gen_gitignore(c.name)) or { panic(err) }
2020-03-15 11:20:12 +01:00
fl.close()
}
}
}
fn create(args []string) {
2019-11-10 22:48:56 +01:00
mut c := Create{}
c.name = check_name(if args.len > 0 { args[0] } else { os.input('Input your project name: ') })
if c.name == '' {
cerror('project name cannot be empty')
exit(1)
}
if c.name.contains('-') {
cerror('"$c.name" should not contain hyphens')
exit(1)
}
2020-03-15 11:20:12 +01:00
if os.is_dir(c.name) {
cerror('$c.name folder already exists')
2020-02-13 19:27:13 +01:00
exit(3)
}
c.description = if args.len > 1 { args[1] } else { os.input('Input your project description: ') }
default_version := '0.0.0'
c.version = os.input('Input your project version: ($default_version) ')
if c.version == '' {
c.version = default_version
}
default_license := 'MIT'
c.license = os.input('Input your project license: ($default_license) ')
if c.license == '' {
c.license = default_license
}
2019-11-10 22:48:56 +01:00
println('Initialising ...')
os.mkdir(c.name) or { panic(err) }
c.write_vmod(true)
c.write_main(true)
2020-03-15 11:20:12 +01:00
c.create_git_repo(c.name)
}
fn init_project() {
2020-03-15 11:20:12 +01:00
if os.exists('v.mod') {
cerror('`v init` cannot be run on existing v modules')
exit(3)
}
mut c := Create{}
c.name = check_name(os.file_name(os.getwd()))
2020-03-15 11:20:12 +01:00
c.description = ''
c.write_vmod(false)
c.write_main(false)
c.create_git_repo('.')
2021-05-24 14:17:57 +02:00
2020-03-15 11:20:12 +01:00
println("Change your module's description in `v.mod`")
}
fn main() {
2021-03-13 07:43:12 +01:00
cmd := os.args[1]
match cmd {
'new' {
create(os.args[2..])
}
'init' {
init_project()
}
else {
cerror('unknown command: $cmd')
exit(1)
}
2020-03-15 11:20:12 +01:00
}
println('Complete!')
2019-11-10 22:48:56 +01:00
}