// Copyright (c) 2019-2023 Alexander Medvednikov. All rights reserved. // Use of this source code is governed by an MIT license that can be found in the LICENSE file. module main import os // Note: this program follows a similar convention to Rust: `init` makes the // structure of the program in the _current_ directory, while `new` // makes the program structure in a _sub_ directory. Besides that, the // functionality is essentially the same. // Note: here are the currently supported invokations so far: // 1) `v init` -> create a new project in the current folder // 2) `v new abc` -> create a new project in the new folder `abc`, by default a "hello world" project. // 3) `v new abcd web` -> create a new project in the new folder `abcd`, using the vweb template. // 4) `v new abcde hello_world` -> create a new project in the new folder `abcde`, using the hello_world template. // Note: run `v cmd/tools/vcreate_test.v` after changes to this program, to avoid regressions. struct Create { mut: name string description string version string license string files []ProjectFiles } struct ProjectFiles { path string content string } fn main() { cmd := os.args[1] match cmd { 'new' { // list of models allowed project_models := ['web', 'hello_world'] if os.args.len == 4 { // validation if os.args.last() !in project_models { mut error_str := 'It is not possible create a "${os.args[os.args.len - 2]}" project.\n' error_str += 'See the list of allowed projects:\n' for model in project_models { error_str += 'v new ${os.args[os.args.len - 2]} ${model}\n' } eprintln(error_str) exit(1) } } new_project(os.args[2..]) } 'init' { init_project() } else { cerror('unknown command: ${cmd}') exit(1) } } println('Complete!') } fn new_project(args []string) { 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) } if os.is_dir(c.name) { cerror('${c.name} folder already exists') 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 := os.getenv_opt('VLICENSE') or { 'MIT' } c.license = os.input('Input your project license: (${default_license}) ') if c.license == '' { c.license = default_license } println('Initialising ...') // `v new abcde hello_world` if args.len == 2 { match os.args.last() { 'web' { c.set_web_project_files() } 'hello_world' { c.set_hello_world_project_files() } else { eprintln('${os.args.last()} model not exist') exit(1) } } } else { // `v new abc` c.set_hello_world_project_files() } // gen project based in the `Create.files` info c.create_files_and_directories() c.write_vmod(true) c.write_gitattributes(true) c.write_editorconfig(true) c.create_git_repo(c.name) } fn init_project() { mut c := Create{} c.name = check_name(os.file_name(os.getwd())) if !os.exists('v.mod') { c.description = '' c.write_vmod(false) println('Change the description of your project in `v.mod`') } if !os.exists('src/main.v') { c.files << ProjectFiles{ path: 'src/main.v' content: hello_world_content() } } c.create_files_and_directories() c.write_gitattributes(false) c.write_editorconfig(false) c.create_git_repo('.') } fn cerror(e string) { eprintln('\nerror: ${e}') } fn check_name(name string) string { if name.trim_space().len == 0 { cerror('project name cannot be empty') exit(1) } 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 "Module { name: '${c.name}' description: '${c.description}' version: '${c.version}' license: '${c.license}' dependencies: [] } " } fn hello_world_content() string { return "module main fn main() { println('Hello World!') } " } fn gen_gitignore(name string) string { return '# Binaries for programs and plugins main ${name} *.exe *.exe~ *.so *.dylib *.dll # Ignore binary output folders bin/ # Ignore common editor/system specific metadata .DS_Store .idea/ .vscode/ *.iml # ENV .env # vweb and database *.db *.js ' } fn gitattributes_content() string { return '* text=auto eol=lf *.bat eol=crlf **/*.v linguist-language=V **/*.vv linguist-language=V **/*.vsh linguist-language=V **/v.mod linguist-language=V ' } fn editorconfig_content() string { return '[*] charset = utf-8 end_of_line = lf insert_final_newline = true trim_trailing_whitespace = true [*.v] indent_style = tab indent_size = 4 ' } fn (c &Create) write_vmod(new bool) { vmod_path := if new { '${c.name}/v.mod' } else { 'v.mod' } os.write_file(vmod_path, vmod_content(c)) or { panic(err) } } fn (c &Create) write_gitattributes(new bool) { gitattributes_path := if new { '${c.name}/.gitattributes' } else { '.gitattributes' } if !new && os.exists(gitattributes_path) { return } os.write_file(gitattributes_path, gitattributes_content()) or { panic(err) } } fn (c &Create) write_editorconfig(new bool) { editorconfig_path := if new { '${c.name}/.editorconfig' } else { '.editorconfig' } if !new && os.exists(editorconfig_path) { return } os.write_file(editorconfig_path, editorconfig_content()) or { panic(err) } } fn (c &Create) create_git_repo(dir string) { // Create Git Repo and .gitignore file if !os.is_dir('${dir}/.git') { res := os.execute('git init ${dir}') if res.exit_code != 0 { cerror('Unable to create git repo') exit(4) } } gitignore_path := '${dir}/.gitignore' if !os.exists(gitignore_path) { os.write_file(gitignore_path, gen_gitignore(c.name)) or {} } } fn (mut c Create) create_files_and_directories() { for file in c.files { // get dir and convert path separator dir := file.path.split('/')#[..-1].join(os.path_separator) // create all directories, if not exist os.mkdir_all(dir) or { panic(err) } os.write_file(file.path, file.content) or { panic(err) } } } // ####################################### PROJECTS CONTENT AND PATH ####################################### fn (mut c Create) set_hello_world_project_files() { c.files << ProjectFiles{ path: '${c.name}/src/main.v' content: hello_world_content() } } fn (mut c Create) set_web_project_files() { c.files << ProjectFiles{ path: '${c.name}/src/databases/config_databases_sqlite.v' content: "module databases import db.sqlite // can change to 'db.mysql', 'db.pg' pub fn create_db_connection() !sqlite.DB { mut db := sqlite.connect('app.db')! return db } " } c.files << ProjectFiles{ path: '${c.name}/src/templates/header_component.html' content: " " } c.files << ProjectFiles{ path: '${c.name}/src/templates/products.css' content: 'h1.title { font-family: Arial, Helvetica, sans-serif; color: #3b7bbf; } div.products-table { border: 1px solid; max-width: 720px; padding: 10px; margin: 10px; }' } c.files << ProjectFiles{ path: '${c.name}/src/templates/products.html' content: " Login @css 'src/templates/products.css'
@include 'header_component.html'

Hi, \${user.username}! you are online

@for product in user.products @end
ID Name Created date
\${product.id} \${product.name} \${product.created_at}
" } c.files << ProjectFiles{ path: '${c.name}/src/auth_controllers.v' content: "module main import vweb ['/controller/auth'; post] pub fn (mut app App) controller_auth(username string, password string) vweb.Result { response := app.service_auth(username, password) or { app.set_status(400, '') return app.text('error: \${err}') } return app.json(response) } " } c.files << ProjectFiles{ path: '${c.name}/src/auth_dto.v' content: 'module main struct AuthRequestDto { username string [required] password string [required] } ' } c.files << ProjectFiles{ path: '${c.name}/src/auth_services.v' content: "module main import crypto.hmac import crypto.sha256 import crypto.bcrypt import encoding.base64 import json import databases import time struct JwtHeader { alg string typ string } struct JwtPayload { sub string // (subject) = Entity to whom the token belongs, usually the user ID; iss string // (issuer) = Token issuer; exp string // (expiration) = Timestamp of when the token will expire; iat time.Time // (issued at) = Timestamp of when the token was created; aud string // (audience) = Token recipient, represents the application that will use it. name string roles string permissions string } fn (mut app App) service_auth(username string, password string) !string { mut db := databases.create_db_connection() or { eprintln(err) panic(err) } defer { db.close() or { panic(err) } } user := sql db { select from User where username == username limit 1 } if user.username != username { return error('user not found') } if !user.active { return error('user is not active') } bcrypt.compare_hash_and_password(password.bytes(), user.password.bytes()) or { return error('Failed to auth user, \${err}') } token := make_token(user) return token } fn make_token(user User) string { secret := 'SECRET_KEY' // os.getenv('SECRET_KEY') jwt_header := JwtHeader{'HS256', 'JWT'} jwt_payload := JwtPayload{ sub: '\${user.id}' name: '\${user.username}' iat: time.now() } header := base64.url_encode(json.encode(jwt_header).bytes()) payload := base64.url_encode(json.encode(jwt_payload).bytes()) signature := base64.url_encode(hmac.new(secret.bytes(), '\${header}.\${payload}'.bytes(), sha256.sum, sha256.block_size).bytestr().bytes()) jwt := '\${header}.\${payload}.\${signature}' return jwt } fn auth_verify(token string) bool { if token == '' { return false } secret := 'SECRET_KEY' // os.getenv('SECRET_KEY') token_split := token.split('.') signature_mirror := hmac.new(secret.bytes(), '\${token_split[0]}.\${token_split[1]}'.bytes(), sha256.sum, sha256.block_size).bytestr().bytes() signature_from_token := base64.url_decode(token_split[2]) return hmac.equal(signature_from_token, signature_mirror) // return true } " } c.files << ProjectFiles{ path: '${c.name}/src/index.html' content: " \${title}
@include 'templates/header_component.html'
" } c.files << ProjectFiles{ path: '${c.name}/src/main.v' content: "module main import vweb import databases import os const ( port = 8082 ) struct App { vweb.Context } pub fn (app App) before_request() { println('[web] before_request: \${app.req.method} \${app.req.url}') } fn main() { mut db := databases.create_db_connection() or { panic(err) } sql db { create table User } or { panic('error on create table: \${err}') } db.close() or { panic(err) } mut app := &App{} app.serve_static('/favicon.ico', 'src/assets/favicon.ico') // makes all static files available. app.mount_static_folder_at(os.resource_abs_path('.'), '/') vweb.run(app, port) } pub fn (mut app App) index() vweb.Result { title := 'vweb app' return \$vweb.html() } " } c.files << ProjectFiles{ path: '${c.name}/src/product_controller.v' content: "module main import vweb import encoding.base64 import json ['/controller/products'; get] pub fn (mut app App) controller_get_all_products() vweb.Result { token := app.req.header.get_custom('token') or { '' } if !auth_verify(token) { app.set_status(401, '') return app.text('Not valid token') } jwt_payload_stringify := base64.url_decode_str(token.split('.')[1]) jwt_payload := json.decode(JwtPayload, jwt_payload_stringify) or { app.set_status(501, '') return app.text('jwt decode error') } user_id := jwt_payload.sub response := app.service_get_all_products_from(user_id.int()) or { app.set_status(400, '') return app.text('\${err}') } return app.json(response) // return app.text('response') } ['/controller/product/create'; post] pub fn (mut app App) controller_create_product(product_name string) vweb.Result { if product_name == '' { app.set_status(400, '') return app.text('product name cannot be empty') } token := app.req.header.get_custom('token') or { '' } if !auth_verify(token) { app.set_status(401, '') return app.text('Not valid token') } jwt_payload_stringify := base64.url_decode_str(token.split('.')[1]) jwt_payload := json.decode(JwtPayload, jwt_payload_stringify) or { app.set_status(501, '') return app.text('jwt decode error') } user_id := jwt_payload.sub app.service_add_product(product_name, user_id.int()) or { app.set_status(400, '') return app.text('error: \${err}') } app.set_status(201, '') return app.text('product created successfully') } " } c.files << ProjectFiles{ path: '${c.name}/src/product_entities.v' content: "module main [table: 'products'] struct Product { id int [primary; sql: serial] user_id int name string [required; sql_type: 'TEXT'] created_at string [default: 'CURRENT_TIMESTAMP'] } " } c.files << ProjectFiles{ path: '${c.name}/src/product_service.v' content: "module main import databases fn (mut app App) service_add_product(product_name string, user_id int) ! { mut db := databases.create_db_connection()! defer { db.close() or { panic(err) } } product_model := Product{ name: product_name user_id: user_id } mut insert_error := '' sql db { insert product_model into Product } or { insert_error = err.msg() } if insert_error != '' { return error(insert_error) } } fn (mut app App) service_get_all_products_from(user_id int) ?[]Product { mut db := databases.create_db_connection() or { println(err) return err } defer { db.close() or { panic(err) } } results := sql db { select from Product where user_id == user_id } return results } " } c.files << ProjectFiles{ path: '${c.name}/src/product_view_api.v' content: "module main import json import net.http pub fn get_products(token string) ![]Product { mut header := http.new_header() header.add_custom('token', token)! url := 'http://localhost:8082/controller/products' mut config := http.FetchConfig{ header: header } resp := http.fetch(http.FetchConfig{ ...config, url: url })! products := json.decode([]Product, resp.body)! return products } pub fn get_product(token string) ![]User { mut header := http.new_header() header.add_custom('token', token)! url := 'http://localhost:8082/controller/product' mut config := http.FetchConfig{ header: header } resp := http.fetch(http.FetchConfig{ ...config, url: url })! products := json.decode([]User, resp.body)! return products } " } c.files << ProjectFiles{ path: '${c.name}/src/product_view.v' content: "module main import vweb ['/products'; get] pub fn (mut app App) products() !vweb.Result { token := app.get_cookie('token') or { app.set_status(400, '') return app.text('\${err}') } user := get_user(token) or { app.set_status(400, '') return app.text('Failed to fetch data from the server. Error: \${err}') } return \$vweb.html() } " } c.files << ProjectFiles{ path: '${c.name}/src/user_controllers.v' content: "module main import vweb import encoding.base64 import json ['/controller/users'; get] pub fn (mut app App) controller_get_all_user() vweb.Result { token := app.req.header.get_custom('token') or { '' } if !auth_verify(token) { app.set_status(401, '') return app.text('Not valid token') } response := app.service_get_all_user() or { app.set_status(400, '') return app.text('\${err}') } return app.json(response) } ['/controller/user'; get] pub fn (mut app App) controller_get_user() vweb.Result { token := app.req.header.get_custom('token') or { '' } if !auth_verify(token) { app.set_status(401, '') return app.text('Not valid token') } jwt_payload_stringify := base64.url_decode_str(token.split('.')[1]) jwt_payload := json.decode(JwtPayload, jwt_payload_stringify) or { app.set_status(501, '') return app.text('jwt decode error') } user_id := jwt_payload.sub response := app.service_get_user(user_id.int()) or { app.set_status(400, '') return app.text('\${err}') } return app.json(response) } ['/controller/user/create'; post] pub fn (mut app App) controller_create_user(username string, password string) vweb.Result { if username == '' { app.set_status(400, '') return app.text('username cannot be empty') } if password == '' { app.set_status(400, '') return app.text('password cannot be empty') } app.service_add_user(username, password) or { app.set_status(400, '') return app.text('error: \${err}') } app.set_status(201, '') return app.text('User created successfully') } " } c.files << ProjectFiles{ path: '${c.name}/src/user_entities.v' content: "module main [table: 'users'] pub struct User { mut: id int [primary; sql: serial] username string [required; sql_type: 'TEXT'; unique] password string [required; sql_type: 'TEXT'] active bool products []Product [fkey: 'user_id'] } " } c.files << ProjectFiles{ path: '${c.name}/src/user_services.v' content: "module main import crypto.bcrypt import databases fn (mut app App) service_add_user(username string, password string) ! { mut db := databases.create_db_connection()! defer { db.close() or { panic(err) } } hashed_password := bcrypt.generate_from_password(password.bytes(), bcrypt.min_cost) or { eprintln(err) return err } user_model := User{ username: username password: hashed_password active: true } mut insert_error := '' sql db { insert user_model into User } or { insert_error = err.msg() } if insert_error != '' { return error(insert_error) } } fn (mut app App) service_get_all_user() ?[]User { mut db := databases.create_db_connection() or { println(err) return err } defer { db.close() or { panic(err) } } results := sql db { select from User } return results } fn (mut app App) service_get_user(id int) ?User { mut db := databases.create_db_connection() or { println(err) return err } defer { db.close() or { panic(err) } } results := sql db { select from User where id == id } return results } " } c.files << ProjectFiles{ path: '${c.name}/src/user_view_api.v' content: "module main import json import net.http pub fn get_users(token string) ![]User { mut header := http.new_header() header.add_custom('token', token)! url := 'http://localhost:8082/controller/users' mut config := http.FetchConfig{ header: header } resp := http.fetch(http.FetchConfig{ ...config, url: url })! users := json.decode([]User, resp.body)! return users } pub fn get_user(token string) !User { mut header := http.new_header() header.add_custom('token', token)! url := 'http://localhost:8082/controller/user' mut config := http.FetchConfig{ header: header } resp := http.fetch(http.FetchConfig{ ...config, url: url })! users := json.decode(User, resp.body)! return users } " } }