proto_to_carbon.cpp 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814
  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 PrimitiveOperatorToCarbon(
  75. const Fuzzing::PrimitiveOperatorExpression& primitive_operator,
  76. llvm::raw_ostream& out) -> void {
  77. const Fuzzing::Expression& arg0 =
  78. !primitive_operator.arguments().empty()
  79. ? primitive_operator.arguments(0)
  80. : Fuzzing::Expression::default_instance();
  81. const Fuzzing::Expression& arg1 =
  82. primitive_operator.arguments().size() > 1
  83. ? primitive_operator.arguments(1)
  84. : Fuzzing::Expression::default_instance();
  85. out << "(";
  86. switch (primitive_operator.op()) {
  87. case Fuzzing::PrimitiveOperatorExpression::UnknownOperator:
  88. // `-` is an arbitrary default to avoid getting invalid syntax.
  89. PrefixUnaryOperatorToCarbon("-", arg0, out);
  90. break;
  91. case Fuzzing::PrimitiveOperatorExpression::AddressOf:
  92. PrefixUnaryOperatorToCarbon("&", arg0, out);
  93. break;
  94. case Fuzzing::PrimitiveOperatorExpression::As:
  95. BinaryOperatorToCarbon(arg0, " as ", arg1, out);
  96. break;
  97. case Fuzzing::PrimitiveOperatorExpression::Deref:
  98. PrefixUnaryOperatorToCarbon("*", arg0, out);
  99. break;
  100. case Fuzzing::PrimitiveOperatorExpression::Mul:
  101. BinaryOperatorToCarbon(arg0, " * ", arg1, out);
  102. break;
  103. case Fuzzing::PrimitiveOperatorExpression::Ptr:
  104. PostfixUnaryOperatorToCarbon(arg0, "*", out);
  105. break;
  106. case Fuzzing::PrimitiveOperatorExpression::Neg:
  107. PrefixUnaryOperatorToCarbon("-", arg0, out);
  108. break;
  109. case Fuzzing::PrimitiveOperatorExpression::Sub:
  110. BinaryOperatorToCarbon(arg0, " - ", arg1, out);
  111. break;
  112. case Fuzzing::PrimitiveOperatorExpression::Not:
  113. // Needs a space to 'unglue' from the operand.
  114. PrefixUnaryOperatorToCarbon("not ", arg0, out);
  115. break;
  116. case Fuzzing::PrimitiveOperatorExpression::Add:
  117. BinaryOperatorToCarbon(arg0, " + ", arg1, out);
  118. break;
  119. case Fuzzing::PrimitiveOperatorExpression::And:
  120. BinaryOperatorToCarbon(arg0, " and ", arg1, out);
  121. break;
  122. case Fuzzing::PrimitiveOperatorExpression::Eq:
  123. BinaryOperatorToCarbon(arg0, " == ", arg1, out);
  124. break;
  125. case Fuzzing::PrimitiveOperatorExpression::Or:
  126. BinaryOperatorToCarbon(arg0, " or ", arg1, out);
  127. break;
  128. case Fuzzing::PrimitiveOperatorExpression::Combine:
  129. BinaryOperatorToCarbon(arg0, " & ", arg1, out);
  130. break;
  131. }
  132. out << ")";
  133. }
  134. static auto FieldInitializerToCarbon(const Fuzzing::FieldInitializer& field,
  135. std::string_view separator,
  136. llvm::raw_ostream& out) -> void {
  137. out << ".";
  138. IdentifierToCarbon(field.name(), out);
  139. out << " " << separator << " ";
  140. ExpressionToCarbon(field.expression(), out);
  141. }
  142. static auto TupleLiteralExpressionToCarbon(
  143. const Fuzzing::TupleLiteralExpression& tuple_literal,
  144. llvm::raw_ostream& out) -> void {
  145. out << "(";
  146. llvm::ListSeparator sep;
  147. for (const auto& field : tuple_literal.fields()) {
  148. out << sep;
  149. ExpressionToCarbon(field, out);
  150. }
  151. if (tuple_literal.fields_size() == 1) {
  152. // Adding a trailing comma so that generated source will be parsed as a
  153. // tuple expression. See docs/design/tuples.md.
  154. out << ", ";
  155. }
  156. out << ")";
  157. }
  158. static auto ExpressionToCarbon(const Fuzzing::Expression& expression,
  159. llvm::raw_ostream& out) -> void {
  160. switch (expression.kind_case()) {
  161. case Fuzzing::Expression::KIND_NOT_SET:
  162. // Arbitrary default for missing expressions to avoid invalid syntax.
  163. out << "true";
  164. break;
  165. case Fuzzing::Expression::kCall: {
  166. const auto& call = expression.call();
  167. ExpressionToCarbon(call.function(), out);
  168. if (call.argument().kind_case() == Fuzzing::Expression::kTupleLiteral) {
  169. TupleLiteralExpressionToCarbon(call.argument().tuple_literal(), out);
  170. } else {
  171. out << "(";
  172. ExpressionToCarbon(call.argument(), out);
  173. out << ")";
  174. }
  175. break;
  176. }
  177. case Fuzzing::Expression::kFunctionType: {
  178. const auto& fun_type = expression.function_type();
  179. out << "__Fn";
  180. TupleLiteralExpressionToCarbon(fun_type.parameter(), out);
  181. out << " -> ";
  182. ExpressionToCarbon(fun_type.return_type(), out);
  183. break;
  184. }
  185. case Fuzzing::Expression::kSimpleMemberAccess: {
  186. const auto& simple_member_access = expression.simple_member_access();
  187. ExpressionToCarbon(simple_member_access.object(), out);
  188. out << ".";
  189. IdentifierToCarbon(simple_member_access.field(), out);
  190. break;
  191. }
  192. case Fuzzing::Expression::kCompoundMemberAccess: {
  193. const auto& simple_member_access = expression.compound_member_access();
  194. ExpressionToCarbon(simple_member_access.object(), out);
  195. out << ".(";
  196. ExpressionToCarbon(simple_member_access.path(), out);
  197. out << ")";
  198. break;
  199. }
  200. case Fuzzing::Expression::kIndex: {
  201. const auto& index = expression.index();
  202. ExpressionToCarbon(index.object(), out);
  203. out << "[";
  204. ExpressionToCarbon(index.offset(), out);
  205. out << "]";
  206. break;
  207. }
  208. case Fuzzing::Expression::kPrimitiveOperator:
  209. PrimitiveOperatorToCarbon(expression.primitive_operator(), out);
  210. break;
  211. case Fuzzing::Expression::kTupleLiteral: {
  212. TupleLiteralExpressionToCarbon(expression.tuple_literal(), out);
  213. break;
  214. }
  215. case Fuzzing::Expression::kStructLiteral: {
  216. const auto& struct_literal = expression.struct_literal();
  217. out << "{";
  218. llvm::ListSeparator sep;
  219. for (const auto& field : struct_literal.fields()) {
  220. out << sep;
  221. FieldInitializerToCarbon(field, "=", out);
  222. }
  223. out << "}";
  224. break;
  225. }
  226. case Fuzzing::Expression::kStructTypeLiteral: {
  227. const auto& struct_type_literal = expression.struct_type_literal();
  228. out << "{";
  229. llvm::ListSeparator sep;
  230. for (const auto& field : struct_type_literal.fields()) {
  231. out << sep;
  232. FieldInitializerToCarbon(field, ":", out);
  233. }
  234. out << "}";
  235. break;
  236. }
  237. case Fuzzing::Expression::kIdentifier: {
  238. const auto& identifier = expression.identifier();
  239. IdentifierToCarbon(identifier.name(), out);
  240. break;
  241. }
  242. case Fuzzing::Expression::kDesignator: {
  243. const auto& designator = expression.designator();
  244. out << ".";
  245. IdentifierToCarbon(designator.name(), out);
  246. break;
  247. }
  248. case Fuzzing::Expression::kIntrinsic: {
  249. const auto& intrinsic = expression.intrinsic();
  250. switch (intrinsic.intrinsic()) {
  251. case Fuzzing::IntrinsicExpression::UnknownIntrinsic:
  252. // Arbitrary default to avoid getting invalid syntax.
  253. out << "__intrinsic_print";
  254. break;
  255. case Fuzzing::IntrinsicExpression::Print:
  256. out << "__intrinsic_print";
  257. break;
  258. case Fuzzing::IntrinsicExpression::Alloc:
  259. out << "__intrinsic_new";
  260. break;
  261. case Fuzzing::IntrinsicExpression::Dealloc:
  262. out << "__intrinsic_delete";
  263. break;
  264. }
  265. TupleLiteralExpressionToCarbon(intrinsic.argument(), out);
  266. } break;
  267. case Fuzzing::Expression::kIfExpression: {
  268. const auto& if_expression = expression.if_expression();
  269. out << "if ";
  270. ExpressionToCarbon(if_expression.condition(), out);
  271. out << " then ";
  272. ExpressionToCarbon(if_expression.then_expression(), out);
  273. out << " else ";
  274. ExpressionToCarbon(if_expression.else_expression(), out);
  275. break;
  276. }
  277. case Fuzzing::Expression::kBoolTypeLiteral:
  278. out << "Bool";
  279. break;
  280. case Fuzzing::Expression::kBoolLiteral: {
  281. const auto& bool_literal = expression.bool_literal();
  282. out << (bool_literal.value() ? "true" : "false");
  283. break;
  284. }
  285. case Fuzzing::Expression::kIntTypeLiteral:
  286. out << "i32";
  287. break;
  288. case Fuzzing::Expression::kIntLiteral: {
  289. out << expression.int_literal().value();
  290. break;
  291. }
  292. case Fuzzing::Expression::kStringLiteral:
  293. StringLiteralToCarbon(expression.string_literal().value(), out);
  294. break;
  295. case Fuzzing::Expression::kStringTypeLiteral:
  296. out << "String";
  297. break;
  298. case Fuzzing::Expression::kContinuationTypeLiteral:
  299. out << "__Continuation";
  300. break;
  301. case Fuzzing::Expression::kTypeTypeLiteral:
  302. out << "Type";
  303. break;
  304. case Fuzzing::Expression::kUnimplementedExpression:
  305. // Not really supported.
  306. // This is an arbitrary default to avoid getting invalid syntax.
  307. out << "1 __unimplemented_example_infix 2";
  308. break;
  309. case Fuzzing::Expression::kArrayTypeLiteral: {
  310. const Fuzzing::ArrayTypeLiteral& array_literal =
  311. expression.array_type_literal();
  312. out << "[";
  313. ExpressionToCarbon(array_literal.element_type(), out);
  314. out << "; ";
  315. ExpressionToCarbon(array_literal.size(), out);
  316. out << "]";
  317. break;
  318. }
  319. case Fuzzing::Expression::kWhere: {
  320. const Fuzzing::WhereExpression& where = expression.where();
  321. ExpressionToCarbon(where.base(), out);
  322. out << " where ";
  323. llvm::ListSeparator sep(" and ");
  324. for (const auto& clause : where.clauses()) {
  325. out << sep;
  326. switch (clause.kind_case()) {
  327. case Fuzzing::WhereClause::kIs:
  328. ExpressionToCarbon(clause.is().type(), out);
  329. out << " is ";
  330. ExpressionToCarbon(clause.is().constraint(), out);
  331. break;
  332. case Fuzzing::WhereClause::kEquals:
  333. ExpressionToCarbon(clause.equals().lhs(), out);
  334. out << " == ";
  335. ExpressionToCarbon(clause.equals().rhs(), out);
  336. break;
  337. case Fuzzing::WhereClause::KIND_NOT_SET:
  338. // Arbitrary default to avoid invalid syntax.
  339. out << ".Self == .Self";
  340. break;
  341. }
  342. }
  343. break;
  344. }
  345. }
  346. }
  347. static auto BindingPatternToCarbon(const Fuzzing::BindingPattern& pattern,
  348. llvm::raw_ostream& out) -> void {
  349. IdentifierToCarbon(pattern.name(), out);
  350. out << ": ";
  351. PatternToCarbon(pattern.type(), out);
  352. }
  353. static auto GenericBindingToCarbon(
  354. const Fuzzing::GenericBinding& generic_binding, llvm::raw_ostream& out) {
  355. IdentifierToCarbon(generic_binding.name(), out);
  356. out << ":! ";
  357. ExpressionToCarbon(generic_binding.type(), out);
  358. }
  359. static auto TuplePatternToCarbon(const Fuzzing::TuplePattern& tuple_pattern,
  360. llvm::raw_ostream& out) -> void {
  361. out << "(";
  362. llvm::ListSeparator sep;
  363. for (const auto& field : tuple_pattern.fields()) {
  364. out << sep;
  365. PatternToCarbon(field, out);
  366. }
  367. if (tuple_pattern.fields_size() == 1) {
  368. // Adding a trailing comma so that generated source will be parsed as a
  369. // tuple pattern expression. See docs/design/tuples.md.
  370. out << ", ";
  371. }
  372. out << ")";
  373. }
  374. static auto PatternToCarbon(const Fuzzing::Pattern& pattern,
  375. llvm::raw_ostream& out) -> void {
  376. switch (pattern.kind_case()) {
  377. case Fuzzing::Pattern::KIND_NOT_SET:
  378. // Arbitrary default to avoid getting invalid syntax.
  379. out << "auto";
  380. break;
  381. case Fuzzing::Pattern::kBindingPattern:
  382. BindingPatternToCarbon(pattern.binding_pattern(), out);
  383. break;
  384. case Fuzzing::Pattern::kTuplePattern:
  385. TuplePatternToCarbon(pattern.tuple_pattern(), out);
  386. break;
  387. case Fuzzing::Pattern::kAlternativePattern: {
  388. const auto& alternative_pattern = pattern.alternative_pattern();
  389. ExpressionToCarbon(alternative_pattern.choice_type(), out);
  390. out << ".";
  391. IdentifierToCarbon(alternative_pattern.alternative_name(), out);
  392. TuplePatternToCarbon(alternative_pattern.arguments(), out);
  393. break;
  394. }
  395. // Arbitrary expression.
  396. case Fuzzing::Pattern::kExpressionPattern: {
  397. const auto& expression_pattern = pattern.expression_pattern();
  398. ExpressionToCarbon(expression_pattern.expression(), out);
  399. break;
  400. }
  401. case Fuzzing::Pattern::kAutoPattern:
  402. out << "auto";
  403. break;
  404. case Fuzzing::Pattern::kVarPattern:
  405. out << "var ";
  406. PatternToCarbon(pattern.var_pattern().pattern(), out);
  407. break;
  408. case Fuzzing::Pattern::kGenericBinding:
  409. GenericBindingToCarbon(pattern.generic_binding(), out);
  410. break;
  411. case Fuzzing::Pattern::kAddrPattern:
  412. out << "addr ";
  413. BindingPatternToCarbon(pattern.addr_pattern().binding_pattern(), out);
  414. break;
  415. }
  416. }
  417. static auto BlockStatementToCarbon(const Fuzzing::BlockStatement& block,
  418. llvm::raw_ostream& out) -> void {
  419. out << "{\n";
  420. for (const auto& statement : block.statements()) {
  421. StatementToCarbon(statement, out);
  422. out << "\n";
  423. }
  424. out << "}\n";
  425. }
  426. static auto StatementToCarbon(const Fuzzing::Statement& statement,
  427. llvm::raw_ostream& out) -> void {
  428. switch (statement.kind_case()) {
  429. case Fuzzing::Statement::KIND_NOT_SET:
  430. // Arbitrary default to avoid getting invalid syntax.
  431. out << "true;\n";
  432. break;
  433. case Fuzzing::Statement::kExpressionStatement: {
  434. const auto& expression_statement = statement.expression_statement();
  435. ExpressionToCarbon(expression_statement.expression(), out);
  436. out << ";";
  437. break;
  438. }
  439. case Fuzzing::Statement::kAssign: {
  440. const auto& assign_statement = statement.assign();
  441. ExpressionToCarbon(assign_statement.lhs(), out);
  442. out << " = ";
  443. ExpressionToCarbon(assign_statement.rhs(), out);
  444. out << ";";
  445. break;
  446. }
  447. case Fuzzing::Statement::kVariableDefinition: {
  448. const auto& def = statement.variable_definition();
  449. if (def.is_returned()) {
  450. out << "returned ";
  451. }
  452. out << "var ";
  453. PatternToCarbon(def.pattern(), out);
  454. if (def.has_init()) {
  455. out << " = ";
  456. ExpressionToCarbon(def.init(), out);
  457. }
  458. out << ";";
  459. break;
  460. }
  461. case Fuzzing::Statement::kIfStatement: {
  462. const auto& if_statement = statement.if_statement();
  463. out << "if (";
  464. ExpressionToCarbon(if_statement.condition(), out);
  465. out << ") ";
  466. BlockStatementToCarbon(if_statement.then_block(), out);
  467. // `else` is optional.
  468. if (if_statement.has_else_block()) {
  469. out << " else ";
  470. BlockStatementToCarbon(if_statement.else_block(), out);
  471. }
  472. break;
  473. }
  474. case Fuzzing::Statement::kReturnVarStatement: {
  475. out << "return var;";
  476. break;
  477. }
  478. case Fuzzing::Statement::kReturnExpressionStatement: {
  479. const auto& ret = statement.return_expression_statement();
  480. out << "return";
  481. if (!ret.is_omitted_expression()) {
  482. out << " ";
  483. ExpressionToCarbon(ret.expression(), out);
  484. }
  485. out << ";";
  486. break;
  487. }
  488. case Fuzzing::Statement::kBlock:
  489. BlockStatementToCarbon(statement.block(), out);
  490. break;
  491. case Fuzzing::Statement::kWhileStatement: {
  492. const auto& while_statement = statement.while_statement();
  493. out << "while (";
  494. ExpressionToCarbon(while_statement.condition(), out);
  495. out << ") ";
  496. BlockStatementToCarbon(while_statement.body(), out);
  497. break;
  498. }
  499. case Fuzzing::Statement::kMatch: {
  500. const auto& match = statement.match();
  501. out << "match (";
  502. ExpressionToCarbon(match.expression(), out);
  503. out << ") {";
  504. for (const auto& clause : match.clauses()) {
  505. if (clause.is_default()) {
  506. out << "default";
  507. } else {
  508. out << "case ";
  509. PatternToCarbon(clause.pattern(), out);
  510. }
  511. out << " => ";
  512. StatementToCarbon(clause.statement(), out);
  513. }
  514. out << "}";
  515. break;
  516. }
  517. case Fuzzing::Statement::kContinuation: {
  518. const auto& continuation = statement.continuation();
  519. out << "__continuation ";
  520. IdentifierToCarbon(continuation.name(), out);
  521. BlockStatementToCarbon(continuation.body(), out);
  522. break;
  523. }
  524. case Fuzzing::Statement::kRun: {
  525. const auto& run = statement.run();
  526. out << "__run ";
  527. ExpressionToCarbon(run.argument(), out);
  528. out << ";";
  529. break;
  530. }
  531. case Fuzzing::Statement::kAwaitStatement:
  532. out << "__await;";
  533. break;
  534. case Fuzzing::Statement::kBreakStatement:
  535. out << "break;";
  536. break;
  537. case Fuzzing::Statement::kContinueStatement:
  538. out << "continue;";
  539. break;
  540. }
  541. }
  542. static auto ReturnTermToCarbon(const Fuzzing::ReturnTerm& return_term,
  543. llvm::raw_ostream& out) -> void {
  544. switch (return_term.kind()) {
  545. case Fuzzing::ReturnTerm::UnknownReturnKind:
  546. case Fuzzing::ReturnTerm::Omitted:
  547. break;
  548. case Fuzzing::ReturnTerm::Auto:
  549. out << " -> auto";
  550. break;
  551. case Fuzzing::ReturnTerm::Expression:
  552. out << " -> ";
  553. ExpressionToCarbon(return_term.type(), out);
  554. break;
  555. }
  556. }
  557. static auto DeclarationToCarbon(const Fuzzing::Declaration& declaration,
  558. llvm::raw_ostream& out) -> void {
  559. switch (declaration.kind_case()) {
  560. case Fuzzing::Declaration::KIND_NOT_SET:
  561. // Arbitrary default to avoid getting invalid syntax.
  562. out << "var x: i32;";
  563. break;
  564. case Fuzzing::Declaration::kFunction: {
  565. const auto& function = declaration.function();
  566. out << "fn ";
  567. IdentifierToCarbon(function.name(), out);
  568. if (!function.deduced_parameters().empty() || function.has_me_pattern()) {
  569. out << "[";
  570. llvm::ListSeparator sep;
  571. for (const Fuzzing::GenericBinding& p : function.deduced_parameters()) {
  572. out << sep;
  573. GenericBindingToCarbon(p, out);
  574. }
  575. if (function.has_me_pattern()) {
  576. // This is a class method.
  577. out << sep;
  578. PatternToCarbon(function.me_pattern(), out);
  579. }
  580. out << "]";
  581. }
  582. TuplePatternToCarbon(function.param_pattern(), out);
  583. ReturnTermToCarbon(function.return_term(), out);
  584. // Body is optional.
  585. if (function.has_body()) {
  586. out << "\n";
  587. BlockStatementToCarbon(function.body(), out);
  588. } else {
  589. out << ";";
  590. }
  591. break;
  592. }
  593. case Fuzzing::Declaration::kClassDeclaration: {
  594. const auto& class_declaration = declaration.class_declaration();
  595. out << "class ";
  596. IdentifierToCarbon(class_declaration.name(), out);
  597. // type_params is optional.
  598. if (class_declaration.has_type_params()) {
  599. TuplePatternToCarbon(class_declaration.type_params(), out);
  600. }
  601. out << "{\n";
  602. for (const auto& member : class_declaration.members()) {
  603. DeclarationToCarbon(member, out);
  604. out << "\n";
  605. }
  606. out << "}";
  607. break;
  608. }
  609. case Fuzzing::Declaration::kChoice: {
  610. const auto& choice = declaration.choice();
  611. out << "choice ";
  612. IdentifierToCarbon(choice.name(), out);
  613. out << "{";
  614. llvm::ListSeparator sep;
  615. for (const auto& alternative : choice.alternatives()) {
  616. out << sep;
  617. IdentifierToCarbon(alternative.name(), out);
  618. TupleLiteralExpressionToCarbon(alternative.signature(), out);
  619. }
  620. out << "}";
  621. break;
  622. }
  623. case Fuzzing::Declaration::kVariable: {
  624. const auto& var = declaration.variable();
  625. out << "var ";
  626. BindingPatternToCarbon(var.binding(), out);
  627. // Initializer is optional.
  628. if (var.has_initializer()) {
  629. out << " = ";
  630. ExpressionToCarbon(var.initializer(), out);
  631. }
  632. out << ";";
  633. break;
  634. }
  635. case Fuzzing::Declaration::kLet: {
  636. const auto& let = declaration.let();
  637. out << "let ";
  638. PatternToCarbon(let.pattern(), out);
  639. // TODO: Print out the initializer once it's supported.
  640. /*
  641. if (let.has_initializer()) {
  642. out << " = ";
  643. ExpressionToCarbon(let.initializer(), out);
  644. }
  645. */
  646. out << ";";
  647. break;
  648. }
  649. case Fuzzing::Declaration::kInterface: {
  650. const auto& interface = declaration.interface();
  651. out << "interface ";
  652. IdentifierToCarbon(interface.name(), out);
  653. out << " {\n";
  654. for (const auto& member : interface.members()) {
  655. DeclarationToCarbon(member, out);
  656. out << "\n";
  657. }
  658. out << "}";
  659. // TODO: need to handle interface.self()?
  660. break;
  661. }
  662. case Fuzzing::Declaration::kImpl: {
  663. const auto& impl = declaration.impl();
  664. if (impl.kind() == Fuzzing::ImplDeclaration::ExternalImpl) {
  665. out << "external ";
  666. }
  667. out << "impl ";
  668. ExpressionToCarbon(impl.impl_type(), out);
  669. out << " as ";
  670. ExpressionToCarbon(impl.interface(), out);
  671. out << " {\n";
  672. for (const auto& member : impl.members()) {
  673. DeclarationToCarbon(member, out);
  674. out << "\n";
  675. }
  676. out << "}";
  677. break;
  678. }
  679. case Fuzzing::Declaration::kAlias: {
  680. const auto& alias = declaration.alias();
  681. out << "alias ";
  682. IdentifierToCarbon(alias.name(), out);
  683. out << " = ";
  684. ExpressionToCarbon(alias.target(), out);
  685. out << ";";
  686. break;
  687. }
  688. }
  689. }
  690. static auto ProtoToCarbon(const Fuzzing::CompilationUnit& compilation_unit,
  691. llvm::raw_ostream& out) -> void {
  692. out << "// Generated by proto_to_carbon.\n\n";
  693. out << "package ";
  694. LibraryNameToCarbon(compilation_unit.package_statement(), out);
  695. out << (compilation_unit.is_api() ? " api" : " impl") << ";\n";
  696. if (!compilation_unit.declarations().empty()) {
  697. out << "\n";
  698. for (const auto& declaration : compilation_unit.declarations()) {
  699. DeclarationToCarbon(declaration, out);
  700. out << "\n";
  701. }
  702. }
  703. }
  704. auto ProtoToCarbon(const Fuzzing::CompilationUnit& compilation_unit)
  705. -> std::string {
  706. std::string source;
  707. llvm::raw_string_ostream out(source);
  708. ProtoToCarbon(compilation_unit, out);
  709. return source;
  710. }
  711. } // namespace Carbon