Remove use of JuliaSyntax DOTOP_FLAG and is_dotted

These have been removed upstream on the main branch and replaced with a
small number of new `Kind`s.

Author: mlechu

src/JuliaLowering.jl

@@ -13,7 +13,7 @@ using JuliaSyntax: highlight, Kind, @KSet_str
 using JuliaSyntax: is_leaf, children, numchildren, head, kind, flags, has_flags, numeric_flags
 using JuliaSyntax: filename, first_byte, last_byte, byte_range, sourcefile, source_location, span, sourcetext
 
-using JuliaSyntax: is_literal, is_number, is_operator, is_prec_assignment, is_prefix_call, is_infix_op_call, is_postfix_op_call, is_error, is_dotted
+using JuliaSyntax: is_literal, is_number, is_operator, is_prec_assignment, is_prefix_call, is_infix_op_call, is_postfix_op_call, is_error
 
 _include("kinds.jl")
 _register_kinds()

src/desugaring.jl

@@ -690,10 +690,10 @@ function expand_dotcall(ctx, ex)
         ]
     elseif k == K"comparison"
         expand_dotcall(ctx, expand_compare_chain(ctx, ex))
-    elseif (k == K"&&" || k == K"||") && is_dotted(ex)
+    elseif k == K".&&" || k == K".||"
         @ast ctx ex [K"call"
             "broadcasted"::K"top"
-            (k == K"&&" ? "andand" : "oror")::K"top"
+            (k == K".&&" ? "andand" : "oror")::K"top"
             (expand_dotcall(ctx, arg) for arg in children(ex))...
         ]
     else
@@ -702,8 +702,7 @@ function expand_dotcall(ctx, ex)
 end
 
 function expand_fuse_broadcast(ctx, ex)
-    if kind(ex) == K"="
-        @assert is_dotted(ex)
+    if kind(ex) == K".=" || kind(ex) == K".op="
         @chk numchildren(ex) == 2
         lhs = ex[1]
         kl = kind(lhs)
@@ -1293,7 +1292,7 @@ end
 
 function expand_update_operator(ctx, ex)
     k = kind(ex)
-    dotted = is_dotted(ex)
+    dotted = k == K".op="
 
     @chk numchildren(ex) == 3
     lhs = ex[1]
@@ -1335,7 +1334,7 @@ function expand_update_operator(ctx, ex)
 
     @ast ctx ex [K"block"
         stmts...
-        [K"="(syntax_flags=(dotted ? JuliaSyntax.DOTOP_FLAG : nothing))
+        [(dotted ? K".=" : K"=")
             lhs
             [(dotted ? K"dotcall" : K"call")
                 op
@@ -4172,7 +4171,7 @@ function expand_forms_2(ctx::DesugaringContext, ex::SyntaxTree, docs=nothing)
         throw(LoweringError(ex, "unimplemented or unsupported atomic declaration"))
     elseif k == K"call"
         expand_call(ctx, ex)
-    elseif k == K"dotcall" || ((k == K"&&" || k == K"||") && is_dotted(ex))
+    elseif k == K"dotcall" || k == K".&&" || k == K".||" || k == K".="
         expand_forms_2(ctx, expand_fuse_broadcast(ctx, ex))
     elseif k == K"."
         expand_forms_2(ctx, expand_dot(ctx, ex))
@@ -4208,14 +4207,10 @@ function expand_forms_2(ctx::DesugaringContext, ex::SyntaxTree, docs=nothing)
             adopt_scope(string(k)::K"Identifier", ex)
             children(ex)...
         ])
-    elseif k == K"op="
+    elseif k == K"op=" || k == K".op="
         expand_forms_2(ctx, expand_update_operator(ctx, ex))
     elseif k == K"="
-        if is_dotted(ex)
-            expand_forms_2(ctx, expand_fuse_broadcast(ctx, ex))
-        else
-            expand_assignment(ctx, ex)
-        end
+        expand_assignment(ctx, ex)
     elseif k == K"break"
         numchildren(ex) > 0 ? ex :
             @ast ctx ex [K"break" "loop_exit"::K"symbolic_label"]

src/syntax_graph.jl

@@ -137,17 +137,20 @@ function Base.getproperty(graph::SyntaxGraph, name::Symbol)
 end
 
 function sethead!(graph, id::NodeId, h::JuliaSyntax.SyntaxHead)
-    graph.kind[id] = kind(h)
-    f = flags(h)
-    if f != 0
-        graph.syntax_flags[id] = f
-    end
+    sethead!(graph, id, kind(h))
+    setflags!(graph, id, flags(h))
 end
 
 function sethead!(graph, id::NodeId, k::Kind)
     graph.kind[id] = k
 end
 
+function setflags!(graph, id::NodeId, f::UInt16)
+    if f != 0
+        graph.syntax_flags[id] = f
+    end
+end
+
 function _convert_nodes(graph::SyntaxGraph, node::SyntaxNode)
     id = newnode!(graph)
     sethead!(graph, id, head(node))
@@ -307,6 +310,7 @@ JuliaSyntax.source_line(src::LineNumberNode) = src.line
 # The follow somewhat strange cases are for where LineNumberNode is standing in
 # for SourceFile because we've only got Expr-based provenance info
 JuliaSyntax.sourcefile(src::LineNumberNode) = src
+JuliaSyntax.sourcetext(src::LineNumberNode) = SubString("")
 JuliaSyntax.source_location(src::LineNumberNode, byte_index::Integer) = (src.line, 0)
 JuliaSyntax.source_location(::Type{LineNumberNode}, src::LineNumberNode, byte_index::Integer) = src
 JuliaSyntax.filename(src::LineNumberNode) = string(src.file)
@@ -536,13 +540,11 @@ end
 JuliaSyntax.sourcefile(ex::SyntaxTree) = sourcefile(sourceref(ex))
 JuliaSyntax.byte_range(ex::SyntaxTree) = byte_range(sourceref(ex))
 
-function JuliaSyntax._expr_leaf_val(ex::SyntaxTree)
+function JuliaSyntax._expr_leaf_val(ex::SyntaxTree, _...)
     name = get(ex, :name_val, nothing)
-    if !isnothing(name)
-        Symbol(name)
-    else
-        ex.value
-    end
+    !isnothing(name) && return Symbol(name)
+    name = get(ex, :value, nothing)
+    return name
 end
 
 Base.Expr(ex::SyntaxTree) = JuliaSyntax.to_expr(ex)
@@ -603,7 +605,7 @@ macro SyntaxTree(ex_old)
         throw(ArgumentError("@SyntaxTree expects a `quote` block or `:`-quoted expression"))
     end
     # 2. Re-parse the current source file as SyntaxTree instead
-    fname = String(__source__.file)
+    fname = isnothing(__source__.file) ? error("No current file") : String(__source__.file)
     if occursin(r"REPL\[\d+\]", fname)
         # Assume we should look at last history entry in REPL
         try