v/vlib/v/ast/attr.v

// 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.
module ast

import v.token

// e.g. `[unsafe]`
pub struct Attr {
pub:
	name               string // [name]
	is_string          bool   // ['name']
	is_comptime_define bool   // [if name]
	arg                string // [name: arg]
	is_string_arg      bool   // [name: 'arg']
	pos                token.Position
}

// no square brackets
pub fn (attr Attr) str() string {
	mut s := ''
	if attr.is_comptime_define {
		s += 'if '
	}
	if attr.is_string {
		s += "'$attr.name'"
	} else {
		s += attr.name
		if attr.arg.len > 0 {
			s += ': '
			if attr.is_string_arg {
				a := attr.arg.replace("'", "\\'")
				s += "'$a'"
			} else {
				s += attr.arg
			}
		}
	}
	return s
}

pub fn (attrs []Attr) contains(str string) bool {
	for a in attrs {
		if a.name == str {
			return true
		}
	}
	return false
}

pub fn (attrs []Attr) has_comptime_define() (bool, string) {
	for a in attrs {
		if a.is_comptime_define {
			return true, a.name
		}
	}
	return false, ''
}