proto_to_carbon.cpp 30 KB

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