schema.h 33.3 KB
Newer Older
Kenton Varda's avatar
Kenton Varda committed
1 2
// Copyright (c) 2013-2014 Sandstorm Development Group, Inc. and contributors
// Licensed under the MIT License:
3
//
Kenton Varda's avatar
Kenton Varda committed
4 5 6 7 8 9
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
10
//
Kenton Varda's avatar
Kenton Varda committed
11 12
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
13
//
Kenton Varda's avatar
Kenton Varda committed
14 15 16 17 18 19 20
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.
21

Kenton Varda's avatar
Kenton Varda committed
22 23
#ifndef CAPNP_SCHEMA_H_
#define CAPNP_SCHEMA_H_
24

Kenton Varda's avatar
Kenton Varda committed
25
#include <capnp/schema.capnp.h>
26

27
namespace capnp {
28 29 30 31 32

class Schema;
class StructSchema;
class EnumSchema;
class InterfaceSchema;
33
class ConstSchema;
34
class ListSchema;
35
class Type;
36 37

template <typename T, Kind k = kind<T>()> struct SchemaType_ { typedef Schema Type; };
Kenton Varda's avatar
Kenton Varda committed
38 39
template <typename T> struct SchemaType_<T, Kind::PRIMITIVE> { typedef schema::Type::Which Type; };
template <typename T> struct SchemaType_<T, Kind::BLOB> { typedef schema::Type::Which Type; };
40 41 42 43 44 45 46 47 48
template <typename T> struct SchemaType_<T, Kind::ENUM> { typedef EnumSchema Type; };
template <typename T> struct SchemaType_<T, Kind::STRUCT> { typedef StructSchema Type; };
template <typename T> struct SchemaType_<T, Kind::INTERFACE> { typedef InterfaceSchema Type; };
template <typename T> struct SchemaType_<T, Kind::LIST> { typedef ListSchema Type; };

template <typename T>
using SchemaType = typename SchemaType_<T>::Type;
// SchemaType<T> is the type of T's schema, e.g. StructSchema if T is a struct.

49 50 51 52 53 54 55 56 57 58
namespace _ {  // private
extern const RawSchema NULL_SCHEMA;
extern const RawSchema NULL_STRUCT_SCHEMA;
extern const RawSchema NULL_ENUM_SCHEMA;
extern const RawSchema NULL_INTERFACE_SCHEMA;
extern const RawSchema NULL_CONST_SCHEMA;
// The schema types default to these null (empty) schemas in case of error, especially when
// exceptions are disabled.
}  // namespace _ (private)

59
class Schema {
60
  // Convenience wrapper around capnp::schema::Node.
61 62

public:
63
  inline Schema(): raw(&_::NULL_SCHEMA.defaultBrand) {}
64 65 66 67 68

  template <typename T>
  static inline SchemaType<T> from() { return SchemaType<T>::template fromImpl<T>(); }
  // Get the Schema for a particular compiled-in type.

Kenton Varda's avatar
Kenton Varda committed
69
  schema::Node::Reader getProto() const;
70 71
  // Get the underlying Cap'n Proto representation of the schema node.  (Note that this accessor
  // has performance comparable to accessors of struct-typed fields on Reader classes.)
72

Kenton Varda's avatar
Kenton Varda committed
73 74 75 76
  kj::ArrayPtr<const word> asUncheckedMessage() const;
  // Get the encoded schema node content as a single message segment.  It is safe to read as an
  // unchecked message.

77 78 79 80 81 82
  Schema getDependency(uint64_t id) const KJ_DEPRECATED("Does not handle generics correctly.");
  // DEPRECATED: This method cannot correctly account for generic type parameter bindings that
  //   may apply to the dependency. Instead of using this method, use a method of the Schema API
  //   that corresponds to the exact kind of dependency. For example, to get a field type, use
  //   StructSchema::Field::getType().
  //
83 84
  // Gets the Schema for one of this Schema's dependencies.  For example, if this Schema is for a
  // struct, you could look up the schema for one of its fields' types.  Throws an exception if this
85 86 87 88 89 90 91 92 93 94 95 96 97 98 99
  // schema doesn't actually depend on the given id.
  //
  // Note that not all type IDs found in the schema node are considered "dependencies" -- only the
  // ones that are needed to implement the dynamic API are.  That includes:
  // - Field types.
  // - Group types.
  // - scopeId for group nodes, but NOT otherwise.
  // - Method parameter and return types.
  //
  // The following are NOT considered dependencies:
  // - Nested nodes.
  // - scopeId for a non-group node.
  // - Annotations.
  //
  // To obtain schemas for those, you would need a SchemaLoader.
100

101 102 103 104 105 106
  bool isBranded() const;
  // Returns true if this schema represents a non-default parameterization of this type.

  Schema getGeneric() const;
  // Get the version of this schema with any brands removed.

107 108 109 110
  class BrandArgumentList;
  BrandArgumentList getBrandArgumentsAtScope(uint64_t scopeId) const;
  // Gets the values bound to the brand parameters at the given scope.

111 112 113
  StructSchema asStruct() const;
  EnumSchema asEnum() const;
  InterfaceSchema asInterface() const;
114 115 116
  ConstSchema asConst() const;
  // Cast the Schema to a specific type.  Throws an exception if the type doesn't match.  Use
  // getProto() to determine type, e.g. getProto().isStruct().
117 118 119 120 121 122 123

  inline bool operator==(const Schema& other) const { return raw == other.raw; }
  inline bool operator!=(const Schema& other) const { return raw != other.raw; }
  // Determine whether two Schemas are wrapping the exact same underlying data, by identity.  If
  // you want to check if two Schemas represent the same type (but possibly different versions of
  // it), compare their IDs instead.

124
  template <typename T>
125
  void requireUsableAs() const;
126 127 128 129 130 131
  // Throws an exception if a value with this Schema cannot safely be cast to a native value of
  // the given type.  This passes if either:
  // - *this == from<T>()
  // - This schema was loaded with SchemaLoader, the type ID matches typeId<T>(), and
  //   loadCompiledTypeAndDependencies<T>() was called on the SchemaLoader.

132 133 134
  kj::StringPtr getShortDisplayName() const;
  // Get the short version of the node's display name.

135
private:
136
  const _::RawBrandedSchema* raw;
137

138
  inline explicit Schema(const _::RawBrandedSchema* raw): raw(raw) {
139 140 141
    KJ_IREQUIRE(raw->lazyInitializer == nullptr,
        "Must call ensureInitialized() on RawSchema before constructing Schema.");
  }
142 143

  template <typename T> static inline Schema fromImpl() {
144
    return Schema(&_::rawSchema<T>());
145 146
  }

147
  void requireUsableAs(const _::RawSchema* expected) const;
148

149 150
  uint32_t getSchemaOffset(const schema::Value::Reader& value) const;

151 152 153 154 155 156 157 158 159 160 161 162 163 164
  Type getBrandBinding(uint64_t scopeId, uint index) const;
  // Look up the binding for a brand parameter used by this Schema. Returns `AnyPointer` if the
  // parameter is not bound.
  //
  // TODO(someday): Public interface for iterating over all bindings?

  Schema getDependency(uint64_t id, uint location) const;
  // Look up schema for a particular dependency of this schema. `location` is the dependency
  // location number as defined in _::RawBrandedSchema.

  Type interpretType(schema::Type::Reader proto, uint location) const;
  // Interpret a schema::Type in the given location within the schema, compiling it into a
  // Type object.

165 166 167
  friend class StructSchema;
  friend class EnumSchema;
  friend class InterfaceSchema;
168
  friend class ConstSchema;
169 170
  friend class ListSchema;
  friend class SchemaLoader;
171
  friend class Type;
172 173
  friend kj::StringTree _::structString(
      _::StructReader reader, const _::RawBrandedSchema& schema);
174 175
};

176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210
class Schema::BrandArgumentList {
  // A list of generic parameter bindings for parameters of some particular type. Note that since
  // parameters on an outer type apply to all inner types as well, a deeply-nested type can have
  // multiple BrandArgumentLists that apply to it.
  //
  // A BrandArgumentList only represents the arguments that the client of the type specified. Since
  // new parameters can be added over time, this list may not cover all defined parameters for the
  // type. Missing parameters should be treated as AnyPointer. This class's implementation of
  // operator[] already does this for you; out-of-bounds access will safely return AnyPointer.

public:
  inline BrandArgumentList(): scopeId(0), size_(0), bindings(nullptr) {}

  inline uint size() const { return size_; }
  Type operator[](uint index) const;

  typedef _::IndexingIterator<const BrandArgumentList, Type> Iterator;
  inline Iterator begin() const { return Iterator(this, 0); }
  inline Iterator end() const { return Iterator(this, size()); }

private:
  uint64_t scopeId;
  uint size_;
  bool isUnbound;
  const _::RawBrandedSchema::Binding* bindings;

  inline BrandArgumentList(uint64_t scopeId, bool isUnbound)
      : scopeId(scopeId), size_(0), isUnbound(isUnbound), bindings(nullptr) {}
  inline BrandArgumentList(uint64_t scopeId, uint size,
                           const _::RawBrandedSchema::Binding* bindings)
      : scopeId(scopeId), size_(size), isUnbound(false), bindings(bindings) {}

  friend class Schema;
};

211 212 213 214
// -------------------------------------------------------------------

class StructSchema: public Schema {
public:
215
  inline StructSchema(): Schema(&_::NULL_STRUCT_SCHEMA.defaultBrand) {}
216

Kenton Varda's avatar
Kenton Varda committed
217
  class Field;
Kenton Varda's avatar
Kenton Varda committed
218 219
  class FieldList;
  class FieldSubset;
220

Kenton Varda's avatar
Kenton Varda committed
221 222 223 224
  FieldList getFields() const;
  // List top-level fields of this struct.  This list will contain top-level groups (including
  // named unions) but not the members of those groups.  The list does, however, contain the
  // members of the unnamed union, if there is one.
Kenton Varda's avatar
Kenton Varda committed
225

Kenton Varda's avatar
Kenton Varda committed
226 227 228 229
  FieldSubset getUnionFields() const;
  // If the field contains an unnamed union, get a list of fields in the union, ordered by
  // ordinal.  Since discriminant values are assigned sequentially by ordinal, you may index this
  // list by discriminant value.
230

Kenton Varda's avatar
Kenton Varda committed
231 232
  FieldSubset getNonUnionFields() const;
  // Get the fields of this struct which are not in an unnamed union, ordered by ordinal.
Kenton Varda's avatar
Kenton Varda committed
233

Kenton Varda's avatar
Kenton Varda committed
234 235 236 237 238 239 240 241 242 243 244 245 246 247
  kj::Maybe<Field> findFieldByName(kj::StringPtr name) const;
  // Find the field with the given name, or return null if there is no such field.  If the struct
  // contains an unnamed union, then this will find fields of that union in addition to fields
  // of the outer struct, since they exist in the same namespace.  It will not, however, find
  // members of groups (including named unions) -- you must first look up the group itself,
  // then dig into its type.

  Field getFieldByName(kj::StringPtr name) const;
  // Like findFieldByName() but throws an exception on failure.

  kj::Maybe<Field> getFieldByDiscriminant(uint16_t discriminant) const;
  // Finds the field whose `discriminantValue` is equal to the given value, or returns null if
  // there is no such field.  (If the schema does not represent a union or a struct containing
  // an unnamed union, then this always returns null.)
248

249
private:
250
  StructSchema(Schema base): Schema(base) {}
251
  template <typename T> static inline StructSchema fromImpl() {
252
    return StructSchema(Schema(&_::rawBrandedSchema<T>()));
253 254
  }
  friend class Schema;
255
  friend class Type;
256 257
};

Kenton Varda's avatar
Kenton Varda committed
258
class StructSchema::Field {
259
public:
Kenton Varda's avatar
Kenton Varda committed
260
  Field() = default;
261

Kenton Varda's avatar
Kenton Varda committed
262
  inline schema::Field::Reader getProto() const { return proto; }
263 264 265
  inline StructSchema getContainingStruct() const { return parent; }

  inline uint getIndex() const { return index; }
Kenton Varda's avatar
Kenton Varda committed
266
  // Get the index of this field within the containing struct or union.
Kenton Varda's avatar
Kenton Varda committed
267

268 269 270 271
  Type getType() const;
  // Get the type of this field. Note that this is preferred over getProto().getType() as this
  // method will apply generics.

Kenton Varda's avatar
Kenton Varda committed
272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291
  uint32_t getDefaultValueSchemaOffset() const;
  // For struct, list, and object fields, returns the offset, in words, within the first segment of
  // the struct's schema, where this field's default value pointer is located.  The schema is
  // always stored as a single-segment unchecked message, which in turn means that the default
  // value pointer itself can be treated as the root of an unchecked message -- if you know where
  // to find it, which is what this method helps you with.
  //
  // For blobs, returns the offset of the begging of the blob's content within the first segment of
  // the struct's schema.
  //
  // This is primarily useful for code generators.  The C++ code generator, for example, embeds
  // the entire schema as a raw word array within the generated code.  Of course, to implement
  // field accessors, it needs access to those fields' default values.  Embedding separate copies
  // of those default values would be redundant since they are already included in the schema, but
  // seeking through the schema at runtime to find the default values would be ugly.  Instead,
  // the code generator can use getDefaultValueSchemaOffset() to find the offset of the default
  // value within the schema, and can simply apply that offset at runtime.
  //
  // If the above does not make sense, you probably don't need this method.

Kenton Varda's avatar
Kenton Varda committed
292 293
  inline bool operator==(const Field& other) const;
  inline bool operator!=(const Field& other) const { return !(*this == other); }
Kenton Varda's avatar
Kenton Varda committed
294

295
private:
Kenton Varda's avatar
Kenton Varda committed
296 297
  StructSchema parent;
  uint index;
Kenton Varda's avatar
Kenton Varda committed
298
  schema::Field::Reader proto;
Kenton Varda's avatar
Kenton Varda committed
299

Kenton Varda's avatar
Kenton Varda committed
300
  inline Field(StructSchema parent, uint index, schema::Field::Reader proto)
Kenton Varda's avatar
Kenton Varda committed
301
      : parent(parent), index(index), proto(proto) {}
302 303 304 305

  friend class StructSchema;
};

Kenton Varda's avatar
Kenton Varda committed
306
class StructSchema::FieldList {
Kenton Varda's avatar
Kenton Varda committed
307
public:
Kenton Varda's avatar
Kenton Varda committed
308
  FieldList() = default;  // empty list
Kenton Varda's avatar
Kenton Varda committed
309

Kenton Varda's avatar
Kenton Varda committed
310 311
  inline uint size() const { return list.size(); }
  inline Field operator[](uint index) const { return Field(parent, index, list[index]); }
Kenton Varda's avatar
Kenton Varda committed
312

Kenton Varda's avatar
Kenton Varda committed
313 314 315
  typedef _::IndexingIterator<const FieldList, Field> Iterator;
  inline Iterator begin() const { return Iterator(this, 0); }
  inline Iterator end() const { return Iterator(this, size()); }
Kenton Varda's avatar
Kenton Varda committed
316 317

private:
Kenton Varda's avatar
Kenton Varda committed
318
  StructSchema parent;
Kenton Varda's avatar
Kenton Varda committed
319
  List<schema::Field>::Reader list;
Kenton Varda's avatar
Kenton Varda committed
320

Kenton Varda's avatar
Kenton Varda committed
321
  inline FieldList(StructSchema parent, List<schema::Field>::Reader list)
Kenton Varda's avatar
Kenton Varda committed
322
      : parent(parent), list(list) {}
Kenton Varda's avatar
Kenton Varda committed
323 324 325 326

  friend class StructSchema;
};

Kenton Varda's avatar
Kenton Varda committed
327
class StructSchema::FieldSubset {
328
public:
Kenton Varda's avatar
Kenton Varda committed
329
  FieldSubset() = default;  // empty list
330

Kenton Varda's avatar
Kenton Varda committed
331 332 333 334
  inline uint size() const { return size_; }
  inline Field operator[](uint index) const {
    return Field(parent, indices[index], list[indices[index]]);
  }
335

Kenton Varda's avatar
Kenton Varda committed
336
  typedef _::IndexingIterator<const FieldSubset, Field> Iterator;
337 338
  inline Iterator begin() const { return Iterator(this, 0); }
  inline Iterator end() const { return Iterator(this, size()); }
339 340 341

private:
  StructSchema parent;
Kenton Varda's avatar
Kenton Varda committed
342
  List<schema::Field>::Reader list;
Kenton Varda's avatar
Kenton Varda committed
343 344
  const uint16_t* indices;
  uint size_;
345

Kenton Varda's avatar
Kenton Varda committed
346
  inline FieldSubset(StructSchema parent, List<schema::Field>::Reader list,
Kenton Varda's avatar
Kenton Varda committed
347 348
                     const uint16_t* indices, uint size)
      : parent(parent), list(list), indices(indices), size_(size) {}
349 350 351 352 353 354 355 356

  friend class StructSchema;
};

// -------------------------------------------------------------------

class EnumSchema: public Schema {
public:
357
  inline EnumSchema(): Schema(&_::NULL_ENUM_SCHEMA.defaultBrand) {}
358

359 360 361
  class Enumerant;
  class EnumerantList;

362
  EnumerantList getEnumerants() const;
Kenton Varda's avatar
Kenton Varda committed
363

364
  kj::Maybe<Enumerant> findEnumerantByName(kj::StringPtr name) const;
365

366
  Enumerant getEnumerantByName(kj::StringPtr name) const;
Kenton Varda's avatar
Kenton Varda committed
367 368
  // Like findEnumerantByName() but throws an exception on failure.

369
private:
370
  EnumSchema(Schema base): Schema(base) {}
371
  template <typename T> static inline EnumSchema fromImpl() {
372
    return EnumSchema(Schema(&_::rawBrandedSchema<T>()));
373 374
  }
  friend class Schema;
375
  friend class Type;
376 377 378 379 380 381
};

class EnumSchema::Enumerant {
public:
  Enumerant() = default;

Kenton Varda's avatar
Kenton Varda committed
382
  inline schema::Enumerant::Reader getProto() const { return proto; }
Kenton Varda's avatar
Kenton Varda committed
383
  inline EnumSchema getContainingEnum() const { return parent; }
384

Kenton Varda's avatar
Kenton Varda committed
385 386
  inline uint16_t getOrdinal() const { return ordinal; }
  inline uint getIndex() const { return ordinal; }
387 388 389 390 391 392 393

  inline bool operator==(const Enumerant& other) const;
  inline bool operator!=(const Enumerant& other) const { return !(*this == other); }

private:
  EnumSchema parent;
  uint16_t ordinal;
Kenton Varda's avatar
Kenton Varda committed
394
  schema::Enumerant::Reader proto;
395

Kenton Varda's avatar
Kenton Varda committed
396
  inline Enumerant(EnumSchema parent, uint16_t ordinal, schema::Enumerant::Reader proto)
397 398 399 400 401 402 403
      : parent(parent), ordinal(ordinal), proto(proto) {}

  friend class EnumSchema;
};

class EnumSchema::EnumerantList {
public:
404 405
  EnumerantList() = default;  // empty list

406 407 408
  inline uint size() const { return list.size(); }
  inline Enumerant operator[](uint index) const { return Enumerant(parent, index, list[index]); }

409
  typedef _::IndexingIterator<const EnumerantList, Enumerant> Iterator;
410 411
  inline Iterator begin() const { return Iterator(this, 0); }
  inline Iterator end() const { return Iterator(this, size()); }
412 413 414

private:
  EnumSchema parent;
Kenton Varda's avatar
Kenton Varda committed
415
  List<schema::Enumerant>::Reader list;
416

Kenton Varda's avatar
Kenton Varda committed
417
  inline EnumerantList(EnumSchema parent, List<schema::Enumerant>::Reader list)
418 419 420 421 422 423 424 425 426
      : parent(parent), list(list) {}

  friend class EnumSchema;
};

// -------------------------------------------------------------------

class InterfaceSchema: public Schema {
public:
427
  inline InterfaceSchema(): Schema(&_::NULL_INTERFACE_SCHEMA.defaultBrand) {}
428

429 430 431
  class Method;
  class MethodList;

432
  MethodList getMethods() const;
Kenton Varda's avatar
Kenton Varda committed
433

434
  kj::Maybe<Method> findMethodByName(kj::StringPtr name) const;
435

436
  Method getMethodByName(kj::StringPtr name) const;
Kenton Varda's avatar
Kenton Varda committed
437 438
  // Like findMethodByName() but throws an exception on failure.

439 440 441 442 443
  class SuperclassList;

  SuperclassList getSuperclasses() const;
  // Get the immediate superclasses of this type, after applying generics.

444 445 446 447 448 449 450
  bool extends(InterfaceSchema other) const;
  // Returns true if `other` is a superclass of this interface (including if `other == *this`).

  kj::Maybe<InterfaceSchema> findSuperclass(uint64_t typeId) const;
  // Find the superclass of this interface with the given type ID.  Returns null if the interface
  // extends no such type.

451
private:
452
  InterfaceSchema(Schema base): Schema(base) {}
453
  template <typename T> static inline InterfaceSchema fromImpl() {
454
    return InterfaceSchema(Schema(&_::rawBrandedSchema<T>()));
455 456
  }
  friend class Schema;
457
  friend class Type;
458 459 460 461 462 463

  kj::Maybe<Method> findMethodByName(kj::StringPtr name, uint& counter) const;
  bool extends(InterfaceSchema other, uint& counter) const;
  kj::Maybe<InterfaceSchema> findSuperclass(uint64_t typeId, uint& counter) const;
  // We protect against malicious schemas with large or cyclic hierarchies by cutting off the
  // search when the counter reaches a threshold.
464 465 466 467 468 469
};

class InterfaceSchema::Method {
public:
  Method() = default;

Kenton Varda's avatar
Kenton Varda committed
470
  inline schema::Method::Reader getProto() const { return proto; }
Kenton Varda's avatar
Kenton Varda committed
471
  inline InterfaceSchema getContainingInterface() const { return parent; }
472

Kenton Varda's avatar
Kenton Varda committed
473 474
  inline uint16_t getOrdinal() const { return ordinal; }
  inline uint getIndex() const { return ordinal; }
475

476 477 478 479
  StructSchema getParamType() const;
  StructSchema getResultType() const;
  // Get the parameter and result types, including substituting generic parameters.

480 481 482 483 484 485
  inline bool operator==(const Method& other) const;
  inline bool operator!=(const Method& other) const { return !(*this == other); }

private:
  InterfaceSchema parent;
  uint16_t ordinal;
Kenton Varda's avatar
Kenton Varda committed
486
  schema::Method::Reader proto;
487 488

  inline Method(InterfaceSchema parent, uint16_t ordinal,
Kenton Varda's avatar
Kenton Varda committed
489
                schema::Method::Reader proto)
490 491 492 493 494 495 496
      : parent(parent), ordinal(ordinal), proto(proto) {}

  friend class InterfaceSchema;
};

class InterfaceSchema::MethodList {
public:
497 498
  MethodList() = default;  // empty list

499 500 501
  inline uint size() const { return list.size(); }
  inline Method operator[](uint index) const { return Method(parent, index, list[index]); }

502
  typedef _::IndexingIterator<const MethodList, Method> Iterator;
503 504
  inline Iterator begin() const { return Iterator(this, 0); }
  inline Iterator end() const { return Iterator(this, size()); }
505 506 507

private:
  InterfaceSchema parent;
Kenton Varda's avatar
Kenton Varda committed
508
  List<schema::Method>::Reader list;
509

Kenton Varda's avatar
Kenton Varda committed
510
  inline MethodList(InterfaceSchema parent, List<schema::Method>::Reader list)
511 512 513 514 515
      : parent(parent), list(list) {}

  friend class InterfaceSchema;
};

516 517 518 519 520 521 522
class InterfaceSchema::SuperclassList {
public:
  SuperclassList() = default;  // empty list

  inline uint size() const { return list.size(); }
  InterfaceSchema operator[](uint index) const;

523
  typedef _::IndexingIterator<const SuperclassList, InterfaceSchema> Iterator;
524 525 526 527 528
  inline Iterator begin() const { return Iterator(this, 0); }
  inline Iterator end() const { return Iterator(this, size()); }

private:
  InterfaceSchema parent;
529
  List<schema::Superclass>::Reader list;
530

531
  inline SuperclassList(InterfaceSchema parent, List<schema::Superclass>::Reader list)
532 533 534 535 536
      : parent(parent), list(list) {}

  friend class InterfaceSchema;
};

537 538
// -------------------------------------------------------------------

539 540 541 542 543 544
class ConstSchema: public Schema {
  // Represents a constant declaration.
  //
  // `ConstSchema` can be implicitly cast to DynamicValue to read its value.

public:
545
  inline ConstSchema(): Schema(&_::NULL_CONST_SCHEMA.defaultBrand) {}
546 547 548 549 550 551 552 553 554 555 556 557

  template <typename T>
  ReaderFor<T> as() const;
  // Read the constant's value.  This is a convenience method equivalent to casting the ConstSchema
  // to a DynamicValue and then calling its `as<T>()` method.  For dependency reasons, this method
  // is defined in <capnp/dynamic.h>, which you must #include explicitly.

  uint32_t getValueSchemaOffset() const;
  // Much like StructSchema::Field::getDefaultValueSchemaOffset(), if the constant has pointer
  // type, this gets the offset from the beginning of the constant's schema node to a pointer
  // representing the constant value.

558 559
  Type getType() const;

560
private:
561
  ConstSchema(Schema base): Schema(base) {}
562 563 564 565 566
  friend class Schema;
};

// -------------------------------------------------------------------

567 568 569 570 571 572
class Type {
public:
  struct BrandParameter {
    uint64_t scopeId;
    uint index;
  };
573 574 575
  struct ImplicitParameter {
    uint index;
  };
576 577 578 579 580 581 582 583

  inline Type();
  inline Type(schema::Type::Which primitive);
  inline Type(StructSchema schema);
  inline Type(EnumSchema schema);
  inline Type(InterfaceSchema schema);
  inline Type(ListSchema schema);
  inline Type(BrandParameter param);
584
  inline Type(ImplicitParameter param);
585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600

  template <typename T>
  inline static Type from();

  inline schema::Type::Which which() const;

  StructSchema asStruct() const;
  EnumSchema asEnum() const;
  InterfaceSchema asInterface() const;
  ListSchema asList() const;
  // Each of these methods may only be called if which() returns the corresponding type.

  kj::Maybe<BrandParameter> getBrandParameter() const;
  // Only callable if which() returns ANY_POINTER. Returns null if the type is just a regular
  // AnyPointer and not a parameter.

601 602 603 604
  kj::Maybe<ImplicitParameter> getImplicitParameter() const;
  // Only callable if which() returns ANY_POINTER. Returns null if the type is just a regular
  // AnyPointer and not a parameter. "Implicit parameters" refer to type parameters on methods.

605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631
  inline bool isVoid() const;
  inline bool isBool() const;
  inline bool isInt8() const;
  inline bool isInt16() const;
  inline bool isInt32() const;
  inline bool isInt64() const;
  inline bool isUInt8() const;
  inline bool isUInt16() const;
  inline bool isUInt32() const;
  inline bool isUInt64() const;
  inline bool isFloat32() const;
  inline bool isFloat64() const;
  inline bool isText() const;
  inline bool isData() const;
  inline bool isList() const;
  inline bool isEnum() const;
  inline bool isStruct() const;
  inline bool isInterface() const;
  inline bool isAnyPointer() const;

  bool operator==(const Type& other) const;
  inline bool operator!=(const Type& other) const { return !(*this == other); }

private:
  schema::Type::Which baseType;  // type not including applications of List()
  uint8_t listDepth;             // 0 for T, 1 for List(T), 2 for List(List(T)), ...

632 633 634 635 636
  bool isImplicitParam;
  // If true, this refers to an implicit method parameter. baseType must be ANY_POINTER, scopeId
  // must be zero, and paramIndex indicates the parameter index.

  uint16_t paramIndex;
637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656
  // If baseType is ANY_POINTER but this Type actually refers to a type parameter, this is the
  // index of the parameter among the parameters at its scope, and `scopeId` below is the type ID
  // of the scope where the parameter was defined.

  union {
    const _::RawBrandedSchema* schema;  // if type is struct, enum, interface...
    uint64_t scopeId;  // if type is AnyPointer but it's actually a type parameter...
  };

  Type(schema::Type::Which baseType, uint8_t listDepth, const _::RawBrandedSchema* schema)
      : baseType(baseType), listDepth(listDepth), schema(schema) {}

  void requireUsableAs(Type expected) const;

  friend class Schema;
  friend class ListSchema;
};

// -------------------------------------------------------------------

657 658 659 660 661 662 663
class ListSchema {
  // ListSchema is a little different because list types are not described by schema nodes.  So,
  // ListSchema doesn't subclass Schema.

public:
  ListSchema() = default;

Kenton Varda's avatar
Kenton Varda committed
664
  static ListSchema of(schema::Type::Which primitiveType);
665 666 667 668
  static ListSchema of(StructSchema elementType);
  static ListSchema of(EnumSchema elementType);
  static ListSchema of(InterfaceSchema elementType);
  static ListSchema of(ListSchema elementType);
669
  static ListSchema of(Type elementType);
670 671
  // Construct the schema for a list of the given type.

672 673 674 675 676 677 678
  static ListSchema of(schema::Type::Reader elementType, Schema context)
      KJ_DEPRECATED("Does not handle generics correctly.");
  // DEPRECATED: This method cannot correctly account for generic type parameter bindings that
  //   may apply to the input type. Instead of using this method, use a method of the Schema API
  //   that corresponds to the exact kind of dependency. For example, to get a field type, use
  //   StructSchema::Field::getType().
  //
679 680 681
  // Construct from an element type schema.  Requires a context which can handle getDependency()
  // requests for any type ID found in the schema.

682 683
  Type getElementType() const;

Kenton Varda's avatar
Kenton Varda committed
684
  inline schema::Type::Which whichElementType() const;
685 686 687 688 689 690 691 692 693 694 695
  // Get the element type's "which()".  ListSchema does not actually store a schema::Type::Reader
  // describing the element type, but if it did, this would be equivalent to calling
  // .getBody().which() on that type.

  StructSchema getStructElementType() const;
  EnumSchema getEnumElementType() const;
  InterfaceSchema getInterfaceElementType() const;
  ListSchema getListElementType() const;
  // Get the schema for complex element types.  Each of these throws an exception if the element
  // type is not of the requested kind.

696 697
  inline bool operator==(const ListSchema& other) const { return elementType == other.elementType; }
  inline bool operator!=(const ListSchema& other) const { return elementType != other.elementType; }
698

699
  template <typename T>
700
  void requireUsableAs() const;
701

702
private:
703 704 705
  Type elementType;

  inline explicit ListSchema(Type elementType): elementType(elementType) {}
706 707 708 709 710 711 712

  template <typename T>
  struct FromImpl;
  template <typename T> static inline ListSchema fromImpl() {
    return FromImpl<T>::get();
  }

713
  void requireUsableAs(ListSchema expected) const;
714

715 716 717 718 719 720
  friend class Schema;
};

// =======================================================================================
// inline implementation

Kenton Varda's avatar
Kenton Varda committed
721 722 723 724 725 726 727 728 729 730 731 732 733 734
template <> inline schema::Type::Which Schema::from<Void>() { return schema::Type::VOID; }
template <> inline schema::Type::Which Schema::from<bool>() { return schema::Type::BOOL; }
template <> inline schema::Type::Which Schema::from<int8_t>() { return schema::Type::INT8; }
template <> inline schema::Type::Which Schema::from<int16_t>() { return schema::Type::INT16; }
template <> inline schema::Type::Which Schema::from<int32_t>() { return schema::Type::INT32; }
template <> inline schema::Type::Which Schema::from<int64_t>() { return schema::Type::INT64; }
template <> inline schema::Type::Which Schema::from<uint8_t>() { return schema::Type::UINT8; }
template <> inline schema::Type::Which Schema::from<uint16_t>() { return schema::Type::UINT16; }
template <> inline schema::Type::Which Schema::from<uint32_t>() { return schema::Type::UINT32; }
template <> inline schema::Type::Which Schema::from<uint64_t>() { return schema::Type::UINT64; }
template <> inline schema::Type::Which Schema::from<float>() { return schema::Type::FLOAT32; }
template <> inline schema::Type::Which Schema::from<double>() { return schema::Type::FLOAT64; }
template <> inline schema::Type::Which Schema::from<Text>() { return schema::Type::TEXT; }
template <> inline schema::Type::Which Schema::from<Data>() { return schema::Type::DATA; }
735

736 737 738 739
inline Schema Schema::getDependency(uint64_t id) const {
  return getDependency(id, 0);
}

740 741 742 743 744 745 746 747
inline bool Schema::isBranded() const {
  return raw != &raw->generic->defaultBrand;
}

inline Schema Schema::getGeneric() const {
  return Schema(&raw->generic->defaultBrand);
}

748
template <typename T>
749
inline void Schema::requireUsableAs() const {
750
  requireUsableAs(&_::rawSchema<T>());
751 752
}

Kenton Varda's avatar
Kenton Varda committed
753 754
inline bool StructSchema::Field::operator==(const Field& other) const {
  return parent == other.parent && index == other.index;
755 756 757 758 759 760 761 762 763
}
inline bool EnumSchema::Enumerant::operator==(const Enumerant& other) const {
  return parent == other.parent && ordinal == other.ordinal;
}
inline bool InterfaceSchema::Method::operator==(const Method& other) const {
  return parent == other.parent && ordinal == other.ordinal;
}

inline ListSchema ListSchema::of(StructSchema elementType) {
764
  return ListSchema(Type(elementType));
765 766
}
inline ListSchema ListSchema::of(EnumSchema elementType) {
767
  return ListSchema(Type(elementType));
768 769
}
inline ListSchema ListSchema::of(InterfaceSchema elementType) {
770
  return ListSchema(Type(elementType));
771 772
}
inline ListSchema ListSchema::of(ListSchema elementType) {
773
  return ListSchema(Type(elementType));
774
}
775
inline ListSchema ListSchema::of(Type elementType) {
776
  return ListSchema(elementType);
777 778 779
}

inline Type ListSchema::getElementType() const {
780
  return elementType;
781
}
782

Kenton Varda's avatar
Kenton Varda committed
783
inline schema::Type::Which ListSchema::whichElementType() const {
784 785 786 787 788 789 790 791 792 793 794 795 796
  return elementType.which();
}

inline StructSchema ListSchema::getStructElementType() const {
  return elementType.asStruct();
}

inline EnumSchema ListSchema::getEnumElementType() const {
  return elementType.asEnum();
}

inline InterfaceSchema ListSchema::getInterfaceElementType() const {
  return elementType.asInterface();
797 798
}

799 800
inline ListSchema ListSchema::getListElementType() const {
  return elementType.asList();
801 802
}

803
template <typename T>
804
inline void ListSchema::requireUsableAs() const {
805 806 807 808 809
  static_assert(kind<T>() == Kind::LIST,
                "ListSchema::requireUsableAs<T>() requires T is a list type.");
  requireUsableAs(Schema::from<T>());
}

810 811 812 813
inline void ListSchema::requireUsableAs(ListSchema expected) const {
  elementType.requireUsableAs(expected.elementType);
}

814 815 816 817 818
template <typename T>
struct ListSchema::FromImpl<List<T>> {
  static inline ListSchema get() { return of(Schema::from<T>()); }
};

819
inline Type::Type(): baseType(schema::Type::VOID), listDepth(0), schema(nullptr) {}
820
inline Type::Type(schema::Type::Which primitive)
821
    : baseType(primitive), listDepth(0), isImplicitParam(false), scopeId(0) {
822 823 824 825 826 827 828
  KJ_IREQUIRE(primitive != schema::Type::STRUCT &&
              primitive != schema::Type::ENUM &&
              primitive != schema::Type::INTERFACE &&
              primitive != schema::Type::LIST);
}

inline Type::Type(StructSchema schema)
829
    : baseType(schema::Type::STRUCT), listDepth(0), schema(schema.raw) {}
830
inline Type::Type(EnumSchema schema)
831
    : baseType(schema::Type::ENUM), listDepth(0), schema(schema.raw) {}
832
inline Type::Type(InterfaceSchema schema)
833
    : baseType(schema::Type::INTERFACE), listDepth(0), schema(schema.raw) {}
834
inline Type::Type(ListSchema schema)
835 836
    : Type(schema.getElementType()) { ++listDepth; }
inline Type::Type(BrandParameter param)
837 838 839 840 841
    : baseType(schema::Type::ANY_POINTER), listDepth(0), isImplicitParam(false),
      paramIndex(param.index), scopeId(param.scopeId) {}
inline Type::Type(ImplicitParameter param)
    : baseType(schema::Type::ANY_POINTER), listDepth(0), isImplicitParam(true),
      paramIndex(param.index), scopeId(0) {}
842

843 844 845 846 847 848 849
inline schema::Type::Which Type::which() const {
  return listDepth > 0 ? schema::Type::LIST : baseType;
}

template <typename T>
inline Type Type::from() { return Type(Schema::from<T>()); }

850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872
inline bool Type::isVoid   () const { return baseType == schema::Type::VOID     && listDepth == 0; }
inline bool Type::isBool   () const { return baseType == schema::Type::BOOL     && listDepth == 0; }
inline bool Type::isInt8   () const { return baseType == schema::Type::INT8     && listDepth == 0; }
inline bool Type::isInt16  () const { return baseType == schema::Type::INT16    && listDepth == 0; }
inline bool Type::isInt32  () const { return baseType == schema::Type::INT32    && listDepth == 0; }
inline bool Type::isInt64  () const { return baseType == schema::Type::INT64    && listDepth == 0; }
inline bool Type::isUInt8  () const { return baseType == schema::Type::UINT8    && listDepth == 0; }
inline bool Type::isUInt16 () const { return baseType == schema::Type::UINT16   && listDepth == 0; }
inline bool Type::isUInt32 () const { return baseType == schema::Type::UINT32   && listDepth == 0; }
inline bool Type::isUInt64 () const { return baseType == schema::Type::UINT64   && listDepth == 0; }
inline bool Type::isFloat32() const { return baseType == schema::Type::FLOAT32  && listDepth == 0; }
inline bool Type::isFloat64() const { return baseType == schema::Type::FLOAT64  && listDepth == 0; }
inline bool Type::isText   () const { return baseType == schema::Type::TEXT     && listDepth == 0; }
inline bool Type::isData   () const { return baseType == schema::Type::DATA     && listDepth == 0; }
inline bool Type::isList   () const { return listDepth > 0; }
inline bool Type::isEnum   () const { return baseType == schema::Type::ENUM     && listDepth == 0; }
inline bool Type::isStruct () const { return baseType == schema::Type::STRUCT   && listDepth == 0; }
inline bool Type::isInterface() const {
  return baseType == schema::Type::INTERFACE && listDepth == 0;
}
inline bool Type::isAnyPointer() const {
  return baseType == schema::Type::ANY_POINTER && listDepth == 0;
}
873

874
}  // namespace capnp
875

Kenton Varda's avatar
Kenton Varda committed
876
#endif  // CAPNP_SCHEMA_H_