compiler/cpp/src/generate/t_xsd_generator.h

   1 // Copyright (c) 2006- Facebook
   2 // Distributed under the Thrift Software License
   3 //
   4 // See accompanying file LICENSE or visit the Thrift site at:
   5 // http://developers.facebook.com/thrift/
   6
   7 #ifndef T_XSD_GENERATOR_H
   8 #define T_XSD_GENERATOR_H
   9
  10 #include <fstream>
  11 #include <iostream>
  12 #include <sstream>
  13 #include "t_generator.h"
  14
  15 /**
  16  * XSD generator, creates an XSD for the base types etc.
  17  *
  18  * @author Mark Slee <mcslee@facebook.com>
  19  */
  20 class t_xsd_generator : public t_generator {
  21  public:
  22   t_xsd_generator(t_program* program) :
  23     t_generator(program) {
  24     out_dir_base_ = "gen-xsd";
  25   }
  26
  27   virtual ~t_xsd_generator() {}
  28
  29   /**
  30    * Init and close methods
  31    */
  32
  33   void init_generator();
  34   void close_generator();
  35
  36   /**
  37    * Program-level generation functions
  38    */
  39
  40   void generate_typedef(t_typedef* ttypedef);
  41   void generate_enum(t_enum* tenum) {}
  42
  43   void generate_service(t_service* tservice);
  44   void generate_struct(t_struct* tstruct);
  45
  46  private:
  47
  48   void generate_element(std::ostream& out, std::string name, t_type* ttype, t_struct* attrs=NULL, bool optional=false, bool nillable=false, bool list_element=false);
  49
  50   std::string ns(std::string in, std::string ns) {
  51     return ns + ":" + in;
  52   }
  53
  54   std::string xsd(std::string in) {
  55     return ns(in, "xsd");
  56   }
  57
  58   std::string type_name(t_type* ttype);
  59   std::string base_type_name(t_base_type::t_base tbase);
  60
  61   /**
  62    * Output xsd/php file
  63    */
  64   std::ofstream f_xsd_;
  65   std::ofstream f_php_;
  66
  67   /**
  68    * Output string stream
  69    */
  70   std::ostringstream s_xsd_types_;
  71
  72 };
  73
  74 #endif