proto_to_carbon.cpp 25 KB

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