tracker

1.9 KB

1. # refparams.py <app dir> [--apply] [files...]: lambda parameters that a body only READS get `&` (hybriel#48: a lambda parameter
2. # copies its argument since 4371b7aa; before it was a reference — `&` restores exactly that for read-only params, without the copy).
3. # Skipped: params assigned (p = …), field-assigned (p.x = / p[k] =), list-mutated (p.push …), aliased (let x = p), defaults.
4. import re,sys,glob,os
5. os.chdir(sys.argv[1]); apply='--apply' in sys.argv
6. files=[a for a in sys.argv[2:] if a!='--apply'] or (glob.glob('*.hl')+glob.glob('lib/*.hl')+glob.glob('components/*.hl')+glob.glob('tools/*.hl'))
7. total=0; changed_files=0
8. for f in sorted(files):
9. src=open(f,encoding='utf-8').read(); out=[]; last=0; n=0
10. for m in re.finditer(r'\(([^()]*)\)(\s*=>\s*\{)',src):
11. params=[p.strip() for p in m.group(1).split(',')]
12. if params==[''] : continue
13. if not all(re.match(r'^&?\w+$',p) for p in params): continue
14. i=m.end(); depth=1
15. while depth and i<len(src):
16. c=src[i]; depth += (c=='{') - (c=='}'); i+=1
17. body=src[m.end():i-1]
18. new=[]
19. for p in params:
20. if p.startswith('&'): new.append(p); continue
21. bad = re.search(r'(?<![\w.])%s\s*=(?!=)'%p,body) or re.search(r'(?<![\w.])%s(?:\.\w+|\[[^\]]*\])+\s*=(?!=)'%p,body) \
22. or re.search(r'(?<![\w.])%s(?:\.\w+|\[[^\]]*\])*\.(push|pop|shift|unshift|splice|sort|reverse)\('%p,body) \
23. or re.search(r'=\s*%s\s*$'%p,body,re.M) or re.search(r'=\s*%s\s*[;}]'%p,body) or re.search(r'(?<![\w.])%s\s*(\+\+|--)'%p,body)
24. new.append(p if bad else '&'+p)
25. if new!=params:
26. n+=1
27. out.append(src[last:m.start()]); out.append('('+', '.join(new)+')'); last=m.start()+1+len(m.group(1))+1
28. out.append(src[last:])
29. if n:
30. total+=n; changed_files+=1
31. print(f'{f}: {n} lambdas')
32. if apply: open(f,'w',encoding='utf-8').write(''.join(out))
33. print('total lambdas changed',total,'files',changed_files)