summaryrefslogtreecommitdiff
path: root/vala/valawithstatement.vala
blob: 52210284a0e8a844ca9f23bcf619afa8dfaf522e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
/* valawithtatement.vala
 *
 * Copyright (C) 2020 Nick Schrader
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.

 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.

 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301  USA
 *
 * Authors:
 * 	Nick Schrader <nick.schrader@mailbox.org>
 */

public class Vala.WithStatement : Block {
	/**
	 * Expression representing the type of body's dominant scope.
	 */
	public Expression expression {
		get { return _expression; }
		private set {
			_expression = value;
			_expression.parent_node = this;
		}
	}

	/**
	 * Specifies the with-variable.
	 */
	public LocalVariable? with_variable { get; private set; }

	/**
	 * The block which dominant scope is type of expression.
	 */
	public Block body {
		get { return _body; }
		private set {
			_body = value;
			_body.parent_node = this;
		}
	}

	Expression _expression;
	Block _body;

	public WithStatement (LocalVariable? variable, Expression expression, Block body, SourceReference? source_reference = null) {
		base (source_reference);
		this.with_variable = variable;
		this.expression = expression;
		this.body = body;
	}

	public override void accept (CodeVisitor visitor) {
		visitor.visit_with_statement (this);
	}

	public override void accept_children (CodeVisitor visitor) {
		if (expression.symbol_reference == with_variable) {
			expression.accept (visitor);
		}

		if (with_variable != null) {
			with_variable.accept (visitor);
		}

		body.accept (visitor);
	}

	bool is_object_or_value_type (DataType? type) {
		if (type == null) {
			return false;
		} else if (type is PointerType) {
			var pointer_type = (PointerType) type;
			return is_object_or_value_type (pointer_type.base_type) && expression is PointerIndirection;
		} else {
			return type is ObjectType || type is ValueType;
		}
	}

	public override bool check (CodeContext context) {
		if (checked) {
			return !error;
		}

		checked = true;

		if (!expression.check (context)) {
			error = true;
			return false;
		}

		if (!is_object_or_value_type (expression.value_type)) {
			error = true;
			Report.error (expression.source_reference, "with statement expects an object or basic type");
			return false;
		}

		var local_var = expression.symbol_reference as LocalVariable;
		if (with_variable != null || local_var == null) {
			if (with_variable != null) {
				local_var = with_variable;
			} else {
				local_var = new LocalVariable (expression.value_type.copy (), get_temp_name (), expression, source_reference);
			}
			body.insert_statement (0, new DeclarationStatement (local_var, source_reference));
		}
		with_variable = local_var;

		var old_symbol = context.analyzer.current_symbol;
		owner = context.analyzer.current_symbol.scope;
		context.analyzer.current_symbol = this;

		if (!body.check (context)) {
			error = true;
		}

		context.analyzer.current_symbol = old_symbol;

		return !error;
	}

	public override void emit (CodeGenerator codegen) {
		if (expression.symbol_reference == with_variable) {
			expression.emit (codegen);
		}
		body.emit (codegen);
	}

	public override void get_error_types (Collection<DataType> collection, SourceReference? source_reference = null) {
		if (source_reference == null) {
			source_reference = this.source_reference;
		}
		expression.get_error_types (collection, source_reference);
		body.get_error_types (collection, source_reference);
	}

	public override void get_defined_variables (Collection<Variable> collection) {
		if (expression.symbol_reference != with_variable) {
			collection.add (with_variable);
		}
	}
}