Packages

spectra

0.5.0
0.13.4 0.13.3 0.13.2 0.13.1 0.13.0 0.12.1 0.12.0 0.11.4 0.11.3 0.11.2 0.11.1 0.11.0 0.10.0 0.9.4 0.9.3 0.9.2 0.9.1 0.9.0 0.8.2 0.8.1 0.8.0 0.7.0 0.6.0 0.5.1 0.5.0 0.4.0 0.3.2 0.3.1 0.3.0 0.2.0 0.1.10 0.1.9 0.1.8 0.1.7 0.1.6 0.1.5 0.1.4 0.1.3 0.1.2 0.1.1 0.1.0

Type safe data (de) serialization.

HexDocs

Current section

Files

Jump to
Readme
41 Versions
0 Dependencies
Dependants
Files
Activity
spectra src spectra_type_info.erl
Raw

src/spectra_type_info.erl 

-module(spectra_type_info).



-include("../include/spectra_internal.hrl").



-ignore_xref([find_function/3]).



-export([new/0]).

-export([add_type/4, find_type/3, get_type/3]).

-export([add_record/3, find_record/2, get_record/2]).

-export([add_function/4, find_function/3]).



-export_type([type_info/0, type_key/0, function_key/0]).



-type type_info() :: #type_info{}.

-type type_key() :: {Name :: atom(), Arity :: arity()}.

-type function_key() :: {Name :: atom(), Arity :: arity()}.



-spec new() -> type_info().

new() ->

    #type_info{}.



-spec add_type(type_info(), atom(), arity(), spectra:sp_type()) -> type_info().

add_type(#type_info{types = Types} = TypeInfo, Name, Arity, Type) ->

    TypeInfo#type_info{types = Types#{{Name, Arity} => Type}}.



-spec find_type(type_info(), atom(), arity()) -> {ok, spectra:sp_type()} | error.

find_type(#type_info{types = Types}, Name, Arity) ->

    maps:find({Name, Arity}, Types).



-spec get_type(type_info(), atom(), arity()) -> spectra:sp_type().

get_type(#type_info{types = Types}, Name, Arity) ->

    case Types of

        #{{Name, Arity} := Type} ->

            Type;

        #{} ->

            erlang:error({type_not_found, Name, Arity})

    end.



-spec add_record(type_info(), atom(), #sp_rec{}) -> type_info().

add_record(#type_info{records = Records} = TypeInfo, Name, Record) ->

    TypeInfo#type_info{records = Records#{Name => Record}}.



-spec find_record(type_info(), atom()) -> {ok, #sp_rec{}} | error.

find_record(#type_info{records = Records}, Name) ->

    maps:find(Name, Records).



-spec get_record(type_info(), atom()) -> #sp_rec{}.

get_record(#type_info{records = Records}, Name) ->

    case Records of

        #{Name := Record} ->

            Record;

        #{} ->

            erlang:error({record_not_found, Name})

    end.



-spec add_function(type_info(), atom(), arity(), [spectra:sp_function_spec()]) ->

    type_info().

add_function(#type_info{functions = Functions} = TypeInfo, Name, Arity, FuncSpec) ->

    TypeInfo#type_info{functions = Functions#{{Name, Arity} => FuncSpec}}.



-spec find_function(type_info(), atom(), arity()) ->

    {ok, [spectra:sp_function_spec()]} | error.

find_function(#type_info{functions = Functions}, Name, Arity) ->

    maps:find({Name, Arity}, Functions).