proto_to_carbon.cpp 30 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001
  1. // Part of the Carbon Language project, under the Apache License v2.0 with LLVM
  2. // Exceptions. See /LICENSE for license information.
  3. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
  4. #include "common/fuzzing/proto_to_carbon.h"
  5. #include <string_view>
  6. #include "common/fuzzing/carbon.pb.h"
  7. #include "llvm/ADT/StringExtras.h"
  8. #include "llvm/Support/raw_ostream.h"
  9. namespace Carbon {
  10. static auto ExpressionToCarbon(const Fuzzing::Expression& expression,
  11. llvm::raw_ostream& out) -> void;
  12. static auto PatternToCarbon(const Fuzzing::Pattern& pattern,
  13. llvm::raw_ostream& out) -> void;
  14. static auto StatementToCarbon(const Fuzzing::Statement& statement,
  15. llvm::raw_ostream& out) -> void;
  16. static auto DeclarationToCarbon(const Fuzzing::Declaration& declaration,
  17. llvm::raw_ostream& out) -> void;
  18. // Produces a valid Carbon identifier, which must match the regex
  19. // `[A-Za-z_][A-Za-z0-9_]*`. In the case when `s` is generated by the
  20. // fuzzing framework, it might contain invalid/non-printable characters.
  21. static auto IdentifierToCarbon(std::string_view s, llvm::raw_ostream& out)
  22. -> void {
  23. if (s.empty()) {
  24. out << "EmptyIdentifier";
  25. } else {
  26. if (!llvm::isAlpha(s[0]) && s[0] != '_') {
  27. // Ensures that identifier starts with a valid character.
  28. out << 'x';
  29. }
  30. for (const char c : s) {
  31. if (llvm::isAlnum(c) || c == '_') {
  32. out << c;
  33. } else {
  34. out << llvm::toHex(c);
  35. }
  36. }
  37. }
  38. }
  39. static auto StringLiteralToCarbon(std::string_view s, llvm::raw_ostream& out) {
  40. out << '"';
  41. out.write_escaped(s, /*UseHexEscapes=*/true);
  42. out << '"';
  43. }
  44. static auto LibraryNameToCarbon(const Fuzzing::LibraryName& library,
  45. llvm::raw_ostream& out) -> void {
  46. IdentifierToCarbon(library.package_name(), out);
  47. // Path is optional.
  48. if (library.has_path()) {
  49. out << " library ";
  50. // library.path() is a string literal.
  51. StringLiteralToCarbon(library.path(), out);
  52. }
  53. }
  54. static auto PrefixUnaryOperatorToCarbon(std::string_view op,
  55. const Fuzzing::Expression& arg,
  56. llvm::raw_ostream& out) -> void {
  57. out << op;
  58. ExpressionToCarbon(arg, out);
  59. }
  60. static auto PostfixUnaryOperatorToCarbon(const Fuzzing::Expression& arg,
  61. std::string_view op,
  62. llvm::raw_ostream& out) -> void {
  63. ExpressionToCarbon(arg, out);
  64. out << op;
  65. }
  66. static auto BinaryOperatorToCarbon(const Fuzzing::Expression& lhs,
  67. std::string_view op,
  68. const Fuzzing::Expression& rhs,
  69. llvm::raw_ostream& out) -> void {
  70. ExpressionToCarbon(lhs, out);
  71. out << op;
  72. ExpressionToCarbon(rhs, out);
  73. }
  74. static auto OperatorToCarbon(const Fuzzing::OperatorExpression& operator_expr,
  75. llvm::raw_ostream& out) -> void {
  76. const Fuzzing::Expression& arg0 =
  77. !operator_expr.arguments().empty()
  78. ? operator_expr.arguments(0)
  79. : Fuzzing::Expression::default_instance();
  80. const Fuzzing::Expression& arg1 =
  81. operator_expr.arguments().size() > 1
  82. ? operator_expr.arguments(1)
  83. : Fuzzing::Expression::default_instance();
  84. out << "(";
  85. switch (operator_expr.op()) {
  86. case Fuzzing::OperatorExpression::UnknownOperator:
  87. // `-` is an arbitrary default to avoid getting invalid syntax.
  88. PrefixUnaryOperatorToCarbon("-", arg0, out);
  89. break;
  90. case Fuzzing::OperatorExpression::AddressOf:
  91. PrefixUnaryOperatorToCarbon("&", arg0, out);
  92. break;
  93. case Fuzzing::OperatorExpression::As:
  94. BinaryOperatorToCarbon(arg0, " as ", arg1, out);
  95. break;
  96. case Fuzzing::OperatorExpression::Deref:
  97. PrefixUnaryOperatorToCarbon("*", arg0, out);
  98. break;
  99. case Fuzzing::OperatorExpression::Mul:
  100. BinaryOperatorToCarbon(arg0, " * ", arg1, out);
  101. break;
  102. case Fuzzing::OperatorExpression::Div:
  103. BinaryOperatorToCarbon(arg0, " / ", arg1, out);
  104. break;
  105. case Fuzzing::OperatorExpression::Mod:
  106. BinaryOperatorToCarbon(arg0, " % ", arg1, out);
  107. break;
  108. case Fuzzing::OperatorExpression::Ptr:
  109. PostfixUnaryOperatorToCarbon(arg0, "*", out);
  110. break;
  111. case Fuzzing::OperatorExpression::Neg:
  112. PrefixUnaryOperatorToCarbon("-", arg0, out);
  113. break;
  114. case Fuzzing::OperatorExpression::Sub:
  115. BinaryOperatorToCarbon(arg0, " - ", arg1, out);
  116. break;
  117. case Fuzzing::OperatorExpression::Not:
  118. // Needs a space to 'unglue' from the operand.
  119. PrefixUnaryOperatorToCarbon("not ", arg0, out);
  120. break;
  121. case Fuzzing::OperatorExpression::Add:
  122. BinaryOperatorToCarbon(arg0, " + ", arg1, out);
  123. break;
  124. case Fuzzing::OperatorExpression::And:
  125. BinaryOperatorToCarbon(arg0, " and ", arg1, out);
  126. break;
  127. case Fuzzing::OperatorExpression::Eq:
  128. BinaryOperatorToCarbon(arg0, " == ", arg1, out);
  129. break;
  130. case Fuzzing::OperatorExpression::Less:
  131. BinaryOperatorToCarbon(arg0, " < ", arg1, out);
  132. break;
  133. case Fuzzing::OperatorExpression::LessEq:
  134. BinaryOperatorToCarbon(arg0, " <= ", arg1, out);
  135. break;
  136. case Fuzzing::OperatorExpression::GreaterEq:
  137. BinaryOperatorToCarbon(arg0, " >= ", arg1, out);
  138. break;
  139. case Fuzzing::OperatorExpression::Greater:
  140. BinaryOperatorToCarbon(arg0, " > ", arg1, out);
  141. break;
  142. case Fuzzing::OperatorExpression::Or:
  143. BinaryOperatorToCarbon(arg0, " or ", arg1, out);
  144. break;
  145. case Fuzzing::OperatorExpression::Complement:
  146. PrefixUnaryOperatorToCarbon("^", arg0, out);
  147. break;
  148. case Fuzzing::OperatorExpression::BitwiseAnd:
  149. BinaryOperatorToCarbon(arg0, " & ", arg1, out);
  150. break;
  151. case Fuzzing::OperatorExpression::BitwiseOr:
  152. BinaryOperatorToCarbon(arg0, " | ", arg1, out);
  153. break;
  154. case Fuzzing::OperatorExpression::BitwiseXor:
  155. BinaryOperatorToCarbon(arg0, " ^ ", arg1, out);
  156. break;
  157. case Fuzzing::OperatorExpression::BitShiftLeft:
  158. BinaryOperatorToCarbon(arg0, " << ", arg1, out);
  159. break;
  160. case Fuzzing::OperatorExpression::BitShiftRight:
  161. BinaryOperatorToCarbon(arg0, " >> ", arg1, out);
  162. break;
  163. case Fuzzing::OperatorExpression::NotEq:
  164. BinaryOperatorToCarbon(arg0, " != ", arg1, out);
  165. break;
  166. }
  167. out << ")";
  168. }
  169. static auto FieldInitializerToCarbon(const Fuzzing::FieldInitializer& field,
  170. std::string_view separator,
  171. llvm::raw_ostream& out) -> void {
  172. out << ".";
  173. IdentifierToCarbon(field.name(), out);
  174. out << " " << separator << " ";
  175. ExpressionToCarbon(field.expression(), out);
  176. }
  177. static auto TupleLiteralExpressionToCarbon(
  178. const Fuzzing::TupleLiteralExpression& tuple_literal,
  179. llvm::raw_ostream& out) -> void {
  180. out << "(";
  181. llvm::ListSeparator sep;
  182. for (const auto& field : tuple_literal.fields()) {
  183. out << sep;
  184. ExpressionToCarbon(field, out);
  185. }
  186. if (tuple_literal.fields_size() == 1) {
  187. // Adding a trailing comma so that generated source will be parsed as a
  188. // tuple expression. See docs/design/tuples.md.
  189. out << ", ";
  190. }
  191. out << ")";
  192. }
  193. static auto ExpressionToCarbon(const Fuzzing::Expression& expression,
  194. llvm::raw_ostream& out) -> void {
  195. switch (expression.kind_case()) {
  196. case Fuzzing::Expression::KIND_NOT_SET:
  197. // Arbitrary default for missing expressions to avoid invalid syntax.
  198. out << "true";
  199. break;
  200. case Fuzzing::Expression::kCall: {
  201. const auto& call = expression.call();
  202. ExpressionToCarbon(call.function(), out);
  203. if (call.argument().kind_case() == Fuzzing::Expression::kTupleLiteral) {
  204. TupleLiteralExpressionToCarbon(call.argument().tuple_literal(), out);
  205. } else {
  206. out << "(";
  207. ExpressionToCarbon(call.argument(), out);
  208. out << ")";
  209. }
  210. break;
  211. }
  212. case Fuzzing::Expression::kFunctionType: {
  213. const auto& fun_type = expression.function_type();
  214. out << "__Fn";
  215. TupleLiteralExpressionToCarbon(fun_type.parameter(), out);
  216. out << " -> ";
  217. ExpressionToCarbon(fun_type.return_type(), out);
  218. break;
  219. }
  220. case Fuzzing::Expression::kSimpleMemberAccess: {
  221. const auto& simple_member_access = expression.simple_member_access();
  222. ExpressionToCarbon(simple_member_access.object(), out);
  223. out << ".";
  224. IdentifierToCarbon(simple_member_access.field(), out);
  225. break;
  226. }
  227. case Fuzzing::Expression::kCompoundMemberAccess: {
  228. const auto& simple_member_access = expression.compound_member_access();
  229. ExpressionToCarbon(simple_member_access.object(), out);
  230. out << ".(";
  231. ExpressionToCarbon(simple_member_access.path(), out);
  232. out << ")";
  233. break;
  234. }
  235. case Fuzzing::Expression::kIndex: {
  236. const auto& index = expression.index();
  237. ExpressionToCarbon(index.object(), out);
  238. out << "[";
  239. ExpressionToCarbon(index.offset(), out);
  240. out << "]";
  241. break;
  242. }
  243. case Fuzzing::Expression::kOperator:
  244. OperatorToCarbon(expression.operator_(), out);
  245. break;
  246. case Fuzzing::Expression::kTupleLiteral: {
  247. TupleLiteralExpressionToCarbon(expression.tuple_literal(), out);
  248. break;
  249. }
  250. case Fuzzing::Expression::kStructLiteral: {
  251. const auto& struct_literal = expression.struct_literal();
  252. out << "{";
  253. llvm::ListSeparator sep;
  254. for (const auto& field : struct_literal.fields()) {
  255. out << sep;
  256. FieldInitializerToCarbon(field, "=", out);
  257. }
  258. out << "}";
  259. break;
  260. }
  261. case Fuzzing::Expression::kStructTypeLiteral: {
  262. const auto& struct_type_literal = expression.struct_type_literal();
  263. out << "{";
  264. llvm::ListSeparator sep;
  265. for (const auto& field : struct_type_literal.fields()) {
  266. out << sep;
  267. FieldInitializerToCarbon(field, ":", out);
  268. }
  269. out << "}";
  270. break;
  271. }
  272. case Fuzzing::Expression::kIdentifier: {
  273. const auto& identifier = expression.identifier();
  274. IdentifierToCarbon(identifier.name(), out);
  275. break;
  276. }
  277. case Fuzzing::Expression::kDesignator: {
  278. const auto& designator = expression.designator();
  279. out << ".";
  280. IdentifierToCarbon(designator.name(), out);
  281. break;
  282. }
  283. case Fuzzing::Expression::kIfExpression: {
  284. const auto& if_expression = expression.if_expression();
  285. out << "if ";
  286. ExpressionToCarbon(if_expression.condition(), out);
  287. out << " then ";
  288. ExpressionToCarbon(if_expression.then_expression(), out);
  289. out << " else ";
  290. ExpressionToCarbon(if_expression.else_expression(), out);
  291. break;
  292. }
  293. case Fuzzing::Expression::kBoolTypeLiteral:
  294. out << "bool";
  295. break;
  296. case Fuzzing::Expression::kBoolLiteral: {
  297. const auto& bool_literal = expression.bool_literal();
  298. out << (bool_literal.value() ? "true" : "false");
  299. break;
  300. }
  301. case Fuzzing::Expression::kIntTypeLiteral:
  302. out << "i32";
  303. break;
  304. case Fuzzing::Expression::kIntLiteral: {
  305. out << expression.int_literal().value();
  306. break;
  307. }
  308. case Fuzzing::Expression::kStringLiteral:
  309. StringLiteralToCarbon(expression.string_literal().value(), out);
  310. break;
  311. case Fuzzing::Expression::kStringTypeLiteral:
  312. out << "String";
  313. break;
  314. case Fuzzing::Expression::kContinuationTypeLiteral:
  315. out << "__Continuation";
  316. break;
  317. case Fuzzing::Expression::kTypeTypeLiteral:
  318. out << "type";
  319. break;
  320. case Fuzzing::Expression::kUnimplementedExpression:
  321. // Not really supported.
  322. // This is an arbitrary default to avoid getting invalid syntax.
  323. out << "1 __unimplemented_example_infix 2";
  324. break;
  325. case Fuzzing::Expression::kArrayTypeLiteral: {
  326. const Fuzzing::ArrayTypeLiteral& array_literal =
  327. expression.array_type_literal();
  328. out << "[";
  329. ExpressionToCarbon(array_literal.element_type(), out);
  330. out << "; ";
  331. ExpressionToCarbon(array_literal.size(), out);
  332. out << "]";
  333. break;
  334. }
  335. case Fuzzing::Expression::kWhere: {
  336. const Fuzzing::WhereExpression& where = expression.where();
  337. ExpressionToCarbon(where.base(), out);
  338. out << " where ";
  339. llvm::ListSeparator sep(" and ");
  340. for (const auto& clause : where.clauses()) {
  341. out << sep;
  342. switch (clause.kind_case()) {
  343. case Fuzzing::WhereClause::kIs:
  344. ExpressionToCarbon(clause.is().type(), out);
  345. out << " is ";
  346. ExpressionToCarbon(clause.is().constraint(), out);
  347. break;
  348. case Fuzzing::WhereClause::kEquals:
  349. ExpressionToCarbon(clause.equals().lhs(), out);
  350. out << " == ";
  351. ExpressionToCarbon(clause.equals().rhs(), out);
  352. break;
  353. case Fuzzing::WhereClause::kRewrite:
  354. out << "." << clause.rewrite().member_name() << " = ";
  355. ExpressionToCarbon(clause.rewrite().replacement(), out);
  356. break;
  357. case Fuzzing::WhereClause::KIND_NOT_SET:
  358. // Arbitrary default to avoid invalid syntax.
  359. out << ".Self == .Self";
  360. break;
  361. }
  362. }
  363. break;
  364. }
  365. }
  366. }
  367. static auto BindingPatternToCarbon(const Fuzzing::BindingPattern& pattern,
  368. llvm::raw_ostream& out) -> void {
  369. IdentifierToCarbon(pattern.name(), out);
  370. out << ": ";
  371. PatternToCarbon(pattern.type(), out);
  372. }
  373. static auto GenericBindingToCarbon(
  374. const Fuzzing::GenericBinding& generic_binding, llvm::raw_ostream& out) {
  375. IdentifierToCarbon(generic_binding.name(), out);
  376. out << ":! ";
  377. ExpressionToCarbon(generic_binding.type(), out);
  378. }
  379. static auto TuplePatternToCarbon(const Fuzzing::TuplePattern& tuple_pattern,
  380. llvm::raw_ostream& out) -> void {
  381. out << "(";
  382. llvm::ListSeparator sep;
  383. for (const auto& field : tuple_pattern.fields()) {
  384. out << sep;
  385. PatternToCarbon(field, out);
  386. }
  387. if (tuple_pattern.fields_size() == 1) {
  388. // Adding a trailing comma so that generated source will be parsed as a
  389. // tuple pattern expression. See docs/design/tuples.md.
  390. out << ", ";
  391. }
  392. out << ")";
  393. }
  394. static auto PatternToCarbon(const Fuzzing::Pattern& pattern,
  395. llvm::raw_ostream& out) -> void {
  396. switch (pattern.kind_case()) {
  397. case Fuzzing::Pattern::KIND_NOT_SET:
  398. // Arbitrary default to avoid getting invalid syntax.
  399. out << "auto";
  400. break;
  401. case Fuzzing::Pattern::kBindingPattern:
  402. BindingPatternToCarbon(pattern.binding_pattern(), out);
  403. break;
  404. case Fuzzing::Pattern::kTuplePattern:
  405. TuplePatternToCarbon(pattern.tuple_pattern(), out);
  406. break;
  407. case Fuzzing::Pattern::kAlternativePattern: {
  408. const auto& alternative_pattern = pattern.alternative_pattern();
  409. ExpressionToCarbon(alternative_pattern.choice_type(), out);
  410. out << ".";
  411. IdentifierToCarbon(alternative_pattern.alternative_name(), out);
  412. TuplePatternToCarbon(alternative_pattern.arguments(), out);
  413. break;
  414. }
  415. // Arbitrary expression.
  416. case Fuzzing::Pattern::kExpressionPattern: {
  417. const auto& expression_pattern = pattern.expression_pattern();
  418. ExpressionToCarbon(expression_pattern.expression(), out);
  419. break;
  420. }
  421. case Fuzzing::Pattern::kAutoPattern:
  422. out << "auto";
  423. break;
  424. case Fuzzing::Pattern::kVarPattern:
  425. out << "var ";
  426. PatternToCarbon(pattern.var_pattern().pattern(), out);
  427. break;
  428. case Fuzzing::Pattern::kGenericBinding:
  429. GenericBindingToCarbon(pattern.generic_binding(), out);
  430. break;
  431. case Fuzzing::Pattern::kAddrPattern:
  432. out << "addr ";
  433. BindingPatternToCarbon(pattern.addr_pattern().binding_pattern(), out);
  434. break;
  435. }
  436. }
  437. static auto BlockStatementToCarbon(const Fuzzing::BlockStatement& block,
  438. llvm::raw_ostream& out) -> void {
  439. out << "{\n";
  440. for (const auto& statement : block.statements()) {
  441. StatementToCarbon(statement, out);
  442. out << "\n";
  443. }
  444. out << "}\n";
  445. }
  446. static auto StatementToCarbon(const Fuzzing::Statement& statement,
  447. llvm::raw_ostream& out) -> void {
  448. switch (statement.kind_case()) {
  449. case Fuzzing::Statement::KIND_NOT_SET:
  450. // Arbitrary default to avoid getting invalid syntax.
  451. out << "true;\n";
  452. break;
  453. case Fuzzing::Statement::kExpressionStatement: {
  454. const auto& expression_statement = statement.expression_statement();
  455. ExpressionToCarbon(expression_statement.expression(), out);
  456. out << ";";
  457. break;
  458. }
  459. case Fuzzing::Statement::kAssign: {
  460. const auto& assign_statement = statement.assign();
  461. ExpressionToCarbon(assign_statement.lhs(), out);
  462. switch (assign_statement.op()) {
  463. case Fuzzing::AssignStatement::Plain:
  464. out << " = ";
  465. break;
  466. case Fuzzing::AssignStatement::Add:
  467. out << " += ";
  468. break;
  469. case Fuzzing::AssignStatement::And:
  470. out << " &= ";
  471. break;
  472. case Fuzzing::AssignStatement::Div:
  473. out << " /= ";
  474. break;
  475. case Fuzzing::AssignStatement::Mod:
  476. out << " %= ";
  477. break;
  478. case Fuzzing::AssignStatement::Mul:
  479. out << " *= ";
  480. break;
  481. case Fuzzing::AssignStatement::Or:
  482. out << " |= ";
  483. break;
  484. case Fuzzing::AssignStatement::ShiftLeft:
  485. out << " <<= ";
  486. break;
  487. case Fuzzing::AssignStatement::ShiftRight:
  488. out << " >>= ";
  489. break;
  490. case Fuzzing::AssignStatement::Sub:
  491. out << " -= ";
  492. break;
  493. case Fuzzing::AssignStatement::Xor:
  494. out << " ^= ";
  495. break;
  496. }
  497. ExpressionToCarbon(assign_statement.rhs(), out);
  498. out << ";";
  499. break;
  500. }
  501. case Fuzzing::Statement::kIncDec: {
  502. const auto& inc_dec_statement = statement.inc_dec();
  503. out << (inc_dec_statement.is_increment() ? "++" : "--");
  504. ExpressionToCarbon(inc_dec_statement.operand(), out);
  505. out << ";";
  506. break;
  507. }
  508. case Fuzzing::Statement::kVariableDefinition: {
  509. const auto& def = statement.variable_definition();
  510. if (def.is_returned()) {
  511. out << "returned ";
  512. }
  513. out << "var ";
  514. PatternToCarbon(def.pattern(), out);
  515. if (def.has_init()) {
  516. out << " = ";
  517. ExpressionToCarbon(def.init(), out);
  518. }
  519. out << ";";
  520. break;
  521. }
  522. case Fuzzing::Statement::kIfStatement: {
  523. const auto& if_statement = statement.if_statement();
  524. out << "if (";
  525. ExpressionToCarbon(if_statement.condition(), out);
  526. out << ") ";
  527. BlockStatementToCarbon(if_statement.then_block(), out);
  528. // `else` is optional.
  529. if (if_statement.has_else_block()) {
  530. out << " else ";
  531. BlockStatementToCarbon(if_statement.else_block(), out);
  532. }
  533. break;
  534. }
  535. case Fuzzing::Statement::kReturnVarStatement: {
  536. out << "return var;";
  537. break;
  538. }
  539. case Fuzzing::Statement::kReturnExpressionStatement: {
  540. const auto& ret = statement.return_expression_statement();
  541. out << "return";
  542. if (!ret.is_omitted_expression()) {
  543. out << " ";
  544. ExpressionToCarbon(ret.expression(), out);
  545. }
  546. out << ";";
  547. break;
  548. }
  549. case Fuzzing::Statement::kBlock:
  550. BlockStatementToCarbon(statement.block(), out);
  551. break;
  552. case Fuzzing::Statement::kWhileStatement: {
  553. const auto& while_statement = statement.while_statement();
  554. out << "while (";
  555. ExpressionToCarbon(while_statement.condition(), out);
  556. out << ") ";
  557. BlockStatementToCarbon(while_statement.body(), out);
  558. break;
  559. }
  560. case Fuzzing::Statement::kForStatement: {
  561. const auto& for_statement = statement.for_statement();
  562. out << "for (";
  563. BindingPatternToCarbon(for_statement.var_decl(), out);
  564. out << " in ";
  565. ExpressionToCarbon(for_statement.target(), out);
  566. out << ") ";
  567. BlockStatementToCarbon(for_statement.body(), out);
  568. break;
  569. }
  570. case Fuzzing::Statement::kMatch: {
  571. const auto& match = statement.match();
  572. out << "match (";
  573. ExpressionToCarbon(match.expression(), out);
  574. out << ") {";
  575. for (const auto& clause : match.clauses()) {
  576. if (clause.is_default()) {
  577. out << "default";
  578. } else {
  579. out << "case ";
  580. PatternToCarbon(clause.pattern(), out);
  581. }
  582. out << " => ";
  583. StatementToCarbon(clause.statement(), out);
  584. }
  585. out << "}";
  586. break;
  587. }
  588. case Fuzzing::Statement::kContinuation: {
  589. const auto& continuation = statement.continuation();
  590. out << "__continuation ";
  591. IdentifierToCarbon(continuation.name(), out);
  592. BlockStatementToCarbon(continuation.body(), out);
  593. break;
  594. }
  595. case Fuzzing::Statement::kRun: {
  596. const auto& run = statement.run();
  597. out << "__run ";
  598. ExpressionToCarbon(run.argument(), out);
  599. out << ";";
  600. break;
  601. }
  602. case Fuzzing::Statement::kAwaitStatement:
  603. out << "__await;";
  604. break;
  605. case Fuzzing::Statement::kBreakStatement:
  606. out << "break;";
  607. break;
  608. case Fuzzing::Statement::kContinueStatement:
  609. out << "continue;";
  610. break;
  611. }
  612. }
  613. static auto ReturnTermToCarbon(const Fuzzing::ReturnTerm& return_term,
  614. llvm::raw_ostream& out) -> void {
  615. switch (return_term.kind()) {
  616. case Fuzzing::ReturnTerm::UnknownReturnKind:
  617. case Fuzzing::ReturnTerm::Omitted:
  618. break;
  619. case Fuzzing::ReturnTerm::Auto:
  620. out << " -> auto";
  621. break;
  622. case Fuzzing::ReturnTerm::Expression:
  623. out << " -> ";
  624. ExpressionToCarbon(return_term.type(), out);
  625. break;
  626. }
  627. }
  628. static auto DeclaredNameToCarbon(const Fuzzing::DeclaredName& name,
  629. llvm::raw_ostream& out) -> void {
  630. for (const std::string& qual : name.qualifiers()) {
  631. IdentifierToCarbon(qual, out);
  632. out << ".";
  633. }
  634. IdentifierToCarbon(name.name(), out);
  635. }
  636. static auto DeclarationToCarbon(const Fuzzing::Declaration& declaration,
  637. llvm::raw_ostream& out) -> void {
  638. switch (declaration.kind_case()) {
  639. case Fuzzing::Declaration::KIND_NOT_SET: {
  640. // Arbitrary default to avoid getting invalid syntax.
  641. out << "var x: i32;";
  642. break;
  643. }
  644. case Fuzzing::Declaration::kNamespace: {
  645. out << "namespace ";
  646. DeclaredNameToCarbon(declaration.namespace_().name(), out);
  647. out << ";";
  648. break;
  649. }
  650. case Fuzzing::Declaration::kDestructor: {
  651. const auto& function = declaration.destructor();
  652. out << "destructor";
  653. llvm::ListSeparator sep;
  654. out << "[";
  655. if (function.has_self_pattern()) {
  656. // This is a class method.
  657. out << sep;
  658. PatternToCarbon(function.self_pattern(), out);
  659. }
  660. out << "]";
  661. // Body is optional.
  662. if (function.has_body()) {
  663. out << "\n";
  664. BlockStatementToCarbon(function.body(), out);
  665. } else {
  666. out << ";";
  667. }
  668. break;
  669. }
  670. case Fuzzing::Declaration::kFunction: {
  671. const auto& function = declaration.function();
  672. out << "fn ";
  673. DeclaredNameToCarbon(function.name(), out);
  674. if (!function.deduced_parameters().empty() ||
  675. function.has_self_pattern()) {
  676. out << "[";
  677. llvm::ListSeparator sep;
  678. for (const Fuzzing::GenericBinding& p : function.deduced_parameters()) {
  679. out << sep;
  680. GenericBindingToCarbon(p, out);
  681. }
  682. if (function.has_self_pattern()) {
  683. // This is a class method.
  684. out << sep;
  685. PatternToCarbon(function.self_pattern(), out);
  686. }
  687. out << "]";
  688. }
  689. TuplePatternToCarbon(function.param_pattern(), out);
  690. ReturnTermToCarbon(function.return_term(), out);
  691. // Body is optional.
  692. if (function.has_body()) {
  693. out << "\n";
  694. BlockStatementToCarbon(function.body(), out);
  695. } else {
  696. out << ";";
  697. }
  698. break;
  699. }
  700. case Fuzzing::Declaration::kClassDeclaration: {
  701. const auto& class_declaration = declaration.class_declaration();
  702. out << "class ";
  703. DeclaredNameToCarbon(class_declaration.name(), out);
  704. // type_params is optional.
  705. if (class_declaration.has_type_params()) {
  706. TuplePatternToCarbon(class_declaration.type_params(), out);
  707. }
  708. out << "{\n";
  709. for (const auto& member : class_declaration.members()) {
  710. DeclarationToCarbon(member, out);
  711. out << "\n";
  712. }
  713. out << "}";
  714. break;
  715. }
  716. // EXPERIMENTAL MIXIN FEATURE
  717. case Fuzzing::Declaration::kMixin: {
  718. const auto& mixin_declaration = declaration.mixin();
  719. out << "__mixin ";
  720. DeclaredNameToCarbon(mixin_declaration.name(), out);
  721. // type params are not implemented yet
  722. // if (mixin_declaration.has_params()) {
  723. // TuplePatternToCarbon(mixin_declaration.params(), out);
  724. //}
  725. out << "{\n";
  726. for (const auto& member : mixin_declaration.members()) {
  727. DeclarationToCarbon(member, out);
  728. out << "\n";
  729. }
  730. out << "}";
  731. // TODO: need to handle interface.self()?
  732. break;
  733. }
  734. // EXPERIMENTAL MIXIN FEATURE
  735. case Fuzzing::Declaration::kMix: {
  736. const auto& mix_declaration = declaration.mix();
  737. out << "__mix ";
  738. ExpressionToCarbon(mix_declaration.mixin(), out);
  739. out << ";";
  740. break;
  741. }
  742. case Fuzzing::Declaration::kChoice: {
  743. const auto& choice = declaration.choice();
  744. out << "choice ";
  745. DeclaredNameToCarbon(choice.name(), out);
  746. out << "{";
  747. llvm::ListSeparator sep;
  748. for (const auto& alternative : choice.alternatives()) {
  749. out << sep;
  750. IdentifierToCarbon(alternative.name(), out);
  751. if (alternative.has_signature()) {
  752. TupleLiteralExpressionToCarbon(alternative.signature(), out);
  753. }
  754. }
  755. out << "}";
  756. break;
  757. }
  758. case Fuzzing::Declaration::kVariable: {
  759. const auto& var = declaration.variable();
  760. out << "var ";
  761. BindingPatternToCarbon(var.binding(), out);
  762. // Initializer is optional.
  763. if (var.has_initializer()) {
  764. out << " = ";
  765. ExpressionToCarbon(var.initializer(), out);
  766. }
  767. out << ";";
  768. break;
  769. }
  770. case Fuzzing::Declaration::kLet: {
  771. const auto& let = declaration.let();
  772. out << "let ";
  773. PatternToCarbon(let.pattern(), out);
  774. // TODO: Print out the initializer once it's supported.
  775. // if (let.has_initializer()) {
  776. // out << " = ";
  777. // ExpressionToCarbon(let.initializer(), out);
  778. // }
  779. out << ";";
  780. break;
  781. }
  782. case Fuzzing::Declaration::kInterfaceExtends: {
  783. const auto& extends = declaration.interface_extends();
  784. out << "extends ";
  785. ExpressionToCarbon(extends.base(), out);
  786. out << ";";
  787. break;
  788. }
  789. case Fuzzing::Declaration::kInterfaceImpl: {
  790. const auto& impl = declaration.interface_impl();
  791. out << "impl ";
  792. ExpressionToCarbon(impl.impl_type(), out);
  793. out << " as ";
  794. ExpressionToCarbon(impl.constraint(), out);
  795. out << ";";
  796. break;
  797. }
  798. case Fuzzing::Declaration::kInterface: {
  799. const auto& interface = declaration.interface();
  800. out << "interface ";
  801. DeclaredNameToCarbon(interface.name(), out);
  802. out << " {\n";
  803. for (const auto& member : interface.members()) {
  804. DeclarationToCarbon(member, out);
  805. out << "\n";
  806. }
  807. out << "}";
  808. break;
  809. }
  810. case Fuzzing::Declaration::kConstraint: {
  811. const auto& constraint = declaration.constraint();
  812. out << "constraint ";
  813. DeclaredNameToCarbon(constraint.name(), out);
  814. out << " {\n";
  815. for (const auto& member : constraint.members()) {
  816. DeclarationToCarbon(member, out);
  817. out << "\n";
  818. }
  819. out << "}";
  820. break;
  821. }
  822. case Fuzzing::Declaration::kImpl: {
  823. const auto& impl = declaration.impl();
  824. if (impl.kind() == Fuzzing::ImplDeclaration::ExternalImpl) {
  825. out << "external ";
  826. }
  827. out << "impl ";
  828. ExpressionToCarbon(impl.impl_type(), out);
  829. out << " as ";
  830. ExpressionToCarbon(impl.interface(), out);
  831. out << " {\n";
  832. for (const auto& member : impl.members()) {
  833. DeclarationToCarbon(member, out);
  834. out << "\n";
  835. }
  836. out << "}";
  837. break;
  838. }
  839. case Fuzzing::Declaration::kMatchFirst: {
  840. const auto& match_first = declaration.match_first();
  841. out << "__match_first {\n";
  842. for (const auto& impl : match_first.impls()) {
  843. DeclarationToCarbon(impl, out);
  844. out << "\n";
  845. }
  846. out << "}";
  847. break;
  848. }
  849. case Fuzzing::Declaration::kAlias: {
  850. const auto& alias = declaration.alias();
  851. out << "alias ";
  852. DeclaredNameToCarbon(alias.name(), out);
  853. out << " = ";
  854. ExpressionToCarbon(alias.target(), out);
  855. out << ";";
  856. break;
  857. }
  858. }
  859. }
  860. static auto ProtoToCarbon(const Fuzzing::CompilationUnit& compilation_unit,
  861. llvm::raw_ostream& out) -> void {
  862. out << "// Generated by proto_to_carbon.\n\n";
  863. out << "package ";
  864. LibraryNameToCarbon(compilation_unit.package_statement(), out);
  865. out << (compilation_unit.is_api() ? " api" : " impl") << ";\n";
  866. if (!compilation_unit.declarations().empty()) {
  867. out << "\n";
  868. for (const auto& declaration : compilation_unit.declarations()) {
  869. DeclarationToCarbon(declaration, out);
  870. out << "\n";
  871. }
  872. }
  873. }
  874. auto ProtoToCarbon(const Fuzzing::CompilationUnit& compilation_unit)
  875. -> std::string {
  876. std::string source;
  877. llvm::raw_string_ostream out(source);
  878. ProtoToCarbon(compilation_unit, out);
  879. return source;
  880. }
  881. } // namespace Carbon