moar functional

[scpubgit/TenDotTcl.git] / json.tcl

namespace eval ten::json {

  namespace eval deparse {

    variable quotes [
      list "\"" "\\\"" / \\/ \\ \\\\ \b \\b \f \\f \n \\n \r \\r \t \\t
    ]

    variable indent
    variable indentBy 0
    variable indentIncr
    variable nl

    proc indent_one {} {
      variable indentIncr
      variable indentBy [ expr $indentBy + $indentIncr ]
      variable indent [ string repeat " " $indentBy ]
    }

    proc outdent_one {} {
      variable indentIncr
      variable indentBy [ expr $indentBy - $indentIncr ]
      variable indent [ string repeat " " $indentBy ]
    }

    proc str {str} {
      variable quotes
      return \"[ string map $quotes $str ]\"
    }

    proc num {num} {
      return $num
    }

    proc list {args} {
      variable indent
      variable nl
      set out \[
      indent_one
      foreach el [lrange $args 0 end] {
        append out $nl$indent
        append out [ deparse $el ],
      }
      outdent_one
      append out $nl$indent\]
      return $out
    }

    proc obj {args} {
      variable indent
      variable nl
      set out \{
      indent_one
      dict for {k v} $args {
        append out $nl$indent[ str $k ]:\ [ deparse $v ],
      }
      outdent_one
      append out $nl$indent\}
    }

    proc deparse {data} {
      switch -regexp [lindex $data 0] {
        ^true|false|null$  { lindex $data 0 }
        ^num|str|obj|list$ { eval $data }
        default { error [ concat "Invalid JSON type " [lindex $data 0 0] ] }
      }
    }
  }

  proc deparse_json {data {indentIncr 2}} {
    set deparse::indentBy 0
    set deparse::indentIncr $indentIncr
    if [expr $indentIncr == 0] {
      set deparse::nl ""
    } else {
      set deparse::nl "\n"
    }
    deparse::deparse $data
  }

  namespace eval tclify {

    proc str {str} { return $str }

    proc num {num} { return $num }

    proc obj {args} {
      return $args
    }

    proc tclify {data} {
      switch -regexp [lindex $data 0] {
        ^true|false|null$ { uplevel 1 return [lindex $data 0] }
        ^num|str|obj|list$ {}
        default { error [ concat "Invalid JSON type " [lindex $data 0 0] ] }
      }
      eval $data
    }
  }

  proc tclify_json {data} {
    tclify::tclify $data
  }

  namespace eval parse {

    variable json

    proc eat_comma {} {
      variable json
      set json [string trimleft $json]
      if {[string index $json 0] eq ","} {
        set json [string range $json 1 end]
      }
    }

    proc parse_list {} {
      variable json
      set json [ string range $json 1 end ]
      set tcl {list}
      while {"$json" ne ""} {
        if {[string index $json 0] eq "]"} {
          return $tcl
        }
        lappend tcl [ parse ]
        eat_comma
      }
      error "Ran out of JSON. Confused now."
    }

    proc parse_str {} {
      variable json
      # like Text::Balanced except ugly (borrowed from tcvJSON's code)
      set reStr {(?:(?:\")(?:[^\\\"]*(?:\\.[^\\\"]*)*)(?:\"))}
      if {![regexp $reStr $json string]} {
        error [ concat "Invalid string: " [string range $json 0 32]
      }
      set json [string range $json [string length $string] end]
      # chop off outer ""s and substitute backslashes
      # This does more than the RFC-specified backslash sequences,
      # but it does cover them all
      list str [subst -nocommand -novariable [string range $string 1 end-1]]
    }

    proc parse {} {
      variable json
      set json [string trimleft $json]
      if {$json eq ""} {
        return
      }
      switch -regexp [string index $json 0] {
        {\{} { parse_obj }
        {\[} { parse_list }
        {\"} { parse_str }

        {[-0-9]} { parse_number }

        default { error "argh" }
      }
    }
  }

  proc parse_json {json} {
    set parse::json $json
    parse::parse
  }
}

set ex_json { list {str foo} {num 0} {obj __remote_object__ {str 512}} {null} }

set jtext {
  [
    "foo",
    0,
    {
      "__remote_object__": "512",
    },
    null,
  ]
}

puts [ ten::json::deparse_json $ex_json 2 ]

dict for {k v} [ ten::json::tclify_json [
  lindex [ ten::json::tclify_json $ex_json ] 2
] ] { puts "$k: $v" }

#puts [ ten::json::parse_json $jtext ]

puts [ ten::json::parse_json {["foo"]} ]