proto_to_carbon.cpp 24 KB

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