proto_to_carbon.cpp 29 KB

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