patrick/WoofWare.Whippet
1
0
Fork 0
mirror of https://github.com/Smaug123/WoofWare.Whippet synced 2025-10-08 09:18:39 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add more helpers to Fantomas library (#7)

Browse Source
This commit is contained in:
Patrick Stevens
2024-10-06 19:45:13 +01:00
committed by GitHub
parent 8834d885de
commit f4da4cd2a1
9 changed files with 762 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
WoofWare.Whippet.Fantomas/SynTypeDefn.fs
View File

@@ -1,5 +1,6 @@
namespace WoofWare.Whippet.Fantomas
open System
open Fantomas.FCS.Syntax
open Fantomas.FCS.SyntaxTrivia
open Fantomas.FCS.Text.Range
@@ -8,6 +9,24 @@ open Fantomas.FCS.Text.Range
[<RequireQualifiedAccess>]
module SynTypeDefn =
/// Returns true iff the input type is an F# record.
let isRecord (SynTypeDefn.SynTypeDefn (_, repr, _, _, _, _)) : bool =
match repr with
| SynTypeDefnRepr.Simple (SynTypeDefnSimpleRepr.Record _, _) -> true
| _ -> false
/// Returns true iff the input type is a discriminated union.
let isDu (SynTypeDefn.SynTypeDefn (_, repr, _, _, _, _)) : bool =
match repr with
| SynTypeDefnRepr.Simple (SynTypeDefnSimpleRepr.Union _, _) -> true
| _ -> false
/// Returns true iff the input type is an enum: `type Foo = | X = 1 ...`
let isEnum (SynTypeDefn.SynTypeDefn (_, repr, _, _, _, _)) : bool =
match repr with
| SynTypeDefnRepr.Simple (SynTypeDefnSimpleRepr.Enum _, _) -> true
| _ -> false
/// Build a `SynTypeDefn` from its components:
/// the "front matter" `SynComponentInfo`, and the "body" `SynTypeDefnRepr`.
let inline create (componentInfo : SynComponentInfo) (repr : SynTypeDefnRepr) : SynTypeDefn =
@@ -36,6 +55,9 @@ module SynTypeDefn =
| SynTypeDefn (SynComponentInfo.SynComponentInfo (_, _, _, id, _, _, _, _), _, _, _, _, _) -> id
/// Select from this type definition the first attribute with the given name: `[<Foo>] type Blah = ...`
///
/// Pass e.g. `typeof<MyAttribute>.Name` as the name for maximum type safety.
/// Don't fully qualify the input; this should basically just be an Ident rather than a LongIdent.
let getAttribute (attrName : string) (defn : SynTypeDefn) : SynAttribute option =
match defn with
| SynTypeDefn (SynComponentInfo.SynComponentInfo (attrs, _, _, _, _, _, _, _), _, _, _, _, _) ->