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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
package org.shedlang.compiler;
import org.shedlang.compiler.parser.Parser;
import org.shedlang.compiler.sources.Sources;
import org.shedlang.compiler.stackir.InstructionSerializer;
import org.shedlang.compiler.stackir.StackCompiler;
import org.shedlang.compiler.typechecker.TypeChecker;
import org.shedlang.compiler.typechecker.TypeCheckerContext;
import org.shedlang.compiler.types.Effects;
import org.shedlang.compiler.types.FunctionType;
import org.shedlang.compiler.types.Types;
import org.shedlang.compiler.util.P;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.nio.file.FileVisitResult;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.SimpleFileVisitor;
import java.nio.file.attribute.BasicFileAttributes;
import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
import java.util.Map;
import static org.shedlang.compiler.parser.Tokeniser.tokenise;
public class ShedCompiler {
public void compile(Path projectRoot) throws IOException {
var sourceRoot = projectRoot.resolve("src");
var sourceFilePaths = collectSourceFilePaths(sourceRoot);
for (var sourceFilePath : sourceFilePaths) {
System.out.println(sourceFilePath);
var sourceContents = Files.readString(sourceFilePath, StandardCharsets.UTF_8);
var source = Sources.file(sourceFilePath.toString(), sourceContents);
var tokens = tokenise(source);
var untypedNode = Parser.parseNamespace(tokens);
var context = TypeCheckerContext.EMPTY
.enter(P.map(
Map.entry("Io", Types.effectType(Effects.IO)),
Map.entry("Unit", Types.metaType(Types.UNIT)),
Map.entry("print", new FunctionType(
List.of(Types.STRING),
List.of(),
Types.UNIT
))
));
var typedNode = TypeChecker.typeCheckNamespace(untypedNode, context);
var instructions = StackCompiler.compileNamespace(typedNode);
var builder = new StringBuilder();
InstructionSerializer.serializeInstructions(instructions, builder);
var instructionsString = builder.toString();
System.out.println(instructionsString);
System.out.println();
}
}
private List<Path> collectSourceFilePaths(Path sourceRoot) throws IOException {
var paths = new ArrayList<Path>();
Files.walkFileTree(sourceRoot, new SimpleFileVisitor<Path>() {
@Override
public FileVisitResult visitFile(
Path file,
BasicFileAttributes attrs
) {
if (file.getFileName().toString().endsWith(".shed")) {
paths.add(file);
}
return FileVisitResult.CONTINUE;
}
@Override
public FileVisitResult postVisitDirectory(
Path dir,
IOException exc
) {
return FileVisitResult.CONTINUE;
}
});
paths.sort(Comparator.naturalOrder());
return paths;
}
}
|
