| David Reiss | ea2cba8 | 2009-03-30 21:35:00 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Licensed to the Apache Software Foundation (ASF) under one |
| 3 | * or more contributor license agreements. See the NOTICE file |
| 4 | * distributed with this work for additional information |
| 5 | * regarding copyright ownership. The ASF licenses this file |
| 6 | * to you under the Apache License, Version 2.0 (the |
| 7 | * "License"); you may not use this file except in compliance |
| 8 | * with the License. You may obtain a copy of the License at |
| 9 | * |
| 10 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 11 | * |
| 12 | * Unless required by applicable law or agreed to in writing, |
| 13 | * software distributed under the License is distributed on an |
| 14 | * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| 15 | * KIND, either express or implied. See the License for the |
| 16 | * specific language governing permissions and limitations |
| 17 | * under the License. |
| 18 | */ |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 19 | |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 20 | #include <string> |
| 21 | #include <fstream> |
| 22 | #include <iostream> |
| 23 | #include <vector> |
| 24 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 25 | #include <stdlib.h> |
| 26 | #include <sys/stat.h> |
| 27 | #include <sys/types.h> |
| 28 | #include <sstream> |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 29 | |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 30 | #include "t_oop_generator.h" |
| David Reiss | 204420f | 2008-01-11 20:59:03 +0000 | [diff] [blame] | 31 | #include "platform.h" |
| Roger Meier | 08d4681 | 2011-04-12 19:08:21 +0000 | [diff] [blame] | 32 | #include "version.h" |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 33 | |
| Jens Geyer | 945537c | 2013-01-04 19:33:29 +0100 | [diff] [blame] | 34 | using std::map; |
| 35 | using std::ofstream; |
| 36 | using std::ostringstream; |
| 37 | using std::string; |
| 38 | using std::stringstream; |
| 39 | using std::vector; |
| 40 | |
| 41 | static const string endl = "\n"; // avoid ostream << std::endl flushes |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 42 | |
| 43 | /** |
| 44 | * Haskell code generator. |
| 45 | * |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 46 | */ |
| 47 | class t_hs_generator : public t_oop_generator { |
| 48 | public: |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 49 | t_hs_generator(t_program* program, |
| 50 | const map<string, string>& parsed_options, |
| 51 | const string& option_string) |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 52 | : t_oop_generator(program) |
| 53 | { |
| Roger Meier | 3b771a1 | 2010-11-17 22:11:26 +0000 | [diff] [blame] | 54 | (void) parsed_options; |
| 55 | (void) option_string; |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 56 | out_dir_base_ = "gen-hs"; |
| 57 | } |
| 58 | |
| 59 | /** |
| 60 | * Init and close methods |
| 61 | */ |
| 62 | |
| 63 | void init_generator(); |
| 64 | void close_generator(); |
| 65 | |
| 66 | /** |
| 67 | * Program-level generation functions |
| 68 | */ |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 69 | void generate_typedef (t_typedef* ttypedef); |
| 70 | void generate_enum (t_enum* tenum); |
| 71 | void generate_const (t_const* tconst); |
| 72 | void generate_struct (t_struct* tstruct); |
| 73 | void generate_xception (t_struct* txception); |
| 74 | void generate_service (t_service* tservice); |
| 75 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 76 | string render_const_value(t_type* type, t_const_value* value); |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 77 | |
| 78 | /** |
| 79 | * Struct generation code |
| 80 | */ |
| 81 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 82 | void generate_hs_struct (t_struct* tstruct, |
| 83 | bool is_exception); |
| 84 | |
| 85 | void generate_hs_struct_definition (ofstream &out, |
| 86 | t_struct* tstruct, |
| 87 | bool is_xception = false, |
| 88 | bool helper = false); |
| 89 | |
| 90 | void generate_hs_struct_reader (ofstream& out, |
| 91 | t_struct* tstruct); |
| 92 | |
| 93 | void generate_hs_struct_writer (ofstream& out, |
| 94 | t_struct* tstruct); |
| 95 | |
| 96 | void generate_hs_function_helpers (t_function* tfunction); |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 97 | |
| 98 | /** |
| 99 | * Service-level generation functions |
| 100 | */ |
| 101 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 102 | void generate_service_helpers (t_service* tservice); |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 103 | void generate_service_interface (t_service* tservice); |
| 104 | void generate_service_client (t_service* tservice); |
| 105 | void generate_service_server (t_service* tservice); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 106 | void generate_process_function (t_service* tservice, |
| 107 | t_function* tfunction); |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 108 | |
| 109 | /** |
| 110 | * Serialization constructs |
| 111 | */ |
| 112 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 113 | void generate_deserialize_field (ofstream &out, |
| 114 | t_field* tfield, |
| 115 | string prefix); |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 116 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 117 | void generate_deserialize_struct (ofstream &out, |
| 118 | t_struct* tstruct); |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 119 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 120 | void generate_deserialize_container (ofstream &out, |
| 121 | t_type* ttype); |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 122 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 123 | void generate_deserialize_set_element (ofstream &out, |
| 124 | t_set* tset); |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 125 | |
| 126 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 127 | void generate_deserialize_list_element (ofstream &out, |
| 128 | t_list* tlist, |
| 129 | string prefix = ""); |
| 130 | |
| 131 | void generate_deserialize_type (ofstream &out, |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 132 | t_type* type); |
| 133 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 134 | void generate_serialize_field (ofstream &out, |
| 135 | t_field* tfield, |
| 136 | string name = ""); |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 137 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 138 | void generate_serialize_struct (ofstream &out, |
| 139 | t_struct* tstruct, |
| 140 | string prefix = ""); |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 141 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 142 | void generate_serialize_container (ofstream &out, |
| 143 | t_type* ttype, |
| 144 | string prefix = ""); |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 145 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 146 | void generate_serialize_map_element (ofstream &out, |
| 147 | t_map* tmap, |
| 148 | string kiter, |
| 149 | string viter); |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 150 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 151 | void generate_serialize_set_element (ofstream &out, |
| 152 | t_set* tmap, |
| 153 | string iter); |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 154 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 155 | void generate_serialize_list_element (ofstream &out, |
| 156 | t_list* tlist, |
| 157 | string iter); |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 158 | |
| 159 | /** |
| 160 | * Helper rendering functions |
| 161 | */ |
| 162 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 163 | string hs_autogen_comment(); |
| 164 | string hs_language_pragma(); |
| 165 | string hs_imports(); |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 166 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 167 | string type_name(t_type* ttype, |
| 168 | string function_prefix = ""); |
| 169 | |
| 170 | string function_type(t_function* tfunc, |
| 171 | bool options = false, |
| 172 | bool io = false, |
| 173 | bool method = false); |
| 174 | |
| 175 | string type_to_enum(t_type* ttype); |
| 176 | |
| 177 | string render_hs_type(t_type* type, |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 178 | bool needs_parens); |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 179 | |
| 180 | private: |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 181 | ofstream f_types_; |
| 182 | ofstream f_consts_; |
| 183 | ofstream f_service_; |
| 184 | ofstream f_iface_; |
| 185 | ofstream f_client_; |
| David Reiss | 42e6d51 | 2008-03-27 21:40:20 +0000 | [diff] [blame] | 186 | }; |
| 187 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 188 | /** |
| 189 | * Prepares for file generation by opening up the necessary file output |
| 190 | * streams. |
| 191 | * |
| 192 | * @param tprogram The program to generate |
| 193 | */ |
| 194 | void t_hs_generator::init_generator() { |
| 195 | // Make output directory |
| David Reiss | 204420f | 2008-01-11 20:59:03 +0000 | [diff] [blame] | 196 | MKDIR(get_out_dir().c_str()); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 197 | |
| 198 | // Make output file |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 199 | string pname = capitalize(program_name_); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 200 | string f_types_name = get_out_dir() + pname + "_Types.hs"; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 201 | f_types_.open(f_types_name.c_str()); |
| 202 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 203 | string f_consts_name = get_out_dir() + pname + "_Consts.hs"; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 204 | f_consts_.open(f_consts_name.c_str()); |
| 205 | |
| 206 | // Print header |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 207 | f_types_ << hs_language_pragma() << endl; |
| 208 | f_types_ << hs_autogen_comment() << endl; |
| 209 | f_types_ << "module " << pname << "_Types where" << endl; |
| 210 | f_types_ << hs_imports() << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 211 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 212 | f_consts_ << hs_language_pragma() << endl; |
| 213 | f_consts_ << hs_autogen_comment() << endl; |
| 214 | f_consts_ << "module " << pname << "_Consts where" << endl; |
| 215 | f_consts_ << hs_imports() << endl; |
| 216 | f_consts_ << "import " << pname << "_Types" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 217 | } |
| 218 | |
| David Reiss | 752529e | 2010-01-11 19:12:56 +0000 | [diff] [blame] | 219 | string t_hs_generator::hs_language_pragma() { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 220 | return string("{-# LANGUAGE DeriveDataTypeable #-}\n" |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 221 | "{-# LANGUAGE OverloadedStrings #-}\n" |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 222 | "{-# OPTIONS_GHC -fno-warn-missing-fields #-}\n" |
| 223 | "{-# OPTIONS_GHC -fno-warn-missing-signatures #-}\n" |
| 224 | "{-# OPTIONS_GHC -fno-warn-name-shadowing #-}\n" |
| 225 | "{-# OPTIONS_GHC -fno-warn-unused-imports #-}\n" |
| 226 | "{-# OPTIONS_GHC -fno-warn-unused-matches #-}\n"); |
| David Reiss | 752529e | 2010-01-11 19:12:56 +0000 | [diff] [blame] | 227 | } |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 228 | |
| 229 | /** |
| 230 | * Autogen'd comment |
| 231 | */ |
| 232 | string t_hs_generator::hs_autogen_comment() { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 233 | return string("-----------------------------------------------------------------\n") + |
| Roger Meier | 08d4681 | 2011-04-12 19:08:21 +0000 | [diff] [blame] | 234 | "-- Autogenerated by Thrift Compiler (" + THRIFT_VERSION + ") --\n" + |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 235 | "-- --\n" + |
| 236 | "-- DO NOT EDIT UNLESS YOU ARE SURE YOU KNOW WHAT YOU ARE DOING --\n" + |
| 237 | "-----------------------------------------------------------------\n"; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 238 | } |
| 239 | |
| 240 | /** |
| 241 | * Prints standard thrift imports |
| 242 | */ |
| 243 | string t_hs_generator::hs_imports() { |
| David Reiss | 752529e | 2010-01-11 19:12:56 +0000 | [diff] [blame] | 244 | const vector<t_program*>& includes = program_->get_includes(); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 245 | string result = string( |
| 246 | "import Prelude ( Bool(..), Enum, Double, String, Maybe(..),\n" |
| 247 | " Eq, Show, Ord,\n" |
| 248 | " return, length, IO, fromIntegral, fromEnum, toEnum,\n" |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 249 | " (.), (&&), (||), (==), (++), ($), (-) )\n" |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 250 | "\n" |
| 251 | "import Control.Exception\n" |
| 252 | "import Data.ByteString.Lazy\n" |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 253 | "import Data.Hashable\n" |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 254 | "import Data.Int\n" |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 255 | "import Data.Text.Lazy ( Text )\n" |
| 256 | "import qualified Data.Text.Lazy as TL\n" |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 257 | "import Data.Typeable ( Typeable )\n" |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 258 | "import qualified Data.HashMap.Strict as Map\n" |
| 259 | "import qualified Data.HashSet as Set\n" |
| 260 | "import qualified Data.Vector as Vector\n" |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 261 | "\n" |
| 262 | "import Thrift\n" |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 263 | "import Thrift.Types ()\n" |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 264 | "\n"); |
| David Reiss | 752529e | 2010-01-11 19:12:56 +0000 | [diff] [blame] | 265 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 266 | for (size_t i = 0; i < includes.size(); ++i) |
| 267 | result += "import qualified " + capitalize(includes[i]->get_name()) + "_Types\n"; |
| 268 | |
| 269 | if (includes.size() > 0) |
| 270 | result += "\n"; |
| 271 | |
| David Reiss | 752529e | 2010-01-11 19:12:56 +0000 | [diff] [blame] | 272 | return result; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 273 | } |
| 274 | |
| 275 | /** |
| 276 | * Closes the type files |
| 277 | */ |
| 278 | void t_hs_generator::close_generator() { |
| 279 | // Close types file |
| 280 | f_types_.close(); |
| 281 | f_consts_.close(); |
| 282 | } |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 283 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 284 | /** |
| 285 | * Generates a typedef. Ez. |
| 286 | * |
| 287 | * @param ttypedef The type definition |
| 288 | */ |
| 289 | void t_hs_generator::generate_typedef(t_typedef* ttypedef) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 290 | string tname = capitalize(ttypedef->get_symbolic()); |
| 291 | string tdef = render_hs_type(ttypedef->get_type(), false); |
| 292 | indent(f_types_) << "type " << tname << " = " << tdef << endl; |
| 293 | f_types_ << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 294 | } |
| 295 | |
| 296 | /** |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 297 | * Generates code for an enumerated type. |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 298 | * the values. |
| 299 | * |
| 300 | * @param tenum The enumeration |
| 301 | */ |
| 302 | void t_hs_generator::generate_enum(t_enum* tenum) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 303 | indent(f_types_) << "data " << capitalize(tenum->get_name()) << " = "; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 304 | indent_up(); |
| 305 | vector<t_enum_value*> constants = tenum->get_constants(); |
| 306 | vector<t_enum_value*>::iterator c_iter; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 307 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 308 | bool first = true; |
| 309 | for (c_iter = constants.begin(); c_iter != constants.end(); ++c_iter) { |
| 310 | string name = capitalize((*c_iter)->get_name()); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 311 | f_types_ << (first ? "" : "|"); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 312 | f_types_ << name; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 313 | first = false; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 314 | } |
| Bryan Duxbury | 0781f2b | 2009-04-07 23:29:42 +0000 | [diff] [blame] | 315 | indent(f_types_) << "deriving (Show,Eq, Typeable, Ord)" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 316 | indent_down(); |
| 317 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 318 | string ename = capitalize(tenum->get_name()); |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 319 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 320 | indent(f_types_) << "instance Enum " << ename << " where" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 321 | indent_up(); |
| 322 | indent(f_types_) << "fromEnum t = case t of" << endl; |
| 323 | indent_up(); |
| 324 | for (c_iter = constants.begin(); c_iter != constants.end(); ++c_iter) { |
| Bryan Duxbury | a406b90 | 2010-09-27 23:37:44 +0000 | [diff] [blame] | 325 | int value = (*c_iter)->get_value(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 326 | string name = capitalize((*c_iter)->get_name()); |
| Bryan Duxbury | a406b90 | 2010-09-27 23:37:44 +0000 | [diff] [blame] | 327 | indent(f_types_) << name << " -> " << value << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 328 | } |
| 329 | indent_down(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 330 | indent(f_types_) << "toEnum t = case t of" << endl; |
| 331 | indent_up(); |
| 332 | for(c_iter = constants.begin(); c_iter != constants.end(); ++c_iter) { |
| Bryan Duxbury | a406b90 | 2010-09-27 23:37:44 +0000 | [diff] [blame] | 333 | int value = (*c_iter)->get_value(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 334 | string name = capitalize((*c_iter)->get_name()); |
| Bryan Duxbury | a406b90 | 2010-09-27 23:37:44 +0000 | [diff] [blame] | 335 | indent(f_types_) << value << " -> " << name << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 336 | } |
| Bryan Duxbury | 0781f2b | 2009-04-07 23:29:42 +0000 | [diff] [blame] | 337 | indent(f_types_) << "_ -> throw ThriftException" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 338 | indent_down(); |
| 339 | indent_down(); |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 340 | |
| 341 | indent(f_types_) << "instance Hashable " << ename << " where" << endl; |
| 342 | indent_up(); |
| 343 | indent(f_types_) << "hashWithSalt salt = hashWithSalt salt . fromEnum" << endl; |
| 344 | indent_down(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 345 | } |
| 346 | |
| 347 | /** |
| 348 | * Generate a constant value |
| 349 | */ |
| 350 | void t_hs_generator::generate_const(t_const* tconst) { |
| 351 | t_type* type = tconst->get_type(); |
| 352 | string name = decapitalize(tconst->get_name()); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 353 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 354 | t_const_value* value = tconst->get_value(); |
| 355 | |
| Kevin Clark | 4798a7a | 2009-03-24 15:56:19 +0000 | [diff] [blame] | 356 | indent(f_consts_) << name << " :: " << render_hs_type(type, false) << endl; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 357 | indent(f_consts_) << name << " = " << render_const_value(type, value) << endl; |
| 358 | f_consts_ << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 359 | } |
| 360 | |
| 361 | /** |
| 362 | * Prints the value of a constant with the given type. Note that type checking |
| 363 | * is NOT performed in this function as it is always run beforehand using the |
| 364 | * validate_types method in main.cc |
| 365 | */ |
| 366 | string t_hs_generator::render_const_value(t_type* type, t_const_value* value) { |
| David Reiss | 9a4edfa | 2008-05-01 05:52:50 +0000 | [diff] [blame] | 367 | type = get_true_type(type); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 368 | ostringstream out; |
| 369 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 370 | if (type->is_base_type()) { |
| 371 | t_base_type::t_base tbase = ((t_base_type*)type)->get_base(); |
| 372 | switch (tbase) { |
| Bryan Duxbury | 75a33e8 | 2010-09-22 00:48:56 +0000 | [diff] [blame] | 373 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 374 | case t_base_type::TYPE_STRING: |
| David Reiss | 82e6fc0 | 2009-03-26 23:32:36 +0000 | [diff] [blame] | 375 | out << '"' << get_escaped_string(value) << '"'; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 376 | break; |
| Bryan Duxbury | 75a33e8 | 2010-09-22 00:48:56 +0000 | [diff] [blame] | 377 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 378 | case t_base_type::TYPE_BOOL: |
| 379 | out << (value->get_integer() > 0 ? "True" : "False"); |
| 380 | break; |
| Bryan Duxbury | 75a33e8 | 2010-09-22 00:48:56 +0000 | [diff] [blame] | 381 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 382 | case t_base_type::TYPE_BYTE: |
| Christian Lavoie | ae7f7fa | 2010-11-02 21:42:53 +0000 | [diff] [blame] | 383 | out << "(" << value->get_integer() << " :: Int8)"; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 384 | break; |
| Bryan Duxbury | 75a33e8 | 2010-09-22 00:48:56 +0000 | [diff] [blame] | 385 | |
| 386 | case t_base_type::TYPE_I16: |
| 387 | out << "(" << value->get_integer() << " :: Int16)"; |
| 388 | break; |
| 389 | |
| 390 | case t_base_type::TYPE_I32: |
| 391 | out << "(" << value->get_integer() << " :: Int32)"; |
| 392 | break; |
| 393 | |
| 394 | case t_base_type::TYPE_I64: |
| 395 | out << "(" << value->get_integer() << " :: Int64)"; |
| 396 | break; |
| 397 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 398 | case t_base_type::TYPE_DOUBLE: |
| 399 | if (value->get_type() == t_const_value::CV_INTEGER) { |
| 400 | out << value->get_integer(); |
| 401 | } else { |
| 402 | out << value->get_double(); |
| 403 | } |
| 404 | break; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 405 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 406 | default: |
| David Reiss | dd7796f | 2007-08-28 21:09:06 +0000 | [diff] [blame] | 407 | throw "compiler error: no const of base type " + t_base_type::t_base_name(tbase); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 408 | } |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 409 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 410 | } else if (type->is_enum()) { |
| 411 | t_enum* tenum = (t_enum*)type; |
| 412 | vector<t_enum_value*> constants = tenum->get_constants(); |
| 413 | vector<t_enum_value*>::iterator c_iter; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 414 | for (c_iter = constants.begin(); c_iter != constants.end(); ++c_iter) { |
| Bryan Duxbury | a406b90 | 2010-09-27 23:37:44 +0000 | [diff] [blame] | 415 | int val = (*c_iter)->get_value(); |
| 416 | if (val == value->get_integer()) { |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 417 | indent(out) << capitalize((*c_iter)->get_name()); |
| 418 | break; |
| 419 | } |
| 420 | } |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 421 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 422 | } else if (type->is_struct() || type->is_xception()) { |
| 423 | string cname = type_name(type); |
| 424 | indent(out) << cname << "{"; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 425 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 426 | const vector<t_field*>& fields = ((t_struct*)type)->get_members(); |
| 427 | vector<t_field*>::const_iterator f_iter; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 428 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 429 | const map<t_const_value*, t_const_value*>& val = value->get_map(); |
| 430 | map<t_const_value*, t_const_value*>::const_iterator v_iter; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 431 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 432 | bool first = true; |
| 433 | for (v_iter = val.begin(); v_iter != val.end(); ++v_iter) { |
| 434 | t_type* field_type = NULL; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 435 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 436 | for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) |
| 437 | if ((*f_iter)->get_name() == v_iter->first->get_string()) |
| 438 | field_type = (*f_iter)->get_type(); |
| 439 | |
| 440 | if (field_type == NULL) |
| 441 | throw "type error: " + type->get_name() + " has no field " + v_iter->first->get_string(); |
| 442 | |
| 443 | string fname = v_iter->first->get_string(); |
| 444 | string const_value = render_const_value(field_type, v_iter->second); |
| 445 | |
| 446 | out << (first ? "" : ","); |
| 447 | out << "f_" << cname << "_" << fname << " = Just (" << const_value << ")"; |
| 448 | first = false; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 449 | } |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 450 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 451 | indent(out) << "}"; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 452 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 453 | } else if (type->is_map()) { |
| 454 | t_type* ktype = ((t_map*)type)->get_key_type(); |
| 455 | t_type* vtype = ((t_map*)type)->get_val_type(); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 456 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 457 | const map<t_const_value*, t_const_value*>& val = value->get_map(); |
| 458 | map<t_const_value*, t_const_value*>::const_iterator v_iter; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 459 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 460 | out << "(Map.fromList ["; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 461 | |
| 462 | bool first = true; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 463 | for (v_iter = val.begin(); v_iter != val.end(); ++v_iter) { |
| 464 | string key = render_const_value(ktype, v_iter->first); |
| 465 | string val = render_const_value(vtype, v_iter->second); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 466 | out << (first ? "" : ","); |
| 467 | out << "(" << key << "," << val << ")"; |
| 468 | first = false; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 469 | } |
| 470 | out << "])"; |
| Kevin Clark | 4798a7a | 2009-03-24 15:56:19 +0000 | [diff] [blame] | 471 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 472 | } else if (type->is_list() || type->is_set()) { |
| 473 | t_type* etype = type->is_list() |
| 474 | ? ((t_list*) type)->get_elem_type() |
| 475 | : ((t_set*) type)->get_elem_type(); |
| Kevin Clark | 4798a7a | 2009-03-24 15:56:19 +0000 | [diff] [blame] | 476 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 477 | const vector<t_const_value*>& val = value->get_list(); |
| 478 | vector<t_const_value*>::const_iterator v_iter; |
| Kevin Clark | 4798a7a | 2009-03-24 15:56:19 +0000 | [diff] [blame] | 479 | |
| 480 | if (type->is_set()) |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 481 | out << "(Set.fromList ["; |
| 482 | else |
| 483 | out << "(Vector.fromList "; |
| Kevin Clark | 4798a7a | 2009-03-24 15:56:19 +0000 | [diff] [blame] | 484 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 485 | bool first = true; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 486 | for (v_iter = val.begin(); v_iter != val.end(); ++v_iter) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 487 | out << (first ? "" : ","); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 488 | out << render_const_value(etype, *v_iter); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 489 | first = false; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 490 | } |
| Kevin Clark | 4798a7a | 2009-03-24 15:56:19 +0000 | [diff] [blame] | 491 | |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 492 | out << "])"; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 493 | |
| David Reiss | 9a4edfa | 2008-05-01 05:52:50 +0000 | [diff] [blame] | 494 | } else { |
| 495 | throw "CANNOT GENERATE CONSTANT FOR TYPE: " + type->get_name(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 496 | } |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 497 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 498 | return out.str(); |
| 499 | } |
| 500 | |
| 501 | /** |
| 502 | * Generates a "struct" |
| 503 | */ |
| 504 | void t_hs_generator::generate_struct(t_struct* tstruct) { |
| 505 | generate_hs_struct(tstruct, false); |
| 506 | } |
| 507 | |
| 508 | /** |
| 509 | * Generates a struct definition for a thrift exception. Basically the same |
| 510 | * as a struct, but also has an exception declaration. |
| 511 | * |
| 512 | * @param txception The struct definition |
| 513 | */ |
| 514 | void t_hs_generator::generate_xception(t_struct* txception) { |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 515 | generate_hs_struct(txception, true); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 516 | } |
| 517 | |
| 518 | /** |
| 519 | * Generates a Haskell struct |
| 520 | */ |
| 521 | void t_hs_generator::generate_hs_struct(t_struct* tstruct, |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 522 | bool is_exception) { |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 523 | generate_hs_struct_definition(f_types_,tstruct, is_exception,false); |
| 524 | } |
| 525 | |
| 526 | /** |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 527 | * Generates a struct definition for a thrift data type. |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 528 | * |
| 529 | * @param tstruct The struct definition |
| 530 | */ |
| 531 | void t_hs_generator::generate_hs_struct_definition(ofstream& out, |
| 532 | t_struct* tstruct, |
| 533 | bool is_exception, |
| 534 | bool helper) { |
| Roger Meier | 3b771a1 | 2010-11-17 22:11:26 +0000 | [diff] [blame] | 535 | (void) helper; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 536 | string tname = type_name(tstruct); |
| 537 | string name = tstruct->get_name(); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 538 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 539 | const vector<t_field*>& members = tstruct->get_members(); |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 540 | vector<t_field*>::const_iterator m_iter; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 541 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 542 | indent(out) << "data " << tname << " = " << tname; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 543 | if (members.size() > 0) { |
| 544 | out << "{"; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 545 | |
| 546 | bool first = true; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 547 | for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) { |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 548 | string mname = (*m_iter)->get_name(); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 549 | out << (first ? "" : ","); |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 550 | out << "f_" << tname << "_" << mname << " :: Maybe " << render_hs_type((*m_iter)->get_type(), true); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 551 | first = false; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 552 | } |
| 553 | out << "}"; |
| 554 | } |
| 555 | |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 556 | out << " deriving (Show,Eq,Typeable)" << endl; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 557 | |
| 558 | if (is_exception) |
| 559 | out << "instance Exception " << tname << endl; |
| 560 | |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 561 | indent(out) << "instance Hashable " << tname << " where" << endl; |
| 562 | indent_up(); |
| 563 | indent(out) << "hashWithSalt salt record = salt"; |
| 564 | for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) { |
| 565 | string mname = (*m_iter)->get_name(); |
| 566 | indent(out) << " `hashWithSalt` " << "f_" << tname << "_" << mname << " record"; |
| 567 | } |
| 568 | indent(out) << endl; |
| 569 | indent_down(); |
| 570 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 571 | generate_hs_struct_writer(out, tstruct); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 572 | generate_hs_struct_reader(out, tstruct); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 573 | } |
| 574 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 575 | /** |
| 576 | * Generates the read method for a struct |
| 577 | */ |
| 578 | void t_hs_generator::generate_hs_struct_reader(ofstream& out, t_struct* tstruct) { |
| 579 | const vector<t_field*>& fields = tstruct->get_members(); |
| 580 | vector<t_field*>::const_iterator f_iter; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 581 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 582 | string sname = type_name(tstruct); |
| 583 | string str = tmp("_str"); |
| 584 | string t = tmp("_t"); |
| 585 | string id = tmp("_id"); |
| 586 | |
| Anthony F. Molinaro | daef1c8 | 2010-09-26 04:25:36 +0000 | [diff] [blame] | 587 | indent(out) << "read_" << sname << "_fields iprot record = do" << endl; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 588 | indent_up(); |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 589 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 590 | // Read beginning field marker |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 591 | indent(out) << "(_," << t << "," << id << ") <- readFieldBegin iprot" << endl; |
| 592 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 593 | // Check for field STOP marker and break |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 594 | indent(out) << "if " << t << " == T_STOP then return record else" << endl; |
| 595 | |
| 596 | indent_up(); |
| 597 | indent(out) << "case " << id << " of " << endl; |
| 598 | indent_up(); |
| 599 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 600 | // Generate deserialization code for known cases |
| 601 | for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 602 | int32_t key = (*f_iter)->get_key(); |
| 603 | string etype = type_to_enum((*f_iter)->get_type()); |
| 604 | indent(out) << key << " -> " << "if " << t << " == " << etype << " then do" << endl; |
| 605 | |
| 606 | indent_up(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 607 | indent(out) << "s <- "; |
| 608 | generate_deserialize_field(out, *f_iter,str); |
| 609 | out << endl; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 610 | |
| 611 | string fname = decapitalize((*f_iter)->get_name()); |
| 612 | indent(out) << "read_" << sname << "_fields iprot record{f_" << sname << "_" << fname << "=Just s}" << endl; |
| 613 | |
| 614 | indent(out) << "else do" << endl; |
| 615 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 616 | indent_up(); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 617 | indent(out) << "skip iprot " << t << endl; |
| 618 | |
| 619 | indent(out) << "read_" << sname << "_fields iprot record" << endl; |
| 620 | |
| 621 | indent_down(); |
| 622 | indent_down(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 623 | } |
| 624 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 625 | // In the default case we skip the field |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 626 | indent(out) << "_ -> do" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 627 | indent_up(); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 628 | indent(out) << "skip iprot " << t << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 629 | indent(out) << "readFieldEnd iprot" << endl; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 630 | indent(out) << "read_" << sname << "_fields iprot record" << endl; |
| 631 | indent_down(); |
| 632 | indent_down(); |
| 633 | indent_down(); |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 634 | indent_down(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 635 | |
| 636 | // read |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 637 | indent(out) << "read_" << sname << " iprot = do" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 638 | indent_up(); |
| Bryan Duxbury | f38b2f1 | 2010-09-20 17:41:40 +0000 | [diff] [blame] | 639 | indent(out) << "_ <- readStructBegin iprot" << endl; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 640 | indent(out) << "record <- read_" << sname << "_fields iprot (" << sname << "{"; |
| 641 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 642 | bool first = true; |
| 643 | for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 644 | out << (first ? "" : ","); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 645 | out << "f_" << sname << "_" << decapitalize((*f_iter)->get_name()) << "=Nothing"; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 646 | first = false; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 647 | } |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 648 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 649 | out << "})" << endl; |
| 650 | indent(out) << "readStructEnd iprot" << endl; |
| Anthony F. Molinaro | daef1c8 | 2010-09-26 04:25:36 +0000 | [diff] [blame] | 651 | indent(out) << "return record" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 652 | indent_down(); |
| 653 | } |
| 654 | |
| 655 | void t_hs_generator::generate_hs_struct_writer(ofstream& out, |
| 656 | t_struct* tstruct) { |
| 657 | string name = type_name(tstruct); |
| Bryan Duxbury | ff219ac | 2009-04-10 21:51:00 +0000 | [diff] [blame] | 658 | const vector<t_field*>& fields = tstruct->get_sorted_members(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 659 | vector<t_field*>::const_iterator f_iter; |
| 660 | string str = tmp("_str"); |
| 661 | string f = tmp("_f"); |
| 662 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 663 | indent(out) << "write_" << name << " oprot record = do" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 664 | indent_up(); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 665 | indent(out) << "writeStructBegin oprot \"" << name << "\"" << endl; |
| 666 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 667 | for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) { |
| 668 | // Write field header |
| 669 | string mname = (*f_iter)->get_name(); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 670 | indent(out) << "case f_" << name << "_" << mname << " record of {Nothing -> return (); Just _v -> do" << endl; |
| 671 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 672 | indent_up(); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 673 | indent(out) << "writeFieldBegin oprot (\"" << (*f_iter)->get_name() << "\"," |
| 674 | << type_to_enum((*f_iter)->get_type()) << "," |
| 675 | << (*f_iter)->get_key() << ")" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 676 | |
| 677 | // Write field contents |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 678 | indent(out); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 679 | generate_serialize_field(out, *f_iter, "_v"); |
| 680 | out << endl; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 681 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 682 | // Write field closer |
| 683 | indent(out) << "writeFieldEnd oprot}" << endl; |
| 684 | indent_down(); |
| 685 | } |
| 686 | |
| 687 | // Write the struct map |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 688 | indent(out) << "writeFieldStop oprot" << endl; |
| 689 | indent(out) << "writeStructEnd oprot" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 690 | |
| 691 | indent_down(); |
| 692 | } |
| 693 | |
| 694 | /** |
| 695 | * Generates a thrift service. |
| 696 | * |
| 697 | * @param tservice The service definition |
| 698 | */ |
| 699 | void t_hs_generator::generate_service(t_service* tservice) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 700 | string f_service_name = get_out_dir() + capitalize(service_name_) + ".hs"; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 701 | f_service_.open(f_service_name.c_str()); |
| 702 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 703 | f_service_ << hs_language_pragma() << endl; |
| 704 | f_service_ << hs_autogen_comment() << endl; |
| 705 | f_service_ << "module " << capitalize(service_name_) << " where" << endl; |
| 706 | f_service_ << hs_imports() << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 707 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 708 | if (tservice->get_extends()) { |
| 709 | f_service_ << "import qualified " << capitalize(tservice->get_extends()->get_name()) << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 710 | } |
| 711 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 712 | f_service_ << "import " << capitalize(program_name_) << "_Types" << endl; |
| 713 | f_service_ << "import qualified " << capitalize(service_name_) << "_Iface as Iface" << endl; |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 714 | |
| 715 | // Generate the three main parts of the service |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 716 | generate_service_helpers(tservice); |
| 717 | generate_service_interface(tservice); |
| 718 | generate_service_client(tservice); |
| 719 | generate_service_server(tservice); |
| 720 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 721 | // Close service file |
| 722 | f_service_.close(); |
| 723 | } |
| 724 | |
| 725 | /** |
| 726 | * Generates helper functions for a service. |
| 727 | * |
| 728 | * @param tservice The service to generate a header definition for |
| 729 | */ |
| 730 | void t_hs_generator::generate_service_helpers(t_service* tservice) { |
| 731 | vector<t_function*> functions = tservice->get_functions(); |
| 732 | vector<t_function*>::iterator f_iter; |
| 733 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 734 | indent(f_service_) << "-- HELPER FUNCTIONS AND STRUCTURES --" << endl; |
| 735 | indent(f_service_) << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 736 | |
| 737 | for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) { |
| 738 | t_struct* ts = (*f_iter)->get_arglist(); |
| 739 | generate_hs_struct_definition(f_service_,ts, false); |
| 740 | generate_hs_function_helpers(*f_iter); |
| 741 | } |
| 742 | } |
| 743 | |
| 744 | /** |
| 745 | * Generates a struct and helpers for a function. |
| 746 | * |
| 747 | * @param tfunction The function |
| 748 | */ |
| 749 | void t_hs_generator::generate_hs_function_helpers(t_function* tfunction) { |
| 750 | t_struct result(program_, decapitalize(tfunction->get_name()) + "_result"); |
| 751 | t_field success(tfunction->get_returntype(), "success", 0); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 752 | |
| 753 | if (!tfunction->get_returntype()->is_void()) |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 754 | result.append(&success); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 755 | |
| 756 | t_struct* xs = tfunction->get_xceptions(); |
| 757 | const vector<t_field*>& fields = xs->get_members(); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 758 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 759 | vector<t_field*>::const_iterator f_iter; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 760 | for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 761 | result.append(*f_iter); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 762 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 763 | generate_hs_struct_definition(f_service_,&result, false); |
| 764 | } |
| 765 | |
| 766 | /** |
| 767 | * Generates a service interface definition. |
| 768 | * |
| 769 | * @param tservice The service to generate a header definition for |
| 770 | */ |
| 771 | void t_hs_generator::generate_service_interface(t_service* tservice) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 772 | string f_iface_name = get_out_dir() + capitalize(service_name_) + "_Iface.hs"; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 773 | f_iface_.open(f_iface_name.c_str()); |
| Bryan Duxbury | e59a80f | 2010-09-20 15:21:37 +0000 | [diff] [blame] | 774 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 775 | f_iface_ << hs_language_pragma() << endl; |
| 776 | f_iface_ << hs_autogen_comment() << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 777 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 778 | f_iface_ << "module " << capitalize(service_name_) << "_Iface where" << endl; |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 779 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 780 | f_iface_ << hs_imports() << endl; |
| 781 | f_iface_ << "import " << capitalize(program_name_) << "_Types" << endl; |
| 782 | f_iface_ << endl; |
| 783 | |
| 784 | string sname = capitalize(service_name_); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 785 | if (tservice->get_extends() != NULL) { |
| 786 | string extends = type_name(tservice->get_extends()); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 787 | |
| 788 | indent(f_iface_) << "import " << extends << "_Iface" << endl; |
| 789 | indent(f_iface_) << "class " << extends << "_Iface a => " << sname << "_Iface a where" << endl; |
| 790 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 791 | } else { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 792 | indent(f_iface_) << "class " << sname << "_Iface a where" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 793 | } |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 794 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 795 | indent_up(); |
| 796 | |
| 797 | vector<t_function*> functions = tservice->get_functions(); |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 798 | vector<t_function*>::iterator f_iter; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 799 | for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 800 | string ft = function_type(*f_iter, true, true, true); |
| 801 | indent(f_iface_) << decapitalize((*f_iter)->get_name()) << " :: a -> " << ft << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 802 | } |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 803 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 804 | indent_down(); |
| 805 | f_iface_.close(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 806 | } |
| 807 | |
| 808 | /** |
| 809 | * Generates a service client definition. Note that in Haskell, the client doesn't implement iface. This is because |
| 810 | * The client does not (and should not have to) deal with arguments being Nothing. |
| 811 | * |
| 812 | * @param tservice The service to generate a server for. |
| 813 | */ |
| 814 | void t_hs_generator::generate_service_client(t_service* tservice) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 815 | string f_client_name = get_out_dir() + capitalize(service_name_) + "_Client.hs"; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 816 | f_client_.open(f_client_name.c_str()); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 817 | f_client_ << hs_language_pragma() << endl; |
| 818 | f_client_ << hs_autogen_comment() << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 819 | |
| 820 | vector<t_function*> functions = tservice->get_functions(); |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 821 | vector<t_function*>::const_iterator f_iter; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 822 | |
| 823 | string extends = ""; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 824 | string exports = ""; |
| 825 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 826 | bool first = true; |
| 827 | for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 828 | exports += (first ? "" : ","); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 829 | string funname = (*f_iter)->get_name(); |
| Anthony F. Molinaro | 71a58a8 | 2010-09-27 19:27:40 +0000 | [diff] [blame] | 830 | exports += decapitalize(funname); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 831 | first = false; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 832 | } |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 833 | |
| 834 | string sname = capitalize(service_name_); |
| 835 | indent(f_client_) << "module " << sname << "_Client(" << exports << ") where" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 836 | |
| 837 | if (tservice->get_extends() != NULL) { |
| 838 | extends = type_name(tservice->get_extends()); |
| 839 | indent(f_client_) << "import " << extends << "_Client" << endl; |
| 840 | } |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 841 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 842 | indent(f_client_) << "import Data.IORef" << endl; |
| 843 | indent(f_client_) << hs_imports() << endl; |
| 844 | indent(f_client_) << "import " << capitalize(program_name_) << "_Types" << endl; |
| 845 | indent(f_client_) << "import " << capitalize(service_name_) << endl; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 846 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 847 | // DATS RITE A GLOBAL VAR |
| 848 | indent(f_client_) << "seqid = newIORef 0" << endl; |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 849 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 850 | // Generate client method implementations |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 851 | for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) { |
| 852 | t_struct* arg_struct = (*f_iter)->get_arglist(); |
| 853 | const vector<t_field*>& fields = arg_struct->get_members(); |
| 854 | vector<t_field*>::const_iterator fld_iter; |
| 855 | string funname = (*f_iter)->get_name(); |
| 856 | |
| 857 | string fargs = ""; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 858 | for (fld_iter = fields.begin(); fld_iter != fields.end(); ++fld_iter) |
| 859 | fargs += " arg_" + decapitalize((*fld_iter)->get_name()); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 860 | |
| 861 | // Open function |
| Anthony F. Molinaro | 71a58a8 | 2010-09-27 19:27:40 +0000 | [diff] [blame] | 862 | indent(f_client_) << decapitalize(funname) << " (ip,op)" << fargs << " = do" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 863 | indent_up(); |
| 864 | indent(f_client_) << "send_" << funname << " op" << fargs; |
| 865 | |
| 866 | f_client_ << endl; |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 867 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 868 | if (!(*f_iter)->is_oneway()) |
| 869 | indent(f_client_) << "recv_" << funname << " ip" << endl; |
| 870 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 871 | indent_down(); |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 872 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 873 | indent(f_client_) << "send_" << funname << " op" << fargs << " = do" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 874 | indent_up(); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 875 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 876 | indent(f_client_) << "seq <- seqid" << endl; |
| 877 | indent(f_client_) << "seqn <- readIORef seq" << endl; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 878 | string argsname = capitalize((*f_iter)->get_name() + "_args"); |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 879 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 880 | // Serialize the request header |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 881 | string fname = (*f_iter)->get_name(); |
| 882 | indent(f_client_) << "writeMessageBegin op (\"" << fname << "\", M_CALL, seqn)" << endl; |
| 883 | indent(f_client_) << "write_" << argsname << " op (" << argsname << "{"; |
| 884 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 885 | bool first = true; |
| 886 | for (fld_iter = fields.begin(); fld_iter != fields.end(); ++fld_iter) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 887 | string fieldname = (*fld_iter)->get_name(); |
| 888 | f_client_ << (first ? "" : ","); |
| 889 | f_client_ << "f_" << argsname << "_" << fieldname << "=Just arg_" << fieldname; |
| 890 | first = false; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 891 | } |
| 892 | f_client_ << "})" << endl; |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 893 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 894 | // Write to the stream |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 895 | indent(f_client_) << "writeMessageEnd op" << endl; |
| 896 | indent(f_client_) << "tFlush (getTransport op)" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 897 | indent_down(); |
| 898 | |
| David Reiss | 4732925 | 2009-03-24 20:01:02 +0000 | [diff] [blame] | 899 | if (!(*f_iter)->is_oneway()) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 900 | string resultname = capitalize((*f_iter)->get_name() + "_result"); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 901 | t_struct noargs(program_); |
| 902 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 903 | string funname = string("recv_") + (*f_iter)->get_name(); |
| 904 | t_function recv_function((*f_iter)->get_returntype(), funname, &noargs); |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 905 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 906 | // Open function |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 907 | indent(f_client_) << funname << " ip = do" << endl; |
| 908 | indent_up(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 909 | |
| 910 | // TODO(mcslee): Validate message reply here, seq ids etc. |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 911 | indent(f_client_) << "(fname, mtype, rseqid) <- readMessageBegin ip" << endl; |
| 912 | indent(f_client_) << "if mtype == M_EXCEPTION then do" << endl; |
| 913 | indent(f_client_) << " x <- readAppExn ip" << endl; |
| 914 | indent(f_client_) << " readMessageEnd ip" << endl; |
| 915 | indent(f_client_) << " throw x" << endl; |
| 916 | indent(f_client_) << " else return ()" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 917 | |
| 918 | t_struct* xs = (*f_iter)->get_xceptions(); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 919 | const vector<t_field*>& xceptions = xs->get_members(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 920 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 921 | indent(f_client_) << "res <- read_" << resultname << " ip" << endl; |
| 922 | indent(f_client_) << "readMessageEnd ip" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 923 | |
| 924 | // Careful, only return _result if not a void function |
| 925 | if (!(*f_iter)->get_returntype()->is_void()) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 926 | indent(f_client_) << "case f_" << resultname << "_success res of" << endl; |
| 927 | indent_up(); |
| 928 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 929 | indent(f_client_) << "Just v -> return v" << endl; |
| 930 | indent(f_client_) << "Nothing -> do" << endl; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 931 | indent_up(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 932 | } |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 933 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 934 | vector<t_field*>::const_iterator x_iter; |
| 935 | for (x_iter = xceptions.begin(); x_iter != xceptions.end(); ++x_iter) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 936 | string xname = (*x_iter)->get_name(); |
| 937 | indent(f_client_) << "case f_" << resultname << "_" << xname << " res of" << endl; |
| 938 | indent_up(); |
| 939 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 940 | indent(f_client_) << "Nothing -> return ()" << endl; |
| Bryan Duxbury | 0781f2b | 2009-04-07 23:29:42 +0000 | [diff] [blame] | 941 | indent(f_client_) << "Just _v -> throw _v" << endl; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 942 | indent_down(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 943 | } |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 944 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 945 | // Careful, only return _result if not a void function |
| 946 | if ((*f_iter)->get_returntype()->is_void()) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 947 | indent(f_client_) << "return ()" << endl; |
| 948 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 949 | } else { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 950 | string tname = (*f_iter)->get_name(); |
| 951 | indent(f_client_) << "throw (AppExn AE_MISSING_RESULT \"" << tname << " failed: unknown result\")" << endl; |
| 952 | indent_down(); |
| 953 | indent_down(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 954 | } |
| 955 | |
| 956 | // Close function |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 957 | indent_down(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 958 | } |
| 959 | } |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 960 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 961 | f_client_.close(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 962 | } |
| 963 | |
| 964 | /** |
| 965 | * Generates a service server definition. |
| 966 | * |
| 967 | * @param tservice The service to generate a server for. |
| 968 | */ |
| 969 | void t_hs_generator::generate_service_server(t_service* tservice) { |
| 970 | // Generate the dispatch methods |
| 971 | vector<t_function*> functions = tservice->get_functions(); |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 972 | vector<t_function*>::iterator f_iter; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 973 | |
| 974 | // Generate the process subfunctions |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 975 | for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 976 | generate_process_function(tservice, *f_iter); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 977 | |
| Anthony F. Molinaro | daef1c8 | 2010-09-26 04:25:36 +0000 | [diff] [blame] | 978 | indent(f_service_) << "proc_ handler (iprot,oprot) (name,typ,seqid) = case name of" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 979 | indent_up(); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 980 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 981 | for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) { |
| 982 | string fname = (*f_iter)->get_name(); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 983 | indent(f_service_) << "\"" << fname << "\" -> process_" << decapitalize(fname) << " (seqid,iprot,oprot,handler)" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 984 | } |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 985 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 986 | indent(f_service_) << "_ -> "; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 987 | if (tservice->get_extends() != NULL) { |
| Anthony F. Molinaro | daef1c8 | 2010-09-26 04:25:36 +0000 | [diff] [blame] | 988 | f_service_ << type_name(tservice->get_extends()) << ".proc_ handler (iprot,oprot) (name,typ,seqid)" << endl; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 989 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 990 | } else { |
| 991 | f_service_ << "do" << endl; |
| 992 | indent_up(); |
| 993 | indent(f_service_) << "skip iprot T_STRUCT" << endl; |
| 994 | indent(f_service_) << "readMessageEnd iprot" << endl; |
| 995 | indent(f_service_) << "writeMessageBegin oprot (name,M_EXCEPTION,seqid)" << endl; |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 996 | indent(f_service_) << "writeAppExn oprot (AppExn AE_UNKNOWN_METHOD (\"Unknown function \" ++ TL.unpack name))" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 997 | indent(f_service_) << "writeMessageEnd oprot" << endl; |
| Bryan Duxbury | 0781f2b | 2009-04-07 23:29:42 +0000 | [diff] [blame] | 998 | indent(f_service_) << "tFlush (getTransport oprot)" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 999 | indent_down(); |
| 1000 | } |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1001 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1002 | indent_down(); |
| 1003 | |
| 1004 | // Generate the server implementation |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1005 | indent(f_service_) << "process handler (iprot, oprot) = do" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1006 | indent_up(); |
| 1007 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1008 | indent(f_service_) << "(name, typ, seqid) <- readMessageBegin iprot" << endl; |
| 1009 | indent(f_service_) << "proc_ handler (iprot,oprot) (name,typ,seqid)" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1010 | indent(f_service_) << "return True" << endl; |
| 1011 | indent_down(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1012 | } |
| 1013 | |
| 1014 | /** |
| 1015 | * Generates a process function definition. |
| 1016 | * |
| 1017 | * @param tfunction The function to write a dispatcher for |
| 1018 | */ |
| 1019 | void t_hs_generator::generate_process_function(t_service* tservice, |
| 1020 | t_function* tfunction) { |
| Roger Meier | 3b771a1 | 2010-11-17 22:11:26 +0000 | [diff] [blame] | 1021 | (void) tservice; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1022 | // Open function |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1023 | string funname = decapitalize(tfunction->get_name()); |
| 1024 | indent(f_service_) << "process_" << funname << " (seqid, iprot, oprot, handler) = do" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1025 | indent_up(); |
| 1026 | |
| 1027 | string argsname = capitalize(tfunction->get_name()) + "_args"; |
| 1028 | string resultname = capitalize(tfunction->get_name()) + "_result"; |
| 1029 | |
| 1030 | // Generate the function call |
| 1031 | t_struct* arg_struct = tfunction->get_arglist(); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1032 | const vector<t_field*>& fields = arg_struct->get_members(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1033 | vector<t_field*>::const_iterator f_iter; |
| 1034 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1035 | indent(f_service_) << "args <- read_" << argsname << " iprot" << endl; |
| 1036 | indent(f_service_) << "readMessageEnd iprot" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1037 | |
| 1038 | t_struct* xs = tfunction->get_xceptions(); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1039 | const vector<t_field*>& xceptions = xs->get_members(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1040 | vector<t_field*>::const_iterator x_iter; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1041 | |
| 1042 | size_t n = xceptions.size(); |
| 1043 | if (!tfunction->is_oneway()) { |
| 1044 | if (!tfunction->get_returntype()->is_void()) |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1045 | n++; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1046 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1047 | indent(f_service_) << "rs <- return (" << resultname; |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 1048 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1049 | for(size_t i = 0; i < n; i++) |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1050 | f_service_ << " Nothing"; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1051 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1052 | f_service_ << ")" << endl; |
| 1053 | } |
| 1054 | |
| 1055 | indent(f_service_) << "res <- "; |
| 1056 | // Try block for a function with exceptions |
| 1057 | if (xceptions.size() > 0) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1058 | for(size_t i = 0; i < xceptions.size(); i++) { |
| Bryan Duxbury | 0781f2b | 2009-04-07 23:29:42 +0000 | [diff] [blame] | 1059 | f_service_ << "(Control.Exception.catch" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1060 | indent_up(); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1061 | indent(f_service_); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1062 | } |
| 1063 | } |
| 1064 | |
| 1065 | f_service_ << "(do" << endl; |
| 1066 | indent_up(); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1067 | indent(f_service_); |
| 1068 | |
| 1069 | if (!tfunction->is_oneway() && !tfunction->get_returntype()->is_void()) |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1070 | f_service_ << "res <- "; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1071 | |
| Anthony F. Molinaro | 71a58a8 | 2010-09-27 19:27:40 +0000 | [diff] [blame] | 1072 | f_service_ << "Iface." << decapitalize(tfunction->get_name()) << " handler"; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1073 | for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1074 | f_service_ << " (f_" << argsname << "_" << (*f_iter)->get_name() << " args)"; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1075 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1076 | if (!tfunction->is_oneway() && !tfunction->get_returntype()->is_void()) { |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1077 | f_service_ << endl; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1078 | indent(f_service_) << "return rs{f_" << resultname << "_success= Just res}"; |
| 1079 | |
| 1080 | } else if (!tfunction->is_oneway()) { |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1081 | f_service_ << endl; |
| 1082 | indent(f_service_) << "return rs"; |
| 1083 | } |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1084 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1085 | f_service_ << ")" << endl; |
| 1086 | indent_down(); |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 1087 | |
| David Reiss | 4732925 | 2009-03-24 20:01:02 +0000 | [diff] [blame] | 1088 | if (xceptions.size() > 0 && !tfunction->is_oneway()) { |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1089 | for (x_iter = xceptions.begin(); x_iter != xceptions.end(); ++x_iter) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1090 | indent(f_service_) << "(\\e -> " << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1091 | indent_up(); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1092 | |
| 1093 | if (!tfunction->is_oneway()) { |
| 1094 | indent(f_service_) << "return rs{f_" << resultname << "_" << (*x_iter)->get_name() << " =Just e}"; |
| 1095 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1096 | } else { |
| 1097 | indent(f_service_) << "return ()"; |
| 1098 | } |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1099 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1100 | f_service_ << "))" << endl; |
| 1101 | indent_down(); |
| 1102 | indent_down(); |
| 1103 | } |
| 1104 | } |
| 1105 | |
| David Reiss | c51986f | 2009-03-24 20:01:25 +0000 | [diff] [blame] | 1106 | // Shortcut out here for oneway functions |
| David Reiss | 4732925 | 2009-03-24 20:01:02 +0000 | [diff] [blame] | 1107 | if (tfunction->is_oneway()) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1108 | indent(f_service_) << "return ()" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1109 | indent_down(); |
| 1110 | return; |
| 1111 | } |
| 1112 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1113 | indent(f_service_ ) << "writeMessageBegin oprot (\"" << tfunction->get_name() << "\", M_REPLY, seqid);" << endl; |
| 1114 | indent(f_service_ ) << "write_" << resultname << " oprot res" << endl; |
| 1115 | indent(f_service_ ) << "writeMessageEnd oprot" << endl; |
| 1116 | indent(f_service_ ) << "tFlush (getTransport oprot)" << endl; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1117 | |
| 1118 | // Close function |
| 1119 | indent_down(); |
| 1120 | } |
| 1121 | |
| 1122 | /** |
| 1123 | * Deserializes a field of any type. |
| 1124 | */ |
| 1125 | void t_hs_generator::generate_deserialize_field(ofstream &out, |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1126 | t_field* tfield, |
| 1127 | string prefix) { |
| Roger Meier | 3b771a1 | 2010-11-17 22:11:26 +0000 | [diff] [blame] | 1128 | (void) prefix; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1129 | t_type* type = tfield->get_type(); |
| 1130 | generate_deserialize_type(out,type); |
| 1131 | } |
| 1132 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1133 | /** |
| 1134 | * Deserializes a field of any type. |
| 1135 | */ |
| 1136 | void t_hs_generator::generate_deserialize_type(ofstream &out, |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1137 | t_type* type) { |
| David Reiss | e087a30 | 2007-08-23 21:43:25 +0000 | [diff] [blame] | 1138 | type = get_true_type(type); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1139 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1140 | if (type->is_void()) |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1141 | throw "CANNOT GENERATE DESERIALIZE CODE FOR void TYPE"; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1142 | |
| 1143 | if (type->is_struct() || type->is_xception()) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1144 | generate_deserialize_struct(out, (t_struct*)type); |
| 1145 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1146 | } else if (type->is_container()) { |
| 1147 | generate_deserialize_container(out, type); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1148 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1149 | } else if (type->is_base_type()) { |
| 1150 | t_base_type::t_base tbase = ((t_base_type*)type)->get_base(); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1151 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1152 | switch (tbase) { |
| 1153 | case t_base_type::TYPE_VOID: |
| 1154 | throw "compiler error: cannot serialize void field in a struct"; |
| 1155 | break; |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 1156 | case t_base_type::TYPE_STRING: |
| Bryan Duxbury | 75a33e8 | 2010-09-22 00:48:56 +0000 | [diff] [blame] | 1157 | out << (((t_base_type*)type)->is_binary() ? "readBinary" : "readString"); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1158 | break; |
| 1159 | case t_base_type::TYPE_BOOL: |
| 1160 | out << "readBool"; |
| 1161 | break; |
| 1162 | case t_base_type::TYPE_BYTE: |
| 1163 | out << "readByte"; |
| 1164 | break; |
| 1165 | case t_base_type::TYPE_I16: |
| 1166 | out << "readI16"; |
| 1167 | break; |
| 1168 | case t_base_type::TYPE_I32: |
| 1169 | out << "readI32"; |
| 1170 | break; |
| 1171 | case t_base_type::TYPE_I64: |
| 1172 | out << "readI64"; |
| 1173 | break; |
| 1174 | case t_base_type::TYPE_DOUBLE: |
| 1175 | out << "readDouble"; |
| 1176 | break; |
| 1177 | default: |
| David Reiss | dd7796f | 2007-08-28 21:09:06 +0000 | [diff] [blame] | 1178 | throw "compiler error: no PHP name for base type " + t_base_type::t_base_name(tbase); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1179 | } |
| 1180 | out << " iprot"; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1181 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1182 | } else if (type->is_enum()) { |
| 1183 | string ename = capitalize(type->get_name()); |
| Anthony F. Molinaro | 71a58a8 | 2010-09-27 19:27:40 +0000 | [diff] [blame] | 1184 | out << "(do {i <- readI32 iprot; return $ toEnum $ fromIntegral i})"; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1185 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1186 | } else { |
| 1187 | printf("DO NOT KNOW HOW TO DESERIALIZE TYPE '%s'\n", |
| 1188 | type->get_name().c_str()); |
| 1189 | } |
| 1190 | } |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 1191 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1192 | |
| 1193 | /** |
| 1194 | * Generates an unserializer for a struct, calling read() |
| 1195 | */ |
| 1196 | void t_hs_generator::generate_deserialize_struct(ofstream &out, |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1197 | t_struct* tstruct) { |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1198 | string name = capitalize(tstruct->get_name()); |
| 1199 | out << "(read_" << name << " iprot)"; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1200 | } |
| 1201 | |
| 1202 | /** |
| 1203 | * Serialize a container by writing out the header followed by |
| 1204 | * data and then a footer. |
| 1205 | */ |
| 1206 | void t_hs_generator::generate_deserialize_container(ofstream &out, |
| 1207 | t_type* ttype) { |
| 1208 | string size = tmp("_size"); |
| 1209 | string ktype = tmp("_ktype"); |
| 1210 | string vtype = tmp("_vtype"); |
| 1211 | string etype = tmp("_etype"); |
| 1212 | string con = tmp("_con"); |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 1213 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1214 | t_field fsize(g_type_i32, size); |
| 1215 | t_field fktype(g_type_byte, ktype); |
| 1216 | t_field fvtype(g_type_byte, vtype); |
| 1217 | t_field fetype(g_type_byte, etype); |
| 1218 | |
| 1219 | // Declare variables, read header |
| 1220 | if (ttype->is_map()) { |
| 1221 | out << "(let {f 0 = return []; f n = do {k <- "; |
| 1222 | generate_deserialize_type(out,((t_map*)ttype)->get_key_type()); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1223 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1224 | out << "; v <- "; |
| 1225 | generate_deserialize_type(out,((t_map*)ttype)->get_val_type()); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1226 | |
| 1227 | out << ";r <- f (n-1); return $ (k,v):r}} in do {(" << ktype << "," << vtype << "," << size << ") <- readMapBegin iprot; l <- f " << size << "; return $ Map.fromList l})"; |
| 1228 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1229 | } else if (ttype->is_set()) { |
| 1230 | out << "(let {f 0 = return []; f n = do {v <- "; |
| 1231 | generate_deserialize_type(out,((t_map*)ttype)->get_key_type()); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1232 | out << ";r <- f (n-1); return $ v:r}} in do {(" << etype << "," << size << ") <- readSetBegin iprot; l <- f " << size << "; return $ Set.fromList l})"; |
| 1233 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1234 | } else if (ttype->is_list()) { |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 1235 | out << "(let f n = Vector.replicateM (fromIntegral n) ("; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1236 | generate_deserialize_type(out,((t_map*)ttype)->get_key_type()); |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 1237 | out << ") in do {(" << etype << "," << size << ") <- readListBegin iprot; f " << size << "})"; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1238 | } |
| 1239 | } |
| 1240 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1241 | /** |
| 1242 | * Serializes a field of any type. |
| 1243 | * |
| 1244 | * @param tfield The field to serialize |
| 1245 | * @param prefix Name to prepend to field name |
| 1246 | */ |
| 1247 | void t_hs_generator::generate_serialize_field(ofstream &out, |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1248 | t_field* tfield, |
| 1249 | string name) { |
| David Reiss | e087a30 | 2007-08-23 21:43:25 +0000 | [diff] [blame] | 1250 | t_type* type = get_true_type(tfield->get_type()); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1251 | |
| 1252 | // Do nothing for void types |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1253 | if (type->is_void()) |
| 1254 | throw "CANNOT GENERATE SERIALIZE CODE FOR void TYPE: " + tfield->get_name(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1255 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1256 | if (name.length() == 0) |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1257 | name = decapitalize(tfield->get_name()); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1258 | |
| 1259 | if (type->is_struct() || type->is_xception()) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1260 | generate_serialize_struct(out, (t_struct*)type, name); |
| 1261 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1262 | } else if (type->is_container()) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1263 | generate_serialize_container(out, type, name); |
| 1264 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1265 | } else if (type->is_base_type() || type->is_enum()) { |
| 1266 | if (type->is_base_type()) { |
| 1267 | t_base_type::t_base tbase = ((t_base_type*)type)->get_base(); |
| 1268 | switch (tbase) { |
| 1269 | case t_base_type::TYPE_VOID: |
| 1270 | throw |
| 1271 | "compiler error: cannot serialize void field in a struct: " + name; |
| 1272 | break; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1273 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1274 | case t_base_type::TYPE_STRING: |
| Bryan Duxbury | 75a33e8 | 2010-09-22 00:48:56 +0000 | [diff] [blame] | 1275 | out << (((t_base_type*)type)->is_binary() ? "writeBinary" : "writeString") << " oprot " << name; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1276 | break; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1277 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1278 | case t_base_type::TYPE_BOOL: |
| 1279 | out << "writeBool oprot " << name; |
| 1280 | break; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1281 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1282 | case t_base_type::TYPE_BYTE: |
| 1283 | out << "writeByte oprot " << name; |
| 1284 | break; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1285 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1286 | case t_base_type::TYPE_I16: |
| 1287 | out << "writeI16 oprot " << name; |
| 1288 | break; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1289 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1290 | case t_base_type::TYPE_I32: |
| 1291 | out << "writeI32 oprot " << name; |
| 1292 | break; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1293 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1294 | case t_base_type::TYPE_I64: |
| 1295 | out << "writeI64 oprot " << name; |
| 1296 | break; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1297 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1298 | case t_base_type::TYPE_DOUBLE: |
| 1299 | out << "writeDouble oprot " << name; |
| 1300 | break; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1301 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1302 | default: |
| David Reiss | dd7796f | 2007-08-28 21:09:06 +0000 | [diff] [blame] | 1303 | throw "compiler error: no hs name for base type " + t_base_type::t_base_name(tbase); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1304 | } |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 1305 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1306 | } else if (type->is_enum()) { |
| 1307 | string ename = capitalize(type->get_name()); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1308 | out << "writeI32 oprot (fromIntegral $ fromEnum " << name << ")"; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1309 | } |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 1310 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1311 | } else { |
| 1312 | printf("DO NOT KNOW HOW TO SERIALIZE FIELD '%s' TYPE '%s'\n", |
| 1313 | tfield->get_name().c_str(), |
| 1314 | type->get_name().c_str()); |
| 1315 | } |
| 1316 | } |
| 1317 | |
| 1318 | /** |
| 1319 | * Serializes all the members of a struct. |
| 1320 | * |
| 1321 | * @param tstruct The struct to serialize |
| 1322 | * @param prefix String prefix to attach to all fields |
| 1323 | */ |
| 1324 | void t_hs_generator::generate_serialize_struct(ofstream &out, |
| 1325 | t_struct* tstruct, |
| 1326 | string prefix) { |
| David Reiss | 752529e | 2010-01-11 19:12:56 +0000 | [diff] [blame] | 1327 | out << type_name(tstruct, "write_") << " oprot " << prefix; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1328 | } |
| 1329 | |
| 1330 | void t_hs_generator::generate_serialize_container(ofstream &out, |
| 1331 | t_type* ttype, |
| 1332 | string prefix) { |
| 1333 | if (ttype->is_map()) { |
| 1334 | string k = tmp("_kiter"); |
| 1335 | string v = tmp("_viter"); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1336 | out << "(let {f [] = return (); f ((" << k << "," << v << "):t) = do {"; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1337 | generate_serialize_map_element(out, (t_map*)ttype, k, v); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1338 | out << ";f t}} in do {writeMapBegin oprot (" << type_to_enum(((t_map*)ttype)->get_key_type()) << "," << type_to_enum(((t_map*)ttype)->get_val_type()) << ",fromIntegral $ Map.size " << prefix << "); f (Map.toList " << prefix << ");writeMapEnd oprot})"; |
| 1339 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1340 | } else if (ttype->is_set()) { |
| 1341 | string v = tmp("_viter"); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1342 | out << "(let {f [] = return (); f (" << v << ":t) = do {"; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1343 | generate_serialize_set_element(out, (t_set*)ttype, v); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1344 | out << ";f t}} in do {writeSetBegin oprot (" << type_to_enum(((t_set*)ttype)->get_elem_type()) << ",fromIntegral $ Set.size " << prefix << "); f (Set.toList " << prefix << ");writeSetEnd oprot})"; |
| 1345 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1346 | } else if (ttype->is_list()) { |
| 1347 | string v = tmp("_viter"); |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 1348 | out << "(let f = Vector.mapM_ (\\" << v << " -> "; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1349 | generate_serialize_list_element(out, (t_list*)ttype, v); |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 1350 | out << ") in do {writeListBegin oprot (" << type_to_enum(((t_list*)ttype)->get_elem_type()) << ",fromIntegral $ Vector.length " << prefix << "); f " << prefix << ";writeListEnd oprot})"; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1351 | } |
| 1352 | |
| 1353 | } |
| 1354 | |
| 1355 | /** |
| 1356 | * Serializes the members of a map. |
| 1357 | * |
| 1358 | */ |
| 1359 | void t_hs_generator::generate_serialize_map_element(ofstream &out, |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1360 | t_map* tmap, |
| 1361 | string kiter, |
| 1362 | string viter) { |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1363 | t_field kfield(tmap->get_key_type(), kiter); |
| 1364 | out << "do {"; |
| 1365 | generate_serialize_field(out, &kfield); |
| 1366 | out << ";"; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1367 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1368 | t_field vfield(tmap->get_val_type(), viter); |
| 1369 | generate_serialize_field(out, &vfield); |
| 1370 | out << "}"; |
| 1371 | } |
| 1372 | |
| 1373 | /** |
| 1374 | * Serializes the members of a set. |
| 1375 | */ |
| 1376 | void t_hs_generator::generate_serialize_set_element(ofstream &out, |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1377 | t_set* tset, |
| 1378 | string iter) { |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1379 | t_field efield(tset->get_elem_type(), iter); |
| 1380 | generate_serialize_field(out, &efield); |
| 1381 | } |
| 1382 | |
| 1383 | /** |
| 1384 | * Serializes the members of a list. |
| 1385 | */ |
| 1386 | void t_hs_generator::generate_serialize_list_element(ofstream &out, |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1387 | t_list* tlist, |
| 1388 | string iter) { |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1389 | t_field efield(tlist->get_elem_type(), iter); |
| 1390 | generate_serialize_field(out, &efield); |
| 1391 | } |
| 1392 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1393 | string t_hs_generator::function_type(t_function* tfunc, bool options, bool io, bool method) { |
| 1394 | string result = ""; |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 1395 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1396 | const vector<t_field*>& fields = tfunc->get_arglist()->get_members(); |
| 1397 | vector<t_field*>::const_iterator f_iter; |
| 1398 | for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1399 | if (options) result += "Maybe "; |
| Kevin Clark | 38c8b5b | 2009-03-24 14:51:51 +0000 | [diff] [blame] | 1400 | result += render_hs_type((*f_iter)->get_type(), options); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1401 | result += " -> "; |
| 1402 | } |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1403 | |
| 1404 | if (fields.empty() && !method) |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1405 | result += "() -> "; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1406 | |
| 1407 | if (io) |
| 1408 | result += "IO "; |
| 1409 | |
| Kevin Clark | 38c8b5b | 2009-03-24 14:51:51 +0000 | [diff] [blame] | 1410 | result += render_hs_type(tfunc->get_returntype(), io); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1411 | return result; |
| 1412 | } |
| 1413 | |
| 1414 | |
| David Reiss | 752529e | 2010-01-11 19:12:56 +0000 | [diff] [blame] | 1415 | string t_hs_generator::type_name(t_type* ttype, string function_prefix) { |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1416 | string prefix = ""; |
| 1417 | t_program* program = ttype->get_program(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1418 | |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1419 | if (program != NULL && program != program_) |
| 1420 | if (!ttype->is_service()) |
| 1421 | prefix = capitalize(program->get_name()) + "_Types."; |
| 1422 | |
| 1423 | return prefix + function_prefix + capitalize(ttype->get_name()); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1424 | } |
| 1425 | |
| 1426 | /** |
| 1427 | * Converts the parse type to a Protocol.t_type enum |
| 1428 | */ |
| 1429 | string t_hs_generator::type_to_enum(t_type* type) { |
| David Reiss | e087a30 | 2007-08-23 21:43:25 +0000 | [diff] [blame] | 1430 | type = get_true_type(type); |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 1431 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1432 | if (type->is_base_type()) { |
| 1433 | t_base_type::t_base tbase = ((t_base_type*)type)->get_base(); |
| 1434 | switch (tbase) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1435 | case t_base_type::TYPE_VOID: return "T_VOID"; |
| 1436 | case t_base_type::TYPE_STRING: return "T_STRING"; |
| 1437 | case t_base_type::TYPE_BOOL: return "T_BOOL"; |
| 1438 | case t_base_type::TYPE_BYTE: return "T_BYTE"; |
| 1439 | case t_base_type::TYPE_I16: return "T_I16"; |
| 1440 | case t_base_type::TYPE_I32: return "T_I32"; |
| 1441 | case t_base_type::TYPE_I64: return "T_I64"; |
| 1442 | case t_base_type::TYPE_DOUBLE: return "T_DOUBLE"; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1443 | } |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1444 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1445 | } else if (type->is_enum()) { |
| 1446 | return "T_I32"; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1447 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1448 | } else if (type->is_struct() || type->is_xception()) { |
| 1449 | return "T_STRUCT"; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1450 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1451 | } else if (type->is_map()) { |
| 1452 | return "T_MAP"; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1453 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1454 | } else if (type->is_set()) { |
| 1455 | return "T_SET"; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1456 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1457 | } else if (type->is_list()) { |
| 1458 | return "T_LIST"; |
| 1459 | } |
| 1460 | |
| 1461 | throw "INVALID TYPE IN type_to_enum: " + type->get_name(); |
| 1462 | } |
| 1463 | |
| 1464 | /** |
| 1465 | * Converts the parse type to an haskell type |
| 1466 | */ |
| Kevin Clark | 38c8b5b | 2009-03-24 14:51:51 +0000 | [diff] [blame] | 1467 | string t_hs_generator::render_hs_type(t_type* type, bool needs_parens) { |
| David Reiss | e087a30 | 2007-08-23 21:43:25 +0000 | [diff] [blame] | 1468 | type = get_true_type(type); |
| Kevin Clark | 38c8b5b | 2009-03-24 14:51:51 +0000 | [diff] [blame] | 1469 | string type_repr; |
| David Reiss | 0c90f6f | 2008-02-06 22:18:40 +0000 | [diff] [blame] | 1470 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1471 | if (type->is_base_type()) { |
| 1472 | t_base_type::t_base tbase = ((t_base_type*)type)->get_base(); |
| 1473 | switch (tbase) { |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1474 | case t_base_type::TYPE_VOID: return "()"; |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 1475 | case t_base_type::TYPE_STRING: return (((t_base_type*)type)->is_binary() ? "ByteString" : "Text"); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1476 | case t_base_type::TYPE_BOOL: return "Bool"; |
| 1477 | case t_base_type::TYPE_BYTE: return "Int8"; |
| 1478 | case t_base_type::TYPE_I16: return "Int16"; |
| 1479 | case t_base_type::TYPE_I32: return "Int32"; |
| 1480 | case t_base_type::TYPE_I64: return "Int64"; |
| 1481 | case t_base_type::TYPE_DOUBLE: return "Double"; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1482 | } |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1483 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1484 | } else if (type->is_enum()) { |
| 1485 | return capitalize(((t_enum*)type)->get_name()); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1486 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1487 | } else if (type->is_struct() || type->is_xception()) { |
| 1488 | return type_name((t_struct*)type); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1489 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1490 | } else if (type->is_map()) { |
| 1491 | t_type* ktype = ((t_map*)type)->get_key_type(); |
| 1492 | t_type* vtype = ((t_map*)type)->get_val_type(); |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 1493 | type_repr = "Map.HashMap " + render_hs_type(ktype, true) + " " + render_hs_type(vtype, true); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1494 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1495 | } else if (type->is_set()) { |
| 1496 | t_type* etype = ((t_set*)type)->get_elem_type(); |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 1497 | type_repr = "Set.HashSet " + render_hs_type(etype, true) ; |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1498 | |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1499 | } else if (type->is_list()) { |
| 1500 | t_type* etype = ((t_list*)type)->get_elem_type(); |
| Roger Meier | 6849f20 | 2012-05-18 07:35:19 +0000 | [diff] [blame] | 1501 | type_repr = "Vector.Vector " + render_hs_type(etype, true); |
| Christian Lavoie | 600a88c | 2010-11-07 18:37:11 +0000 | [diff] [blame] | 1502 | |
| Kevin Clark | 38c8b5b | 2009-03-24 14:51:51 +0000 | [diff] [blame] | 1503 | } else { |
| 1504 | throw "INVALID TYPE IN type_to_enum: " + type->get_name(); |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1505 | } |
| 1506 | |
| Kevin Clark | 38c8b5b | 2009-03-24 14:51:51 +0000 | [diff] [blame] | 1507 | return needs_parens ? "(" + type_repr + ")" : type_repr; |
| iproctor | ff8eb92 | 2007-07-25 19:06:13 +0000 | [diff] [blame] | 1508 | } |
| David Reiss | af3ab26 | 2008-03-27 21:40:16 +0000 | [diff] [blame] | 1509 | |
| Roger Meier | 0069cc4 | 2010-10-13 18:10:18 +0000 | [diff] [blame] | 1510 | THRIFT_REGISTER_GENERATOR(hs, "Haskell", "") |