1
0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00

comptime: add support for T is $Alias and T is $Function (#16929)

This commit is contained in:
Delyan Angelov 2023-01-10 08:49:04 +02:00 committed by GitHub
parent 6a32c81070
commit 1582db1a0a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
6 changed files with 80 additions and 1 deletions

View File

@ -126,6 +126,8 @@ pub enum ComptimeTypeKind {
array
sum_type
enum_
alias
function
}
pub struct ComptimeType {
@ -144,6 +146,8 @@ pub fn (cty ComptimeType) str() string {
.array { '\$Array' }
.sum_type { '\$Sumtype' }
.enum_ { '\$Enum' }
.alias { '\$Alias' }
.function { '\$Function' }
}
}

View File

@ -2262,6 +2262,12 @@ pub fn (t &Table) is_comptime_type(x Type, y ComptimeType) bool {
.enum_ {
return x_kind == .enum_
}
.alias {
return x_kind == .alias
}
.function {
return x_kind == .function
}
}
}

View File

@ -731,6 +731,8 @@ pub fn (mut f Fmt) expr(node_ ast.Expr) {
.float { f.write('\$Float') }
.sum_type { f.write('\$Sumtype') }
.enum_ { f.write('\$Enum') }
.alias { f.write('\$Alias') }
.function { f.write('\$Function') }
}
}
}

View File

@ -659,6 +659,8 @@ pub fn (mut f Gen) expr(node_ ast.Expr) {
.float { f.write('\$Float') }
.sum_type { f.write('\$Sumtype') }
.enum_ { f.write('\$Enum') }
.alias { f.write('\$Alias') }
.function { f.write('\$Function') }
}
}
}

View File

@ -12,7 +12,7 @@ const (
supported_comptime_calls = ['html', 'tmpl', 'env', 'embed_file', 'pkgconfig', 'compile_error',
'compile_warn']
comptime_types = ['Map', 'Array', 'Int', 'Float', 'Struct', 'Interface', 'Enum',
'Sumtype']
'Sumtype', 'Alias', 'Function']
)
pub fn (mut p Parser) parse_comptime_type() ast.ComptimeType {
@ -40,6 +40,12 @@ pub fn (mut p Parser) parse_comptime_type() ast.ComptimeType {
'Float' {
cty = .float
}
'Alias' {
cty = .alias
}
'Function' {
cty = .function
}
'Array' {
cty = .array
}

View File

@ -82,3 +82,62 @@ fn test_kind_struct() {
assert_not_struct[[]int]()
assert_not_struct[map[int]int]()
}
//
type AliasOfAbc = Abc
type AliasOfint = int
type AliasOfstring = string
fn assert_alias[T]() {
$if T is $Alias {
assert true
} $else {
assert false
}
}
fn assert_not_alias[T]() {
$if T is $Alias {
assert false
} $else {
assert true
}
}
fn test_kind_alias() {
assert_alias[AliasOfAbc]()
assert_alias[AliasOfint]()
assert_alias[AliasOfstring]()
//
assert_not_alias[int]()
assert_not_alias[f32]()
assert_not_alias[[]int]()
assert_not_alias[map[int]int]()
assert_not_alias[Abc]()
}
//
fn assert_function[T](f T) {
$if T is $Function {
assert true
} $else {
assert false
}
}
fn assert_not_function[T](f T) {
$if T is $Function {
assert false
} $else {
assert true
}
}
fn test_kind_function() {
assert_function(test_kind_function)
assert_not_function(123)
assert_function('abc'.contains)
i := 5
assert_function(i.str) // TODO: 5.str currently leads to a cgen error
}