summaryrefslogtreecommitdiff
path: root/Jennisys/TypeChecker.fs
blob: c9e5f308f970074e71ba6a5b4d3b6ba53409931a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
module TypeChecker

open Ast
open System.Collections.Generic

let GetClass name decls =
  match decls |> List.tryFind (function Class(_,_,_) -> true | _ -> false) with
    | Some(cl) -> cl
    | None -> Class(name,[],[])

let GetModel name decls =
  match decls |> List.tryFind (function Model(_,_,_,_,_) -> true | _ -> false) with
    | Some(m) -> m
    | None -> Model(name,[],[],[],IdLiteral("true"))

let GetCode name decls =
  match decls |> List.tryFind (function Code(_,_) -> true | _ -> false) with
    | Some(c) -> c
    | None -> Code(name,[])

let TypeCheck prog =
  match prog with
  | SProgram(decls) ->
      let componentNames = decls |> List.choose (function Class(name,_,_) -> Some(name) | _ -> None)
      let clist = componentNames |> List.map (fun name -> Component(GetClass name decls, GetModel name decls, GetCode name decls))
      Some(Program(clist))