Add is, | and & to flagsets

Author: nicolasstucki

compiler/src/dotty/tools/dotc/tastyreflect/FlagSet.scala

@@ -3,7 +3,7 @@ package dotty.tools.dotc.tastyreflect
 import dotty.tools.dotc.core.Flags
 import dotty.tools.dotc.core.Flags._
 
-class FlagSet(flags: Flags.FlagSet) extends scala.tasty.reflect.FlagSet {
+class FlagSet(val flags: Flags.FlagSet) extends scala.tasty.reflect.FlagSet {
 
   def isProtected: Boolean = flags.is(Protected)
   def isAbstract: Boolean = flags.is(Abstract)

compiler/src/dotty/tools/dotc/tastyreflect/FlagsOpsImpl.scala

@@ -3,8 +3,16 @@ package tastyreflect
 
 import dotty.tools.dotc.core.Decorators._
 
+import scala.tasty.reflect
+
 trait FlagsOpsImpl extends scala.tasty.reflect.FlagsOps with CoreImpl {
 
+  def FlagsDeco(flagSet: scala.tasty.reflect.FlagSet): FlagsAPI = new FlagsAPI {
+    def is(that: reflect.FlagSet): Boolean = flagSet.asInstanceOf[FlagSet].flags is that.asInstanceOf[FlagSet].flags
+    def |(that: reflect.FlagSet): reflect.FlagSet = new FlagSet(flagSet.asInstanceOf[FlagSet].flags | that.asInstanceOf[FlagSet].flags)
+    def &(that: reflect.FlagSet): reflect.FlagSet = new FlagSet(flagSet.asInstanceOf[FlagSet].flags & that.asInstanceOf[FlagSet].flags)
+  }
+
   object Flags extends FlagsModule {
     def Protected: FlagSet = new FlagSet(core.Flags.Protected)
     def Abstract: FlagSet = new FlagSet(core.Flags.Abstract)

library/src/scala/tasty/reflect/FlagsOps.scala

@@ -2,6 +2,16 @@ package scala.tasty.reflect
 
 trait FlagsOps extends Core {
 
+  trait FlagsAPI {
+    /** Is the given flag set a subset of this flag sets */
+    def is(flagSet: FlagSet): Boolean
+    /** Union of the two flag sets */
+    def |(flagSet: FlagSet): FlagSet
+    /** Intersection of the two flag sets */
+    def &(flagSet: FlagSet): FlagSet
+  }
+  implicit def FlagsDeco(flagSet: FlagSet): FlagsAPI
+
   val Flags: FlagsModule
   abstract class FlagsModule {
     /** Is this symbol `protected` */