model: add nitunit for model_json

[nit.git] / src / model / model_json.nit

# This file is part of NIT ( http://www.nitlanguage.org ).
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# Make model entities Jsonable.
#
# To avoid cycles, every reference from a MEntity to another is replaced by a
# MEntityRef.
#
# How subobjects are retrieved using the MEntityRef is the responsability of the
# client. Json objects can be returned as this or inflated with concrete objet
# rather than the refs.
#
# TODO consider serialization module?
module model_json

import model::model_collect
import json::static
import json::serialization_write
import loader

# A reference to another mentity.
class MEntityRef
        super MEntity

        # MEntity to link to.
        var mentity: MEntity

        redef fun core_serialize_to(v) do
                v.serialize_attribute("full_name", mentity.full_name)
        end
end

redef class MEntity
        super Jsonable
        serialize

        redef fun core_serialize_to(v) do
                v.serialize_attribute("name", name)
                v.serialize_attribute("class_name", class_name)
                v.serialize_attribute("full_name", full_name)
                v.serialize_attribute("mdoc", mdoc_or_fallback)
                v.serialize_attribute("visibility", visibility.to_s)
                v.serialize_attribute("modifiers", collect_modifiers)
                v.serialize_attribute("location", location)
        end

        # Serialize the full version of `self` to JSON
        #
        # See: `FullJsonSerializer`
        fun serialize_to_full_json(plain, pretty: nullable Bool): String do
                var stream = new StringWriter
                var serializer = new FullJsonSerializer(stream)
                serializer.plain_json = plain or else false
                serializer.pretty_json = pretty or else false
                serializer.serialize self
                stream.close
                return stream.to_s
        end

        # Return the full json representation of `self` with references.
        #
        # By default, every reference to another MEntity is replaced by a pointer
        # to the MEntity::json_id.
        # Use this method to obtain a full object with mentities instead of pointers.
        fun to_full_json: String do return serialize_to_full_json(plain=true)

        # Same as `to_full_json` but with pretty json.
        fun to_pretty_full_json: String do return serialize_to_full_json(plain=true, pretty=true)
end

redef class MDoc
        super Jsonable
        serialize

        redef fun core_serialize_to(v) do
                super
                v.serialize_attribute("content", content.join("\n"))
                v.serialize_attribute("location", location)
        end
end

redef class Location
        super Jsonable
        serialize

        redef fun core_serialize_to(v) do
                v.serialize_attribute("column_end", column_end)
                v.serialize_attribute("column_start", column_start)
                v.serialize_attribute("line_end", line_end)
                v.serialize_attribute("line_start", line_start)
                var file = self.file
                if file != null then
                        v.serialize_attribute("file", file.filename)
                end
        end
end

redef class MPackage
        redef fun core_serialize_to(v) do
                super
                if v isa FullJsonSerializer then
                        v.serialize_attribute("root", to_mentity_ref(root))
                        v.serialize_attribute("mgroups", to_mentity_refs(mgroups))
                        var ini = self.ini
                        if ini != null then v.serialize_attribute("ini", ini.to_map)
                end
        end
end

redef class MGroup
        redef fun core_serialize_to(v) do
                super
                if v isa FullJsonSerializer then
                        v.serialize_attribute("is_root", is_root)
                        v.serialize_attribute("mpackage", to_mentity_ref(mpackage))
                        v.serialize_attribute("default_mmodule", to_mentity_ref(default_mmodule))
                        v.serialize_attribute("parent", to_mentity_ref(parent))
                        v.serialize_attribute("mmodules", to_mentity_refs(mmodules))
                        v.serialize_attribute("mgroups", to_mentity_refs(in_nesting.direct_smallers))
                end
        end
end

redef class MModule
        redef fun core_serialize_to(v) do
                super
                if v isa FullJsonSerializer then
                        var view = private_view
                        v.serialize_attribute("mpackage", to_mentity_ref(mpackage))
                        v.serialize_attribute("mgroup", to_mentity_ref(mgroup))
                        v.serialize_attribute("intro_mclasses", to_mentity_refs(intro_mclasses))
                        v.serialize_attribute("mclassdefs", to_mentity_refs(mclassdefs))
                        v.serialize_attribute("intro_mclassdefs", to_mentity_refs(collect_intro_mclassdefs(view)))
                        v.serialize_attribute("redef_mclassdefs", to_mentity_refs(collect_redef_mclassdefs(view)))
                        v.serialize_attribute("imports", to_mentity_refs(in_importation.direct_greaters))
                end
        end
end

redef class MClass
        redef fun core_serialize_to(v) do
                super
                v.serialize_attribute("mparameters", mparameters)
                if v isa FullJsonSerializer then
                        var view = private_view
                        v.serialize_attribute("intro", to_mentity_ref(intro))
                        v.serialize_attribute("intro_mmodule", to_mentity_ref(intro_mmodule))
                        v.serialize_attribute("mpackage", to_mentity_ref(intro_mmodule.mpackage))
                        v.serialize_attribute("mclassdefs", to_mentity_refs(mclassdefs))
                        v.serialize_attribute("all_mproperties", to_mentity_refs(collect_accessible_mproperties(view)))
                        v.serialize_attribute("intro_mproperties", to_mentity_refs(collect_intro_mproperties(view)))
                        v.serialize_attribute("redef_mproperties", to_mentity_refs(collect_redef_mproperties(view)))
                        v.serialize_attribute("parents", to_mentity_refs(collect_parents(view)))
                end
        end
end

redef class MClassDef
        redef fun core_serialize_to(v) do
                super
                v.serialize_attribute("is_intro", is_intro)
                v.serialize_attribute("mparameters", mclass.mparameters)
                if v isa FullJsonSerializer then
                        var view = private_view
                        v.serialize_attribute("mmodule", to_mentity_ref(mmodule))
                        v.serialize_attribute("mclass", to_mentity_ref(mclass))
                        v.serialize_attribute("mpropdefs", to_mentity_refs(mpropdefs))
                        v.serialize_attribute("intro_mproperties", to_mentity_refs(intro_mproperties))
                        v.serialize_attribute("intro", to_mentity_ref(mclass.intro))
                        v.serialize_attribute("mpackage", to_mentity_ref(mmodule.mpackage))
                        v.serialize_attribute("intro_mpropdefs", to_mentity_refs(collect_intro_mpropdefs(view)))
                        v.serialize_attribute("redef_mpropdefs", to_mentity_refs(collect_redef_mpropdefs(view)))
                end
        end
end

redef class MProperty
        redef fun core_serialize_to(v) do
                super
                if v isa FullJsonSerializer then
                        v.serialize_attribute("intro", to_mentity_ref(intro))
                        v.serialize_attribute("intro_mclassdef", to_mentity_ref(intro_mclassdef))
                        v.serialize_attribute("mpropdefs", to_mentity_refs(mpropdefs))
                        v.serialize_attribute("intro_mclass", to_mentity_ref(intro_mclassdef.mclass))
                        v.serialize_attribute("mpackage", to_mentity_ref(intro_mclassdef.mmodule.mpackage))
                end
        end
end

redef class MMethod
        redef fun core_serialize_to(v) do
                super
                v.serialize_attribute("is_init", is_init)
                v.serialize_attribute("msignature", intro.msignature)
        end
end

redef class MAttribute
        redef fun core_serialize_to(v) do
                super
                v.serialize_attribute("static_mtype", to_mentity_ref(intro.static_mtype))
        end
end

redef class MVirtualTypeProp
        redef fun core_serialize_to(v) do
                super
                v.serialize_attribute("mvirtualtype", to_mentity_ref(mvirtualtype))
                v.serialize_attribute("bound", to_mentity_ref(intro.bound))
        end
end

redef class MPropDef
        redef fun core_serialize_to(v) do
                super
                v.serialize_attribute("is_intro", is_intro)
                if v isa FullJsonSerializer then
                        v.serialize_attribute("mclassdef", to_mentity_ref(mclassdef))
                        v.serialize_attribute("mproperty", to_mentity_ref(mproperty))
                        v.serialize_attribute("intro", to_mentity_ref(mproperty.intro))
                        v.serialize_attribute("intro_mclassdef", to_mentity_ref(mproperty.intro.mclassdef))
                        v.serialize_attribute("mmodule", to_mentity_ref(mclassdef.mmodule))
                        v.serialize_attribute("mgroup", to_mentity_ref(mclassdef.mmodule.mgroup))
                        v.serialize_attribute("mpackage", to_mentity_ref(mclassdef.mmodule.mpackage))
                end
        end
end

redef class MMethodDef
        redef fun core_serialize_to(v) do
                super
                v.serialize_attribute("msignature", msignature)
        end
end

redef class MAttributeDef
        redef fun core_serialize_to(v) do
                super
                v.serialize_attribute("static_mtype", to_mentity_ref(static_mtype))
        end
end

redef class MVirtualTypeDef
        redef fun core_serialize_to(v) do
                super
                v.serialize_attribute("bound", to_mentity_ref(bound))
                v.serialize_attribute("is_fixed", is_fixed)
        end
end

redef class MSignature
        redef fun core_serialize_to(v) do
                v.serialize_attribute("arity", arity)
                v.serialize_attribute("mparams", mparameters)
                v.serialize_attribute("return_mtype", to_mentity_ref(return_mtype))
                v.serialize_attribute("vararg_rank", vararg_rank)
        end
end

redef class MParameterType
        redef fun core_serialize_to(v) do
                v.serialize_attribute("name", name)
                v.serialize_attribute("rank", rank)
                v.serialize_attribute("mtype", to_mentity_ref(mclass.intro.bound_mtype.arguments[rank]))
        end
end

redef class MParameter
        redef fun core_serialize_to(v) do
                v.serialize_attribute("is_vararg", is_vararg)
                v.serialize_attribute("name", name)
                v.serialize_attribute("mtype", to_mentity_ref(mtype))
        end
end

# Create a ref to a `mentity`.
fun to_mentity_ref(mentity: nullable MEntity): nullable MEntityRef do
        if mentity == null then return null
        return new MEntityRef(mentity)
end

# Return a collection of `mentities` as a JsonArray of MEntityRefs.
fun to_mentity_refs(mentities: Collection[MEntity]): Array[MEntityRef] do
        var array = new Array[MEntityRef]
        for mentity in mentities do
                var ref = to_mentity_ref(mentity)
                if ref == null then continue
                array.add ref
        end
        return array
end

# Use the FullJsonSerializer to generate the full json representation of a MEntity.
#
# See MEntity::to_full_json.
class FullJsonSerializer
        super JsonSerializer
end